Tensorflow对象检测API内存不足

时间:2017-09-26 03:10:15

标签: python tensorflow object-detection-api

我是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已关闭,这就是我所拥有的) error message OOM

批量大小:3

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

我有571个图像来训练模型,还有250个左右的图像用于测试,所有尺寸相同。

有关向模型传递少量数据的建议,以便我可以继续运行吗?

正如其中一条评论建议,我可以调整大小,但这会如何影响我的XML文件?

0 个答案:

没有答案