hpc203 / yolov5-dnn-cpp-python

用opencv的dnn模块做yolov5目标检测,包含C++和Python两个版本的程序

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

导出ONNX的问题

Caliburne opened this issue · comments

看了您的博文,明白了要想将导出的onnx模型供cv2.dnn调用,需要改写官方源码。但作为小白,怎么改写还是不太懂。能将这个详细过程再讲一遍吗?十分感谢。

:~/code/cplus/yolov5-dnn-cpp-python/build$ ./opencv_test_project
Net use yolov5s
terminate called after throwing an instance of 'cv::Exception'
what(): OpenCV(4.5.0) /home/code/opencv-4.5.0/modules/dnn/src/onnx/onnx_importer.cpp:73: error: (-5:Bad argument) Can't read ONNX file: yolov5s.onnx in function 'ONNXImporter'

g++ main_yolo.cpp -o main -lopencv_dnn -lopencv_imgproc -lopencv_core -lopencv_imgcodecs