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

Is there a limit about image size when we train custom object with already trained models?

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

我遇到了两个模型的问题。当我的数据集包含大小超过 400kb 的图像时,经过训练的模型将不起作用。有时会出现 "allocation of memory" 问题。我通过改变参数(批量大小等)解决了它们。但是还是不知道我们在准备数据集的时候有没有图片大小的限制?

为什么超过 400kb 的图像对我的系统来说是个问题? 我的问题不是关于像素大小,而是关于图像文件大小。

谢谢...

系统信息

在此处(回答部分)提供解决方案,即使它出现在评论部分(感谢 dasmehdix 的更新),也是为了社区的利益

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