python - 仅检测人类

Question

我正在尝试使用yolo v3在视频中仅检测“人”类。

为此，我进行了以下修改:

I changed number of filters (lines 603, 689, 776 ) from 255 to 18 in yolov3.cfg.
I changed the number of classes from 80 to 1 in the yolov3.cfg and the coco.data files.
I edited the coco.names file in order to leave only the "person" class.

该模型正在运行，但未在视频供稿中检测到任何东西。

您能否提出可能出现问题的建议，并请提供建议？

谢谢，
帕夫洛斯

Answer 1

如果要更改配置文件，则必须更改权重文件-训练自定义网络:instructions。在您的情况下，darknet无法正确解析权重文件，因此不起作用。

或者，您可以使用默认权重和cfg，检测所有对象，但在Darknet源(C代码)中仅使人员通过。