本文介绍了使用torch_pruning库对YOLOv5模型进行剪枝的方法，参照了DepGraph论文的任何结构剪枝思想，通过物理移除参数并识别层间依赖来自动裁剪模型。实施剪枝的步骤包括安装torch_pruning库和在train.py中添加微调训练代码。
参考论文：DepGraph: Towards Any Structural Pruning（https://arxiv.org/abs/2301.12900）
主要原理：物理的移除参数，并自动找出层与层以及层之间的依赖，完成模型的自动裁剪
模型压缩效果：
yolov5剪枝流程如下：
pip install torch_pruning
新建prune.py：
import torch_pruning as tp
from models.experimental import attempt_load
import 
订阅专栏 解锁全文
扫一扫
1589
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
