resizeImage_vr01.py

import matplotlib.pyplot as plt

from skimage.color import rgba2rgb
from skimage.transform import resize
from skimage import io, img_as_ubyte

import os
import psutil
import shutil

def resize_all_images_to_8x8():
    for dirpath , _ , filenames in os.walk("."):
        for f in filenames: 
            if f.endswith(".jpg") or f.endswith(".jpeg") or f.endswith(".png"):
#                print(os.path.abspath(os.path.join(dirpath, f)))
                inputImage = os.path.abspath(os.path.join(dirpath, f))

                fac = None

                mode = 'constant'
                
                img = io.imread(inputImage, plugin="matplotlib")
                
                print(img.shape)
                
                plt.imshow(img)
                plt.show()
                
                #mode='constant', cval=0, clip=True, preserve_range=False, anti_aliasing=True, anti_aliasing_sigma=None
                if img.shape[-1] == 4:
                    img_resized = resize( rgba2rgb(img),
                                          (8,8),
                                          mode=mode,
                                          cval=0,
                                          clip=True,
                                          preserve_range=False,
                                          anti_aliasing=False,
                                          anti_aliasing_sigma=fac )
                else:
                    img_resized = resize( img,
                                          (8,8),
                                          mode=mode,
                                          cval=0,
                                          clip=True,
                                          preserve_range=False,
                                          anti_aliasing=False,
                                          anti_aliasing_sigma=fac )
                plt.imshow(img_resized)
                plt.show()
                
                img_resized_binary = img_as_ubyte( img_resized )
                print(img_resized_binary.shape)
                
                if f.endswith(".jpeg"):
                    outputImage = f[:-5]
                else:
                    outputImage = f[:-4]
                
                with open(outputImage+".data","wb") as f:
                    f.write(bytes(img_resized_binary))
                    
def list_data_files():
    data_files = list()
    for dirpath , _ , filenames in os.walk("."):
        for f in filenames: 
            if f.endswith(".data"):
    #            print(os.path.abspath(os.path.join(dirpath, f)))
                data_files.append(os.path.abspath(os.path.join(dirpath, f)))
    return data_files


def get_partition():
    allParitions = psutil.disk_partitions()
    if os.name == 'nt':
        for partition in allParitions:
            if 'removable' in partition.opts:
                return partition.device
    else:
        for partition in allParitions:
            if 'CIRCUITPY' in partition.mountpoint:
                return partition.mountpoint
    raise ValueError('could not find partition')


def copy_data_files_to_board():
    data_files = list_data_files()
    path = get_partition()
    for dir_file in data_files:
        dst = os.path.join(path, os.path.basename(dir_file))
        shutil.copyfile(dir_file, dst)
        print("copy %s %s" % (dir_file, dst))

resize_all_images_to_8x8()
copy_data_files_to_board()