TensorFlow对象检测训练让python崩溃

Question

我正在训练来自 Tensorflow 对象检测模型动物园的 ssd_mobilenet_v1_coco_2017_11_17 模型。 我的数据集是卫星图像，我的目标是检测图像中的车辆。 但训练因 python 内存问题而失败。我正在 CPU 上进行培训，我的 Windows 10 机器有 32 GB RAM。用于训练的 TF 记录文件大小约为 1.7 GB。

我无法检测到此失败的原因。

请帮忙。

Answer 1

尝试调整批量大小、队列容量、读取器线程。