掌握YOLOv8模型剪枝方法
学习Network Slimming模型剪枝原理和代码
你将收获
掌握YOLOv8模型剪枝方法
学习Network Slimming模型剪枝原理和代码
适用人群
课程介绍
YOLOv8是一个当前非常流行的目标检测器,本课程使用Network Slimming(网络瘦身)剪枝方法对YOLOv8进行模型剪枝,使其更加轻量和实用。
Network Slimming是一种经典实用的模型压缩方法,可实现方便高效的通道级别的结构化剪枝。
该方法通过利用BN(Batch Normalization)层中缩放因子较小的值来剪裁的相应通道,达到精简网络的目的。步骤包括:增加稀疏正则化的训练、网络模型剪枝、剪枝后的网络模型微调。
本课程在YOLOv8 v8.1版本代码的基础上增加Network Slimming剪枝方法,在Windows和Ubuntu系统上演示针对自己的数据集训练和进行网络剪枝过程,并讲解原代码针对剪枝的修改和增加部分。
本课程分为原理篇、实战篇、代码讲解篇。
课程目录