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

Question

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

我尝试使用一组不同的 mac n chesse 图像来训练模型，这些图像有很多变化。然后该模型运行良好，并在输入图像中检测到 mac n 国际象棋的图像。但是当我有单个对象的图片时，模型无法检测到。请帮助我了解我在这里做错了什么

Answer 1

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