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

我已经使用tensorflow上的自定义数据集对ssd\u mobilenet\u v2\u coco进行了培训。我还用我的数据集训练了YOLO。我解决了所有的问题，它们都起作用了

这两种模型都有问题。当我的数据集包含大小超过400kb的图像时，经过训练的模型不起作用。有时出现“内存分配”问题。我通过改变参数（批量大小等）解决了这些问题。但我仍然不知道在准备数据集时，图像大小是否有限制

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

谢谢

系统信息

• 英伟达RTX 2060 6G
• AMD Ryzen 7
• 16GB DDR4 2600Mhz Ram
• Cuda:10.0
• CudNN:7.4.2（我也尝试了不同的版本，结果相同）
• 张量流：2

---

在此处提供解决方案（答案部分），即使该解决方案出现在评论部分（感谢dasmehdix的更新）中，也是为了社区的利益

---

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

编辑：否，我们用于训练模型的图像大小没有限制。我的模型中的问题是标签文件中的一些图像大小错误。我解决了我的问题。