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

Question

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中包含它)

Answer 1

该steps_per_epoch会是由一批大小除以你的训练集(增强前)样品总数.所以 -

steps_per_epoch = 5000 / 32 ~ 156

使用数据扩充不会影响此计算.您还可以获得有关使用此参数的更多信息,以及fit_generator()有关使用Keras培训CNN的视频中的信息.该steps_per_epoch报道开始围绕4:08.