首页 > 解决方案 > keras imagedatagenerator 不使用部分批处理

问题描述

我正在使用 keras imagedatagenerator(在 tf api 中),在 anaconda 上使用 python 3,并尝试使用 imagedatagenerator.flow_from_directory()

图像数量在批次中的图像数量中不均等,我需要将最后的部分批次排除在训练之外,如果我可以告诉 imagedatagenerator 只是将随机 n 个图像排除在外以保留所有批次在同样的尺寸下,这也是可以的。

如何才能做到这一点?

谢谢

标签: pythontensorflowimage-processingkeras

解决方案


推荐阅读