YOLO 使用 Open Images Dataset V4 预训练的权重

数据挖掘 喀拉斯 约洛
2022-02-12 22:12:02

我想为某些用例使用对象检测模型。

我从 YOLOv3 开始,因为我需要能够在不到几秒钟的时间内对多张图像执行检测(即使不是实时/视频)。

因此,我使用具有预训练权重的 keras 模型成功加载并生成了预测

问题是该模型的预训练权重是使用 COCO 数据集生成的,该数据集包含很少的类 (80)。

真正让我吃惊的是,我能为这类算法找到的所有预训练权重都使用 COCO 数据集,而没有一个使用Open Images Dataset V4(包含 600 个类)。

这有什么原因吗?如果没有,有没有地方可以找到这样的预训练权重?

1个回答

Darknet 有一个使用 Open Images 的 yolo 实现检查这个页面(向下滚动到最后,就在引用之前)

您可以简单地使用 wget https://pjreddie.com/media/files/yolov3-openimages.weights