Darknet YOLOv4：仅显示选定的类

Question

我正在尝试使用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.name：

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

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

Answer 1

您不仅可以更改这些文件，还需要为所选的类（如this）重新训练网络

Answer 2

如果您没有时间或资源进行训练，可以通过在原始coco.names文件中用dont_show替换不需要的类来防止绘制其他76个对象的边界框

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