YOLOv3-model-pruning icon indicating copy to clipboard operation
YOLOv3-model-pruning copied to clipboard
verified publisher icon Lam1360

关于剪枝方式的问题

Open YuGuii opened this issue 6 years ago • 3 comments

你好，首先很感谢你对于yolo3的剪枝工作。 我想询问一下你对于darknet的卷积层剪枝是直接将需要剪枝的卷积层的权重设为0，还是将该层卷积层从网络中剔除？ 谢谢！

YuGuii avatar Jul 12 '19 09:07 YuGuii

这个项目剪枝方式是卷积核的数量变少

lz20061213 avatar Jul 12 '19 11:07 lz20061213

直接剔除某一层卷积核的 channel

Lam1360 avatar Jul 12 '19 13:07 Lam1360

直接剔除某一层卷积核的 channel

好的，非常感谢，同时还是很期待prune_utils.py文件的，想在自己的训练集上试试效果

YuGuii avatar Jul 12 '19 14:07 YuGuii