Rust

[OpenCV] DNN failing to load an ONNX file: computeShapeByReshapeMask

Ninestar 2023. 7. 5. 16:32
반응형

[ERROR:0@0.058] global onnx_importer.cpp:1054 cv::dnn::dnn4_v20221220::ONNXImporter::handleNode DNN/ONNX: ERROR during processing node with 2 inputs and 1 outputs: [Reshape]:(onnx_node!/model.22/dfl/Reshape) from domain='ai.onnx'
Traceback (most recent call last):

[ERROR:0@0.058] global onnx_importer.cpp:1054 cv::dnn::dnn4_v20221220::ONNXImporter::handleNode DNN/ONNX: ERROR during processing node with 2 inputs and 1 outputs: [Reshape]:(onnx_node!/model.22/dfl/Reshape) from domain='ai.onnx'
Traceback (most recent call last):
  File ".\run.py", line 12, in <module>
    opencv_net = cv2.dnn.readNetFromONNX(onnx_model_path)

 

[Reshape] 관련 에러 메시지가 뜨면 onnx모델을 opset=12 옵션을 적용해야 합니다

아래 명령어를 입력하여 onnx모델을 새로 다운로드 합니다.

pip install ultralytics
yolo export model=yolov8n.pt imgsz=640 format=onnx opset=12