ZJU-lishuang / yolov5_prune

yolov5 prune,Support V2, V3, V4 and V6 versions of yolov5
Apache License 2.0
556 stars 136 forks source link

生成mxl的cfg是否能够直接支持剪枝操作 #90

Open AllenZYJ opened 2 years ago

AllenZYJ commented 2 years ago

作者您好,想问一下,我直接生成mxl的cfg是否能够直接支持对应模型的剪枝操作呢

ZJU-lishuang commented 2 years ago

要看网络层是否支持。也就是代码是否能解析cfg文件。不支持可以在代码里添加对不支持的层的支持。

AllenZYJ commented 2 years ago

要看网络层是否支持。也就是代码是否能解析cfg文件。不支持可以在代码里添加对不支持的层的支持。

感谢您的回复,我还想问下您有没有生成cfg的脚本可以参考,如果有的话,那就太感谢了

ZJU-lishuang commented 2 years ago

没有