量化过程:
模型提取:onnx
模型转化:ncnn
int8量化
一、环境
shufflev2-yolov5的源码和权重 地址链接:https://github.com/ppogg/shufflev2-yolov5
ncnn部署:Linux环境下ncnn安装+protobuf+opencv_zhangxue1232的博客-CSDN博客
二、onnx模型提取
cd /home/ding/zhangxue/YOLOv5-Lite/YOLOv5-Lite/
#将.pt文件用onnx转化为.onnx文件
python models/export.py --weights weights/v5lite-s.pt --img 640 --batch 1
#onnxsim 精简模型
python -m onnxsim weight/v5lite-s.onnx weight/yolov5-lite-sim.onnx
三、