shaoshengsong / DeepSORT

support deepsort and bytetrack MOT(Multi-object tracking) using yolov5 with C++
812 stars 188 forks source link

weight权重文件 #49

Open ZhongmingZhan opened 1 year ago

ZhongmingZhan commented 1 year ago

作者你好,使用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'”,请问这是为什么,应该怎样修改。

KinokoC commented 7 months ago

用v5 7.0版本的官方脚本导出的onnx模型输出的命名是output0,我把官方export.py line 173后面的output0改为output后再导出的就可以了。把作者代码里的取的输出名改为output0应该也行,你可以试试。

YangSangWan commented 6 months ago

作者你好,使用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");