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目录下
2.run demo
python main.py
3.tensorrt int8 convert
python convert_int8.py --onnx_file weights/yolov5s.onnx --data_path $your coustom datasets