OpenCV未知层类型运行暗网检测

Question

我正在尝试从使用Yolo v2切换到Yolo v3，并且无法进行检测。我遇到了错误

OpenCV Error: Parsing error (Unknown layer type: shortcut) in ReadDarknetFromCfgFile

有人知道我该如何解决？使用Yolo v2中的cfg文件可以正常工作。

Answer 1

对运行YOLOv3的支持已添加到OpenCV主分支（3.4.3）。

您应该能够将YOLOv3配置和权重与OpenCV DNN模块一起使用来执行对象检测。

net = cv2.dnn.readNet(args.weights, args.config)

blob = cv2.dnn.blobFromImage(image, scale, (416,416), (0,0,0), True, crop=False)

net.setInput(blob)

outs = net.forward(get_output_layers(net))

完整代码在这里。

看一下OpenCV的github repo上的samples / dnn目录中的对象检测示例。

这篇博客文章（由我撰写）详细说明了如何使用预训练的YOLOv3权重对COCO数据集执行对象检测以检测上下文中的80个常见对象。

希望这可以帮助。