yolov3-channel-and-layer-pruning icon indicating copy to clipboard operation
yolov3-channel-and-layer-pruning copied to clipboard

yolov3-spp-pan-scale.cfg

Open JingyuLi-code opened this issue 5 years ago • 3 comments
trafficstars

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

JingyuLi-code avatar Mar 14 '20 13:03 JingyuLi-code

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

tanluren avatar Apr 03 '20 15:04 tanluren

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

嗯,感谢认可,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怎么得到,哈哈,感谢回复交流。

JingyuLi-code avatar Apr 03 '20 15:04 JingyuLi-code

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

我的问题和你一样,请问你解决了吗?

chengzi999 avatar Aug 28 '20 18:08 chengzi999