暗网 YOLOv4:仅显示选定内容 类

Darknet YOLOv4: show only selected classes

我正在尝试使用 coco 数据集,只是为了检测 4 类。我使用 YOLOv4 网络。我这样更改了 coco.data 文件:

classes= 4
train = /home/pjreddie/data/coco/trainvalno5k.txt
valid = coco_testdev
#valid = data/coco_val_5k.list
names = data/coco.names
backup = /home/pjreddie/backup/
eval=coco

,而 coco.names 是这样的:

person
car
bus
truck

我还在 yolov4.cfg 文件中将所有 类 行从 80 行更改为 4 行,并根据 3*(类+5) 公式更改了过滤器。当我使用以下方式将电影传递给检测器时:

./darknet detector demo cfg/coco.data cfg/yolov4.cfg yolov4.weights -dont_show /mydrive/praca/videoname1.mkv -thresh 0.4 -i 0 -out_filename results.avi

它没有检测到任何东西。可能是什么问题?

您不能只更改这些文件,您需要为所选 类 重新训练网络,例如

如果您没有时间或资源进行训练,您可以通过将原始 coco.names 文件

person
dont_show
car
dont_show
dont_show
bus
dont_show
truck
dont_show
.
.
.
(and so on)

参考:https://github.com/AlexeyAB/darknet/issues/7318#issuecomment-771603265