yolov5量化

Killwaiter

于 2021-11-16 14:43:58 发布

阅读量6.1k

点赞数

CC 4.0 BY-SA版权

分类专栏： pytorch 文章标签：目标检测

本文链接：https://blog.csdn.net/zhangxue1232/article/details/121340277

本文介绍了将yolov5模型进行量化的过程，包括从onnx模型转换为ncnn模型，进行int8量化，并详细说明了环境配置、模型转换、reshape操作修改、yolov5.cpp的调整以及校验数据集和量化参数的匹配。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

量化过程：

模型提取：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

最低0.47元/天解锁文章