Создать файл npz с изображениями и соответствующими ярлыками для CNN? - PullRequest
Новая
       3

Создать файл npz с изображениями и соответствующими ярлыками для CNN?

0 голосов
/ 26 января 2020

Для многомасштабной модели CNN мой набор данных содержит 10 классов по 2880 изображений. Генераторы Keras не эффективны в colab при обучении модели.

Как создать файл npz с изображениями и их метками. 

def get_training_data(datafolder):
        print("Loading training data...")

        training_data = []
        #Finds all files in datafolder
        filenames = os.listdir(datafolder)
        for filename in tqdm(filenames):
            #Combines folder name and file name.
            path = os.path.join(datafolder,filename)
            #Opens an image as an Image object.
            image = Image.open(path)
            #Resizes to a desired size.
            image = image.resize((image_width,image_height),)
            #Creates an array of pixel values from the image.
            pixel_array = np.asarray(image)

            training_data.append(pixel_array)

        #training_data is converted to a numpy array
        training_data = np.reshape(training_data,(-1,image_width,image_height,channels))
        return training_data 
image_width=800
image_height=800
channels=3
x=get_training_data("/content/drive/My Drive/DL_2_Dataset/data/Week1")
x2=get_training_data("/content/drive/My Drive/DL_2_Dataset/data/Week1")

Над кодом, используемым для создания numpy массивов из dir, и 

np.savez_compressed('new', week1=x, week2=x2)````


Saved into npz file,how to load labels for images ?
...