你将收获

理解深度学习剪枝问题

理解并掌握FPGM剪枝算法

掌握yolov5剪枝技术

适用人群

小白、新手、有一定深度学习基础的学员!

课程介绍

在深度学习落地过程中,为了适应嵌入端AI算力不足的问题,通常需要对深度学习模型进行压缩,而剪枝技术是最常用的深度学习模型压缩技术之一。本课程结合讲师本身的工程实践,给出基于FPGM算法的yolov5s模型剪枝方法,并在coco数据集上进行剪枝训练,在保证mPA微小下降的前提下,大大提高的yolov5s的inference速度。

剪枝前yolov5s在coco val数据集上评估指标如下:                         

 

FPGM剪枝后在coco val数据集上评估指标如下:

I