Как проверить мои дополненные изображения (если они увеличены)? - PullRequest
Новая
       5

Как проверить мои дополненные изображения (если они увеличены)?

0 голосов
/ 14 октября 2019

Я тренирую модель на MobileNetv2. У меня есть примеры, состоящие из 11998 изображений. Я разделил их на обучающие и проверочные изображения в соотношении 9: 1.

Общее количество примеров изображений: 10798

Общее количество изображений для проверки: 1200

Я подготовил данные обучения и проверки в виде: 

#Grabbing the labels from the CSV file (for all 11998 images)
import pandas as pd
training_data_with_label = pd.read_csv("/content/DataSet/train.csv")

#Splitting the data
from sklearn.model_selection import train_test_split
training_set, validation_set = train_test_split(training_data_with_label, random_state = 0, test_size = 0.1)

BATCH_SIZE = 32
TARGET_SIZE = 224
train_dataGen = ImageDataGenerator(rescale = 1./255,
                                   rotation_range=40,
                                   width_shift_range=0.2,
                                   height_shift_range=0.2,
                                      shear_range = 0.2,
                                      zoom_range = 0.2,
                                      horizontal_flip = True,
                                  fill_mode='nearest')

validation_datagen = ImageDataGenerator(rescale=1./255)

train_generator = train_dataGen.flow_from_dataframe(dataframe = training_set, 
                                                    directory="/content/DataSet/Combined_Train_Images",
                                                    x_col="Image_File",
                                                    y_col="Class",
                                                    shuffle=True,
                                                    class_mode="binary",
                                                    #class_mode="categorical",
                                                    target_size=(TARGET_SIZE,TARGET_SIZE),
                                                    batch_size=BATCH_SIZE)

validation_generator = validation_datagen.flow_from_dataframe(dataframe= validation_set,
                                                              directory="/content/DataSet/Combined_Train_Images",
                                                              x_col="Image_File",
                                                              y_col="Class",
                                                              shuffle=True,
                                                              class_mode="binary",
                                                              #class_mode="categorical",
                                                              target_size=(TARGET_SIZE,TARGET_SIZE),
                                                              batch_size=32)

Это (я думаю) означает, что ImageDataGenerator также увеличивает изображения.

Редактировать:

Мне удалось распечатать увеличенные изображения. Вот как я это делаю: 

#Images after augmentation
augmented_images = [train_generator[0][0][0] for i in range(5)]
plotImages(augmented_images)

Я получаю: 1x5 plot

Q1) Где сохраняются дополненные изображения?

Q2) Как проверить, были ли изображения добавлены в существующий набор обучающих данных (с такими же метками)?

...