项目场景:
搞标注数据太麻烦,于是尝试做一个resnet接yolo (v3)的demo,然后出现如下报错,目前仍未解决。先mark下。
问题描述:
具体是,用pytorch建立模型后,把.pth文件转化为.onnx文件,并且用onnx-simplifier简化模型细节。
python3 -m onnxsim old_file.onnx new_file.onnx
得到的onnx模型放到opencv里,在做yolo的detection之前,也就是outs=net.forward()环节出现如下BUG:computeShapeByReshapeMask
outs = net.forward()
cv2.error: OpenCV(4.4.0) /private/var/folders/nz/vv4_9tw56nv9k3tkvyszvwg80000gn/T/pip-req-build-hzfthxlr/opencv/modules/dnn/src/layers/reshape_layer.cpp:154: error: (-1:Backtrace) Can't infer a dim denoted by -1 in function 'computeShapeByReshapeMask'
原因分析:
未知
解决方案:
未知