【发布时间】:2021-05-18 07:06:27
【问题描述】:
我正在尝试进行人体检测。我认为 YOLOv4 适合这个目的。 YOLOv4 使用 COCO 数据集。该数据集包括 80 个类。我运行了 YOLOv4 代码,但它没有给出我想要的好结果。我认为它没有给出我想要的结果的原因是因为 YOLOv4 .weights。这个权重是为 80 个班级创建的。我尝试在自定义数据集上训练 YOLOv4 以获得更好的结果。我使用预训练的权重文件进行训练。但是这个权重也是为 80 个班级创建的。是否有仅为人类检测创建的预训练权重?
【问题讨论】:
标签: machine-learning computer-vision object-detection yolo