SpursLipu / YOLOv3v4-ModelCompression-MultidatasetTraining-Multibackbone

YOLO ModelCompression MultidatasetTraining
GNU General Public License v3.0
445 stars 136 forks source link

prune #28

Closed SISTMrL closed 4 years ago

SISTMrL commented 4 years ago

你好,能不能在readme中贴出prune的参考论文链接,小白第一次接触压缩,不知道code中的剪枝是对应哪些论文。还有在进行剪枝稀疏化训练时我查看了代码好像并没有生成新的cfg文件,请问还是使用原来的yolov3.cfg文件吗?还有i请问下,你在训练时percent的参数一般给的是多少,希望能做下参考,十分感谢

SpursLipu commented 4 years ago

network slimming你搜这个就行了

chumingqian commented 3 years ago

@SISTMrL @SpursLipu > 你好,能不能在readme中贴出prune的参考论文链接,小白第一次接触压缩,不知道code中的剪枝是对应哪些论文。还有在进行剪枝稀疏化训练时我查看了代码好像并没有生成新的cfg文件,请问还是使用原来的yolov3.cfg文件吗?还有i请问下,你在训练时percent的参数一般给的是多少,希望能做下参考,十分感谢

你好, 我使用稀疏化训练 ,并且进行python regular-prune.py 规整剪枝这一步骤之后, 生成了剪枝之后的 网络cfg 文件 ,以及对应的权值文件; +------------+----------+----------+ | mAP | 0.804108 | 0.000000 | | Parameters | 61523734 | 12098758 | | Inference | 0.1753 | 0.0866 | +------------+----------+----------+ Config file has been saved: cfg/prune_0.79_yolov3/prune_0.79_yolov3-hand.cfg Compact model has been saved: weights/yolov3_hand_regular_pruning_0.79percent.weights