代码之家 › 专栏 › 技术社区 › Vikas

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

object-detection tensorflow

Vikas · 技术社区 · 6 年前

我正在为自定义对象检测训练一个基于tensorflow的预构建模型。我只想探测一种物体。我从不同的角度,在不同的光线条件下拍摄了很多照片。我正在培训K80英伟达GPU。一切正常,当我训练时,我可以看到损失函数降到0.3。但当我开始训练时,损失值很快降到1以下。我使用ssd移动网络作为模型的基本配置。当我尝试测试模型时,它只是在输入图像上画一个大正方形,而不是检测图像中所需的对象。基本上,它无法检测出物体。

我试着用一组不同的mac n chesse图片来训练这个模型,这些图片有很多变化。然后模型运行良好,在输入图像中检测到mac n-chesse图像。但当我有单个物体的照片时,模型就无法检测到。请帮助我理解我在这里做错了什么

1 回复 | 直到 6 年前

Vikas 6 年前

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

推荐文章

Danish Bansal · 如何编辑已注释的图像(矩形边框)?

2 年前

jkn · 在训练时,是否有可能在目标检测中使用较少的数据为类赋予更多的权重?

6 年前

Benjamin · 将tfrecord中的灰度转换为RGB

6 年前

Mark Anthony Sulleza · 一个Tensorflow会话中的多个图

6 年前

Chamod Pathirana · 使用更快的\u rcnn\u inception\u v2\u pets模型运行tensorflow对象检测api时出错

7 年前

Ivan Shelonik · 基于图像帧的tensorflow目标检测api和bboxes

7 年前

Heldap · 为YOLO和SSD型号创建数据集时,为什么要进行注释

7 年前

Ajinkya · 以CSV格式存储Tensorflow对象检测API图像输出和方框

7 年前

Pais · Tensorflow对象检测,生成tfrecord时出错[类型错误:None的类型为NoneType,但应为:int,long]

7 年前

tobycoleman · 使用Tensorflow对象检测API检测图像中的小对象

7 年前