如何在使用Image augumentation时确定steps_per_epoch,因为它会增加图像数量

时间:2017-12-16 10:28:50

标签: deep-learning keras

model.fit_generator(datagen.flow(X_train,y_train,batch_size=32),epochs=10,steps_per_epoch=5000,validation_data=(X_test,y_test))

我的总数据大小为5000,批量大小为32,那么如何确定steps_per_epoch的值

案例1:何时不使用ImageAugumentation

cas2 2:使用ImageAugumentation时(Coz数字图像会增加以及如何在steps_per_epoch中包含它)

1 个答案:

答案 0 :(得分:5)

steps_per_epoch将是训练集中(扩充前)的样本总数除以批量大小。所以 -

steps_per_epoch = 5000 / 32 ~ 156

使用数据扩充不会影响此计算。您还可以在我Training a CNN with Keras的视频中获取有关使用此参数以及fit_generator()的更多信息。 steps_per_epoch报道大约在4:08开始。