关于Keras Fit Generator批次大小的问题

时间:2019-10-07 16:33:20

标签: python machine-learning keras

我正在使用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设置的数字肩是什么?

0 个答案:

没有答案