当我们使用已经训练好的模型训练自定义对象时,图像大小是否有限制?
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 的图像对我的系统来说是个问题? 我的问题不是关于像素大小,而是关于图像文件大小。
谢谢...
系统信息
- 英伟达 RTX 2060 6G
- AMD 锐龙 7
- 16GB DDR4 2600Mhz 内存
- Cuda:10.0
- CudNN:7.4.2(我也尝试了不同的版本,结果相同)
- 张量流:2
在此处(回答部分)提供解决方案,即使它出现在评论部分(感谢 dasmehdix 的更新),也是为了社区的利益。
否,我们用于训练模型的图像大小没有限制。
我已经在 tensorflow 上使用我的自定义数据集训练 ssd_mobilenet_v2_coco
。我也用我的数据集训练了 YOLO。我解决了所有问题,它们都有效。
我遇到了两个模型的问题。当我的数据集包含大小超过 400kb 的图像时,经过训练的模型将不起作用。有时会出现 "allocation of memory"
问题。我通过改变参数(批量大小等)解决了它们。但是还是不知道我们在准备数据集的时候有没有图片大小的限制?
为什么超过 400kb 的图像对我的系统来说是个问题? 我的问题不是关于像素大小,而是关于图像文件大小。
谢谢...
系统信息
- 英伟达 RTX 2060 6G
- AMD 锐龙 7
- 16GB DDR4 2600Mhz 内存
- Cuda:10.0
- CudNN:7.4.2(我也尝试了不同的版本,结果相同)
- 张量流:2
在此处(回答部分)提供解决方案,即使它出现在评论部分(感谢 dasmehdix 的更新),也是为了社区的利益。
否,我们用于训练模型的图像大小没有限制。