0

我是 tensorflow 的新手,正在尝试训练自己的对象检测模型。

硬件:尝试了很多东西,但最大的是 32gb cpu。还尝试了 8gb cpu 和 2gb gpu。

所有正在运行的窗口,python 3.6.2 和 tensorflow 1.3

我一直在关注这里的指南https://pythonprogramming.net/introduction-use-tensorflow-object-detection-api-tutorial/

错误是:(对不起,VM 已关闭,这就是我所拥有的) 错误消息 OOM

批量:3

我开始认为我的问题是初始图像大小(4mb,4288x3216),所以我一直在尝试其他资源(纸张空间)并且仍然在 8gb gpu 上耗尽内存。

我有 571 张图像来训练模型,还有 250 多张图像用于测试,大小都一样。

关于传递模型少量数据以便我可以保持运行的任何建议?

作为建议的评论之一,我可以调整大小,但这将如何影响我的 XML 文件?

4

0 回答 0