yolov5剪枝开源分享

AI算法网奇

已于 2022-04-20 13:27:40 修改

阅读量4.8k

点赞数 3

分类专栏：深度学习宝典 yolo 文章标签：剪枝

于 2021-06-19 10:44:34 首次发布

本文链接：https://blog.csdn.net/jacke121/article/details/117490928

版权

深度学习宝典同时被 2 个专栏收录

984 篇文章 710 订阅 ¥29.90 ¥99.00

订阅专栏

超级会员免费看

yolo

68 篇文章 27 订阅

订阅专栏

剪枝分软剪枝和硬剪枝，软剪枝的概念来源于Soft Filter Pruning (SFP)这篇论文，下图阐述了软剪枝(SFP)和硬剪枝(Hard Filter Pruning，HFP)的区别，它们的不同点是HFP在每个epoch后会将卷积核直接剪掉，被剪掉的卷积核在下一个epoch中不会再出现，而SFP在每个epoch后，会将需要剪枝的卷积核仍然保留，只是将其参数置为0，依旧参与下一轮的迭代更新，在所有的epoch循环结束后再进行权重修剪。

剪枝之后，模型变小了4m多，但是cpu速度640*640 150ms。

https://github.com/ZJU-lishuang/yolov5_prune

slim_prune_yolov5s.py 剪枝：

    def prune_and_eval(model, CBL_idx, CBLidx2mask):
        model_copy = deepcopy(model)

        for idx in CBL_idx:
            # bn_module = model_copy.module_list[idx

了解本专栏