使用tensorflow对单个对象进行培训时使用自定义对象检测的问题
使用tensorflow对单个对象进行培训时使用自定义对象检测的问题
提问于 2018-06-29 18:17:00
我正在训练一个预先建立的基于tensorflow的模型，用于定制对象检测。我只想检测到一种类型的对象。我从不同的角度，在不同的光照条件下拍摄了很多照片。我正在K80 Nvidia GPU上训练。一切正常，当我训练时，我可以看到损失函数下降到0.3。但是当我开始训练的时候，损失值很快下降到1以下。我使用SSD移动网络作为模型的基本配置。当我尝试测试模型时，它只是在输入图像上画一个大的正方形，而不是在图像中检测到想要的对象。基本上，它无法检测到对象。
我试着用一组不同的mac图像训练模型，这些图像有很多不同的变化。然后，该模型在输入图像中对mac的图像进行了精细的检测。但是当我有单个对象的图片时，模型就无法检测到。请帮助我理解我在这里做错了什么
回答 1
Stack Overflow用户
回答已采纳
发布于 2018-07-11 03:49:57
问题是我的训练数据集。我没有正确地将对象从原始图像中剪切出来。另外，我需要大约300张图片来正确地训练模型。SSD工作良好后，提供了一个良好的裁剪图像。
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/51107377
复制相关文章
相似问题
腾讯云开发者
