YOLOv3-model-pruning

YOLOv3-model-pruning copied to clipboard

有点看不懂CBL_idx，这是什么index？ What is the difference between pruned_model and compact_model in test_prune.py? 谢谢

containing BN and LeakyReLU?

pruned_model是裁剪了的模型，filters还是跟原始模型一样，compact_model 修改了filters，需要给每一层的权重用init_weights_from_loose_model初始化一下

请问pruned_model里面包括对BN的bias的裁剪吗

@violet17 ，CBL_idx 是卷积层+BN+LeakyReLU的简写而已，C是指卷积

@Lam1360 哦哦 谢谢哈，还有一个问题remaining channel 会不会只剩下一两个channel啊？我现在剪枝完之后就只剩下一两个channel。

layer index: 0 total channel: 32 remaining channel: 19 layer index: 2 total channel: 32 remaining channel: 10 layer index: 6 total channel: 64 remaining channel: 2 layer index: 9 total channel: 64 remaining channel: 24 layer index: 13 total channel: 128 remaining channel: 20 layer index: 16 total channel: 128 remaining channel: 36 layer index: 19 total channel: 128 remaining channel: 27 layer index: 22 total channel: 128 remaining channel: 24 layer index: 25 total channel: 128 remaining channel: 12 layer index: 28 total channel: 128 remaining channel: 11 layer index: 31 total channel: 128 remaining channel: 9 layer index: 34 total channel: 128 remaining channel: 13 layer index: 38 total channel: 256 remaining channel: 61 layer index: 41 total channel: 256 remaining channel: 41 layer index: 44 total channel: 256 remaining channel: 7 layer index: 47 total channel: 256 remaining channel: 10 layer index: 50 total channel: 256 remaining channel: 4 layer index: 53 total channel: 256 remaining channel: 2 layer index: 56 total channel: 256 remaining channel: 18 layer index: 59 total channel: 256 remaining channel: 5 layer index: 63 total channel: 512 remaining channel: 86 layer index: 66 total channel: 512 remaining channel: 7 layer index: 69 total channel: 512 remaining channel: 1 layer index: 72 total channel: 512 remaining channel: 1 layer index: 75 total channel: 512 remaining channel: 94 layer index: 76 total channel: 1024 remaining channel: 190 layer index: 77 total channel: 512 remaining channel: 83 layer index: 78 total channel: 1024 remaining channel: 186 layer index: 79 total channel: 512 remaining channel: 96 layer index: 80 total channel: 1024 remaining channel: 211 layer index: 87 total channel: 256 remaining channel: 59 layer index: 88 total channel: 512 remaining channel: 94 layer index: 89 total channel: 256 remaining channel: 37 layer index: 90 total channel: 512 remaining channel: 87 layer index: 91 total channel: 256 remaining channel: 52 layer index: 92 total channel: 512 remaining channel: 114 layer index: 99 total channel: 128 remaining channel: 29 layer index: 100 total channel: 256 remaining channel: 62 layer index: 101 total channel: 128 remaining channel: 23 layer index: 102 total channel: 256 remaining channel: 43 layer index: 103 total channel: 128 remaining channel: 21 layer index: 104 total channel: 256 remaining channel: 77