在单一类型的对象上训练时使用tensorflow进行自定义对象检测的问题

Question

我正在训练用于自定义对象检测的基于预构建张量流的模型。 我只想检测一种对象。我从不同的角度和不同的光照条件下拍摄了很多图像。我正在使用K80 Nvidia GPU进行培训。一切正常，当我训练时，我可以看到损失函数降至0.3。但是当我开始训练时，损耗值很快就会下降到1以下。我正在使用SSD mobile Net作为该模型的基本配置。当我尝试测试模型时，它只是在输入图像上绘制了一个大正方形，而不是在图像中检测到所需的对象。基本上，它无法检测到对象。

我试图用一组具有很大差异的Mac n Chesse的不同图像来训练模型。然后，该模型运行良好，并在输入图像中检测到了棋盘图像。但是，当我有单个对象的图片时，该模型将无法检测到。请帮助我了解我在这里做错了

Answer 1

问题出在我的训练数据集上。我没有从原始图像正确裁剪对象。我还需要大约300张图像来正确训练模型。提供良好裁剪的图像后，SSD运作良好。