yolov5_tensorRT

1.Export tensorrt model

git clone https://github.com/ycdhqzhiai/yolov5_tensorRT
cd yolov5_tensorRT
pip install -r requirements.txt

复制export.py到原始yolov5仓库models目录下
python models/export.py 生成yolov5s.onnx,拷贝到weights目录下

python main.py

python  convert_int8.py --onnx_file  weights/yolov5s.onnx  --data_path $your coustom datasets