lishuang
lishuang
剪枝减少了参数量和计算量,如果速度瓶颈不在这里,就不会有速度提升。该项目没有对层剪枝,只对通道剪枝,对层剪枝改变了网络结构,影响比较大。
数据格式都是一样的,很容易配置训练文件
没遇到过相关问题
https://github.com/ZJU-lishuang/yolov5-v6
项目[yolov5](https://github.com/ZJU-lishuang/yolov5-v6)
please refer to https://github.com/tanluren/yolov3-channel-and-layer-pruning. No experience in distillation.
修改配置文件cfg中的类别数吧,具体可以边debug边修改 这代码跑过多个模型,不只是适配单类别的
不建议零基础直接来剪枝
直接复制地址到下载器中下载
多选一