我在尝试进行人体检测。我认为YOLOv4非常适合这个目的。YOLOv4使用的是COCO数据集。这个数据集包含80个类别。我运行了YOLOv4的代码,但结果并不如我所愿。我认为没有达到我想要的结果的原因是YOLOv4的权重。这个权重是为80个类别创建的。我尝试在自定义数据集上训练YOLOv4以获得更好的结果。我使用了一个预训练的权重文件进行训练,但这个权重也是为80个类别创建的。是否有专为人体检测创建的预训练权重?
回答:
您可以按照这个问题中详细说明的步骤,仅针对人类训练您的数据库。我也这样做了,结果效率很好。