理解深度学习剪枝问题
理解并掌握FPGM剪枝算法
掌握yolov5剪枝技术
你将收获
理解深度学习剪枝问题
理解并掌握FPGM剪枝算法
掌握yolov5剪枝技术
适用人群
课程介绍
在深度学习落地过程中,为了适应嵌入端AI算力不足的问题,通常需要对深度学习模型进行压缩,而剪枝技术是最常用的深度学习模型压缩技术之一。本课程结合讲师本身的工程实践,给出基于FPGM算法的yolov5s模型剪枝方法,并在coco数据集上进行剪枝训练,在保证mPA微小下降的前提下,大大提高的yolov5s的inference速度。
剪枝前yolov5s在coco val数据集上评估指标如下:
FPGM剪枝后在coco val数据集上评估指标如下:
I
课程目录