三、INT8量化 C++
3.1下载coco128数据集
cd /mnt/workspace/yolov5/data/scripts
sh get_coco128.sh
3.2 模型准备
#onnx转simple_onnx
pip install onnx-simplifier
python -m onnxsim yolov5s.onnx yolov5s-simple.onnx
3.3 下载量化代码库
git clone https://github.com/Wulingtian/yolov5_tensorrt_int8_tools.git
#安装pycuda
pip install pycuda
cd yolov5_tensorrt_int8_tools
vim<