YOLOv4 训练:无法在单个图像上检测多个类

问题描述

我使用 Open Images 数据集为两个类训练 CNN:人、手机。问题是 Yolo 每张图像只检测 1 个类。例如,如果图像包含人和手机,则只会检测到手机。 Example Output。如果只有人类,那么所有人类都会被成功检测到。我认为问题在于数据集只有每个图像标记了 1 个类的图像。我在数据集中有很多图像,其中两个类都存在,但只有其中一个用标签标记。这可能是问题吗?

我该如何解决问题?

  1. 添加几张标有两个类的图像
  2. 标记所有图像以确保每个包含电话和人物的图像都为每个对象都有适当的标签
  3. 或者也许有一种方法可以下载数据集,以便正确标记两个类?

这是我的 .names 文件中的内容:

Person
Mobile_phone

这里是 .data 文件:

classes = 2
train = data/train.txt
valid = data/test.txt
names = data/obj.names
backup = /mydrive/yolov4/backup

感谢您的帮助!

解决方法

暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!

如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@)