我正在使用keras ImageDataGenerator和flow_from_directory来解决回归问题。有些参数我不知道。
这是我的数据结构:
data\
1st label folder(e.g. 1.3)\
image for 1st label
...
2nd label folder\
image for 2nd label
...
...
这是一些代码:
gen = ImageDataGenerator(width_shift_range=0.2,height_shift_range=0.1,..........)
train_datagen = gen.flow_from_directory(path, batch_size=?, class_mode='sparse')
def regression_flow_from_directory(gen, label_list):
for x, y in flow_from_directory_gen:
y=y.astype('int32')
yield x, label_list[y]
model.fit_generator(train_datagen ,steps_per_epoch=?,.......)
Q1。如果每个标签文件夹中的图像很少(假设2张)。我为batch_size设置的数字是多少?
Q2。与Q1相关。如果我有10000个标签文件夹(即总共20000张图像)。我为steps_per_epoch设置的数字肩是什么?