我是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文件?