yolov5模型压缩-PAGCP

参考论文:Performance-aware Approximation of Global Channel Pruning for Multitask CNNs(https://arxiv.org/pdf/2303.11923.pdf)
基本原理:研究不同卷积核之间的联合重要性来实现全局剪枝策略
模型压缩效果
在这里插入图片描述
在yolov5上进行剪枝训练,流程如下:
1、按照yolo正常训练
2、和models、utils同级新建一个文件夹pagcp用来存放相关文件,文件下新建prune_zoo.py,添加如下代码:


import torch
from torch_pruning.prune.strategy import round_pruning_amount


def group_l1prune(model, group, compress_rate, round_to=1):
    "

相关推荐

  1. yolov5模型压缩-shufflenetv3

    2024-03-13 18:22:05       36 阅读
  2. 博客摘录「 YOLOv5模型剪枝压缩」2024年5月11日

    2024-03-13 18:22:05       37 阅读
  3. yolov5导出onnx模型问题

    2024-03-13 18:22:05       47 阅读

最近更新

  1. docker php8.1+nginx base 镜像 dockerfile 配置

    2024-03-13 18:22:05       94 阅读
  2. Could not load dynamic library ‘cudart64_100.dll‘

    2024-03-13 18:22:05       100 阅读
  3. 在Django里面运行非项目文件

    2024-03-13 18:22:05       82 阅读
  4. Python语言-面向对象

    2024-03-13 18:22:05       91 阅读

热门阅读

  1. data engineer

    2024-03-13 18:22:05       41 阅读
  2. 对盒子模型的理解(box-sizing)

    2024-03-13 18:22:05       48 阅读
  3. 【子串】76. 最小覆盖子串【困难】

    2024-03-13 18:22:05       39 阅读
  4. ElasticSearch集群的备份和恢复

    2024-03-13 18:22:05       39 阅读
  5. 保研复习数据结构记(5)--并查集

    2024-03-13 18:22:05       46 阅读
  6. yield代码解释

    2024-03-13 18:22:05       42 阅读
  7. 蓝桥杯 图形排版

    2024-03-13 18:22:05       45 阅读