我正在从Tensorflow对象检测模型动物园训练ssd_mobilenet_v1_coco_2017_11_17模型。我的数据集是卫星图像,我的目标是检测图像中的车辆。但是由于python内存问题,训练失败了。我正在训练CPU,我的Windows10机器有32 gb的RAM。用于训练的TF记录文件大小约为1.7 GB。
我无法检测到此失败的原因。
请帮帮忙。
发布于 2017-11-28 00:08:13
尝试调整批处理大小、队列容量、读取器线程。
https://stackoverflow.com/questions/47484883
复制相似问题