yolov5模型压缩-torch_pruning

作者：很楠不爱3 | 2024-05-21 22:15:13

踩

参考论文：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 torch
import math

weights = "best.pt"
model = att1
2
3
4
5
6

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/很楠不爱3/article/detail/604912