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.

该模型运行但未检测到视频源中的任何内容。

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

谢谢，巴甫洛斯

标签： pythonopencvdarknet

Answer 1

如果要更改配置文件，则必须更改权重文件 - 训练您的自定义网络：说明。在您的情况下，暗网无法正确解析权重文件并且不起作用。

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