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

Question

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

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

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

Answer 1

OpenCV 主分支 (3.4.3) 添加了对运行 YOLOv3 的支持。

您应该能够将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 目录中的对象检测示例。

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

希望这可以帮助。