【问题标题】:Tensorflow Object Detection Unusually large bounding boxes and wrong resultsTensorFlow 对象检测异常大的边界框和错误的结果
【发布时间】:2018-10-29 08:19:33
【问题描述】:

我正在 TensorFlow 中构建一个对象检测器来检测有和没有头盔的摩托车骑手,我有 1000 张图像,分别用于有头盔的骑手、没有头盔的骑手和行人(加起来 -- 3000 张图像),我的最后一个检查点是 35267 步,我已经使用交通视频进行了测试,但我看到异常大的边界框结果错误。有人可以解释这种检测的原因吗?我需要等待至少 50000 步吗?或者我需要添加数据集(与交通摄像头成角度的图像)吗?

模型 - SSD Mobilenet COCO - 自定义对象检测, 培训平台 - Google Colab

请查找附件Video Snapshot 1

Video Snapshot 2

第 2 天 - 2018 年 10 月 30 日

我今天用图像进行了测试,我得到了不同的结果,似乎是正确的,如果我用图像中的单个对象进行测试,第二天。请查找结果 Single Object IMage Test 1 Single Object Image Test 2

经过测试的检查点 - 52,000 步

但是,如果我在道路上使用多个对象的图像进行测试,检测结果是错误的,并且边界框变大了,是不是因为数据集,因为我正在与 One Motorbike 骑手(带或不带头盔)一起训练) 每张图片。

请找出错误的结果

Multi Object Image Test Multi Object Image Test

我也测试了场景中所有摩托车的图像,在这种情况下,我没有得到任何结果,请查找图像

No Result Image No Result Image

结果很混乱,有什么我遗漏的吗?,

【问题讨论】:

  • 请分享您的.config 文件,用于您的培训。所以我们可以提供更好的建议。

标签: tensorflow bounding-box object-detection-api


【解决方案1】:

没有必要等到 50000 epocs 你应该在 35k 甚至 10k 中获得不错的结果。我建议

  1. 再次检查您的数据集并检查所有边界框(数据清理)
  2. 使用推理代码检查您的模型是否有批量标准化等更改
  3. 添加更多具有不同特征、角度和颜色复杂性的数据

在继续之前,我会检查这些要点。

【讨论】:

  • 谢谢@Smit Shilu。我会处理上述建议并回复您。
  • 首先,我可以调整较大图像的大小并保持图像大小和边界框的相对统一,因为我的一些图像是 920x800 和一些是 600x400 ...
  • 您能否分享您对我第 2 天测试结果的看法。
  • 看起来您没有使用不同的图像数据集进行训练。您正在对一些近距离拍摄的图像进行训练,这会给您带来非常好的结果,但是当您尝试从上方或侧面拍摄的图像时,它不起作用。我建议添加适当数量的所有类型的图像,然后再次训练。
  • 是的,这是一个很好的建议,我已经开始准备不同角度的数据集,我们已经将图像尺寸缩小到 720 像素作为标准,您能否分享您对以下问题的想法.1) 我可以在同一张图片中标注多个类别标签。在诸如戴头盔的摩托车骑手和不戴头盔的后座骑手或Vice Versa之类的情况下会有所帮助吗? 2) 我可以复制相同的图像来增加数据集的大小吗?
猜你喜欢
  • 2020-07-20
  • 1970-01-01
  • 2019-04-22
  • 2014-08-02
  • 1970-01-01
  • 2020-07-23
  • 1970-01-01
  • 2021-02-09
  • 1970-01-01
相关资源
最近更新 更多