项目介绍:
本仓库是基于官方yolov5源码的基础上,进行的改进。
新增支持更换yolov5的backbone主干网络为Efficientnetv2, shufflenetv2, 具体网络模型实现可在model-->中查看。剪枝方面,新增三种剪枝算法的实现,分别为network slimming,sfp,fpgm算法,由于最近比较忙,三种算法并未在visdrone数据集上做测试,但是在其他数据集上已经过测试,效果良好。
后续,将会添加更多更轻量,更优秀的主干网络,比如swintranst等,其他剪枝方法,以及量化,蒸馏对于yolov5系列的支持。
欢迎大家有问题,错误,想法都可以积极交流,探讨,学习,本项目长期维护,持续更新,各位的star,就是对本项目的大力支持。
项目地址(GitHub):/Ranking666/Yolov5-Processing
~~~欢迎各位交流、star、fork、issues~~~