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