ZJU-lishuang / yolov5_prune

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

咨询个问题哈,通过这种剪枝,那么删除了一些层和通道,那么会导致转换ncnn有些问题吧,相当远原网络结构变了 #7

Closed EricHuiK closed 3 years ago

ZJU-lishuang commented 3 years ago

If you want to keep the network structure the same ,you can refer to Pruning/Sparsity Tutorial. this project provides the the pruned network structure in cfg.