三、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<