yolo进行剪枝、蒸馏、量化研究

一、剪枝研究

个人以ultralyrics-8.3.179版本和torch_pruning，github上某人研究，研究链接：https://github.com/heyongxin233/YOLO-Pruning-RKNN

修改内容：

ultralytics\engine\model.py第798行添加：

 self.trainer.prune = kwargs.get("prune", False)
 self.trainer.prune_ratio = kwargs.get("prune_ratio", 0.5)
 self.trainer.prune_iterative_steps = kwargs.get("prune_iterative_steps", 1)

ultralytics\engine\trainer.py这个替换比较多，使用文件对比工具

配置文件修改：ultralytics\cfg\default.yaml第46行添加下面东西

# prune setting --------------------------------
prune: False # Prune or not
prune_ratio: 0.66874
prune_iterative_steps: 1
prune_load: False
sparse_training: False

二、蒸馏研究

三、量化研究