Open ZhongmingZhan opened 1 year ago
用v5 7.0版本的官方脚本导出的onnx模型输出的命名是output0,我把官方export.py line 173后面的output0改为output后再导出的就可以了。把作者代码里的取的输出名改为output0应该也行,你可以试试。
作者你好,使用yolov5s.pt导出的onnx模型替换到您的yolov5s.onnx会报错“terminate called after throwing an instance of 'cv::Exception' what(): OpenCV(3.4.16) /home/db/opencv/opencv-3.4.16/modules/dnn/src/dnn.cpp:1406: error: (-2:Unspecified error) Requested layer "output" not found in function 'getLayerData'”,请问这是为什么,应该怎样修改。
edit file YOLOv5Detector.cpp
cv::Mat preds = this->net.forward("output");
-> cv::Mat preds = this->net.forward("output0");
作者你好,使用yolov5s.pt导出的onnx模型替换到您的yolov5s.onnx会报错“terminate called after throwing an instance of 'cv::Exception' what(): OpenCV(3.4.16) /home/db/opencv/opencv-3.4.16/modules/dnn/src/dnn.cpp:1406: error: (-2:Unspecified error) Requested layer "output" not found in function 'getLayerData'”,请问这是为什么,应该怎样修改。