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

Question

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

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

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

Answer 1

对运行 YOLOv3 的支持已添加到 OpenCV master 分支 (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存储库上的samples/dnn目录中的对象检测示例。

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

希望这有帮助。