yolov5prune icon indicating copy to clipboard operation
yolov5prune copied to clipboard

请问代码支持yolo5m6、l6模型的剪枝吗

Open yazheng0307 opened this issue 2 years ago • 4 comments

yazheng0307 avatar Aug 15 '22 04:08 yazheng0307

理论上是支持的,之前没有实验过,yolov5l6模型并没有特殊的操作

midasklr avatar Aug 15 '22 07:08 midasklr

请问你解决了嘛,修改成功能剪枝yolo5m6模型了嘛

WZJAI2018 avatar Aug 23 '22 07:08 WZJAI2018

请问你解决了嘛,修改成功能剪枝yolo5m6模型了嘛

直接对m6剪枝会有key找不到报错,我用的方法是直接跳过没有找到的层

yazheng0307 avatar Aug 24 '22 05:08 yazheng0307

@yazheng0307 您好,请问可以分享修改后的代码吗?

zxsitu avatar Aug 24 '22 05:08 zxsitu