search

tanluren / yolov3-channel-and-layer-pruning

yolov3 yolov4 channel and layer pruning, Knowledge Distillation 层剪枝，通道剪枝，知识蒸馏
Apache License 2.0
1.5k stars 446 forks source link

yolov3-spp-pan-scale.cfg #45

Open JingyuLi-code opened 4 years ago

JingyuLi-code commented 4 years ago

想问一下，目前项目支持yolov3和yolov3-spp。如何修改使用yolov3-spp-pan-scale.cfg？ 基础训练和稀疏训练可以支持yolov3-spp-pan-scale.cfg，在层和通道剪枝运行layer_channel_prune.py 时AttributeError: 'NoneType' object has no attribute 'reshape'，如何修改这部分剪枝代码？

tanluren commented 4 years ago

嗯有意思。。。有空写多几个支持的结构。其实也不难

JingyuLi-code commented 4 years ago

嗯有意思。。。有空写多几个支持的结构。其实也不难

嗯，感谢认可，yolov3这块比较熟悉，剪枝压缩这块代码原理还不太懂，修改了部分代码，现在有一个问题，对于panet的pan结构，#121 [route] layers=1

122

[reorg3d] stride=2

123

[route] layers=5,-1

124

[reorg3d] stride=2

125

[route] layers=12,-1

126

[reorg3d] stride=2

127

[route] layers=37,-1

128

[reorg3d] stride=2

129

[route] layers=62,-1 在 slim_prune.py 时怎么处理，对应的CBLidx2mask怎么得到，哈哈，感谢回复交流。

chengzi999 commented 4 years ago

想问一下，目前项目支持yolov3和yolov3-spp。如何修改使用yolov3-spp-pan-scale.cfg？ 基础训练和稀疏训练可以支持yolov3-spp-pan-scale.cfg，在层和通道剪枝运行layer_channel_prune.py 时AttributeError: 'NoneType' object has no attribute 'reshape'，如何修改这部分剪枝代码？

我的问题和你一样，请问你解决了吗？