我現在使用ImageDataGenerator的資料擴增,並想將強化後的資料儲存 但是在設定的資料夾裡面卻完全沒有資料產生 該如何處理呢?
from keras.preprocessing.image import ImageDataGenerator
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
)
datagen = ImageDataGenerator(rotation_range=10, width_shift_range=0.1,
height_shift_range=0.1,shear_range=0.15,
zoom_range=0.1,channel_shift_range = 10, horizontal_flip=True)
train_data='G:/cnn/py/CNN/train/'
save_here ='G:/cnn/py/CNN/train2'
train_generator = train_datagen.flow_from_directory(
train_data,
batch_size=10,
class_mode='categorical',
save_to_dir=save_here,
)
這是執行成果