为什么shortcut_prune剪枝方法中，上采样前的卷积层不能进行剪枝？

coldlarry / YOLOv3-complete-pruning

提供对YOLOv3及Tiny的多种剪枝版本，以适应不同的需求。

793 stars 214 forks source link

Open worldzhang1996 opened 3 years ago

worldzhang1996 commented 3 years ago

在prune_utils.py文件中的 line 60-62 中代码如下： #上采样层前的卷积层不裁剪 ignore_idx.add(84) ignore_idx.add(96)

按照我的个人理解，上采样的话是只影响H和W两个维度，BN剪枝的话主要是对channel这个维度进行修剪。如果是对上采样前的卷积层进行剪枝，那么其实并不影响上采样操作？

hzq-zjm commented 3 years ago

你好，请问理解了吗，我也有相同的困惑，望解答一下谢谢