YOLOv8剪枝代码方法（一）

该剪枝代码是基于L1、L2、RandomStrategy进行剪枝

过程如下：

pip install torch_pruning == 0.2.7

剪枝过程如下：

①yolov8-train.py进行训练模型权重，此时fintune = False

②yolov8_pruning.py剪枝yolov8-train.py训练的模型权重

③yolov8-train.py再次训练模型权重，此时fintune = True，微调模型参数（需注意权重需要在GPU上）

④draw_channels.py绘制剪枝前后通道对比

至此，剪枝完成

ID:54300

详询客服 微信shujuqudong1 或shujuqudong6 或 qq68823886 或 27699885

图文详情请查看： http://matup.cn/740267147119.html