我已经训练了 ssd_mobilenet_v2_coco 用我在tensorflow上的自定义数据集。另外我也用我的数据集训练了YOLO。我解决了所有的问题，它们都能正常工作。
我在这两个模型中遇到了一个问题。当我的数据集包含的图片大小超过400kb时，训练的模型就无法工作。有些时候 "allocation of memory" 问题的发生。我通过改变参数（批量大小等）解决了这些问题。但是我不知道在准备数据集时，是否有图像大小的限制？
为什么超过400kb的图像对我的系统来说是个问题？我的问题不是像素大小的问题，而是图像文件大小的问题。
谢谢...
在这里提供解决方案（答案部分），即使它是存在于评论部分（感谢dasmehdix的更新）的 造福社会.
没有，我们用来训练模型的图像大小没有限制。