小编Mat*_*qua的帖子

使用 .onnx 时 Yolov5 OpenCV 错误:(-215:断言失败)

我刚刚创建了一个 yolov5 模型,并将其导出为 onnx 格式,因此它可以与 opencv 一起使用,但我不断收到错误:

[错误:0] 全局 D:\a\opencv-python\opencv-python\opencv\modules\dnn\src\dnn.cpp (3554) cv::dnn::dnn4_v20211004::Net::Impl::getLayerShapesRecursively OPENCV /DNN: [Reshape]:(466): getMemoryShapes() 抛出异常。输入=1 输出=1/1 blob=0 [错误:0] 全局 D:\a\opencv-python\opencv-python\opencv\modules\dnn\src\dnn.cpp (3557) cv::dnn:: dnn4_v20211004::Net::Impl::getLayerShapes递归输入[0] = [ 1 24 52 52 ] [ERROR:0] 全局 D:\a\opencv-python\opencv-python\opencv\modules\dnn\src\dnn. cpp (3561) cv::dnn::dnn4_v20211004::Net::Impl::getLayerShapes递归输出[0] = [ ] [错误:0] 全局 D:\a\opencv-python\opencv-python\opencv\modules\ dnn\src\dnn.cpp (3567) cv::dnn::dnn4_v20211004::Net::Impl::getLayerShapesRecursively 异常消息:OpenCV(4.5.4) D:\a\opencv-python\opencv-python\opencv\模块\dnn\src\layers\reshape_layer.cpp:106:错误:(-215:断言失败)total(srcShape,srcRange.start,srcRange.end)== maskTotal在函数'cv::dnn::computeShapeByReshapeMask'中

这是代码:

onnx_model = onnx.load('best.onnx')
onnx.checker.check_model(onnx_model)
net = cv.dnn.readNetFromONNX('best.onnx')


classes = []

with open("coco-dataset.labels", "r") as f:
    classes = [line.strip() for line in f.readlines()]

layerNames = net.getLayerNames()
outputLayers=[]
for i in …
Run Code Online (Sandbox Code Playgroud)

python opencv onnx yolov5

4
推荐指数
1
解决办法
6604
查看次数

标签 统计

onnx ×1

opencv ×1

python ×1

yolov5 ×1