当我们使用已经训练好的模型训练自定义对象时，图像大小是否有限制？

Question

我已经使用我在tensorflow上的自定义数据集训练了ssd_mobilenet_v2_coco。我也用数据集训练了YOLO。我解决了所有问题，它们就起作用了。

两种型号都遇到问题。当我的数据集包含大小超过400kb的图像时，经过训练的模型不起作用。有时发生"allocation of memory"问题。我通过更改参数（批量大小等）解决了它们。但是我仍然不知道在准备数据集时图像大小是否受到限制？

为什么我的系统有超过400kb的图像问题？ 我的问题不是像素大小，而是图像文件大小。

谢谢...

系统信息

• Nvidia RTX 2060 6G
• AMD锐龙7
• 16GB DDR4 2600Mhz Ram
• CUDA：10.0
• CudNN：7.4.2（我也尝试使用不同版本，并且会出现相同的结果）
• Tensorflow：2

Answer 1

即使在注释部分（感谢dasmehdix进行更新）中提供了解决方案，也可以在此处（答案部分）提供该解决方案，以获取社区的好处。

否，我们用于训练模型的图像大小没有限制。