wenjuntan / ONNX-yolov5

deploy yolov5 in c++

Introdution

Deploy ultralytics Yolov5 pretained model with C++ language ;

Env

1. GCC 7.5
2. Opencv 4.5.4

Get ONNX Model

1. go to yolov5 release page download yolov5 pretrained model（official onnx can't work right)，such as yolv5s.pt

2. usetrace.py convert yolov5s.pt to yolov5.onnx:

   git clone https://github.com/ultralytics/yolov5.git
   cd yolov5
   python3.7 export.py --data data/coco128.yaml --weights yolov5s.pt --include onnx

Build

git clone https://github.com/Hexmagic/ONNX_yolov5.git
mkdir build&&cd build
cmake ..
make -j4
cd ..
./build/main yolov5s.onnx data/images/zidane.jpg