公开(公告)号：CN116992945A

公开(公告)日：2023-11-03

申请号：CN202311257346.5

申请日：2023-09-27

Applicant: 之江实验室

Inventor： 刁博宇 ,  杨雨 ,  李超

IPC: G06N3/082 ,  G06N3/063 ,  G06V10/82 ,  G06N3/0464

Abstract: 本发明公开了一种基于贪心策略反向通道剪枝的图像处理方法及装置，该方法包括：获取图像数据集和待剪枝图像处理网络，对待剪枝图像处理网络进行预训练，得到预训练模型，其中预训练过程中所使用的图像集与图像数据集分布相似或相同；将预训练模型划分为特征提取量化模块和输出模块，其中特征提取量化模块作为剪枝对象；根据基于贪心策略的通道选择方法对特征提取量化模块进行反向的逐层通道剪枝，并在每层剪枝完成后进行预定轮次的微调；对进行反向的逐层通道剪枝后的预训练模型在图像数据集上进行一定轮次的微调，得到微调后的图像处理模型；将微调后的图像处理模型部署于终端，以通过微调后的图像处理模型进行图像处理。

公开(公告)号：CN116992945B

公开(公告)日：2024-02-13

申请号：CN202311257346.5

申请日：2023-09-27

Applicant: 之江实验室

Inventor： 刁博宇 ,  杨雨 ,  李超

IPC: G06N3/082 ,  G06N3/063 ,  G06V10/82 ,  G06N3/0464

Abstract: 本发明公开了一种基于贪心策略反向通道剪枝的图像处理方法及装置，该方法包括：获取图像数据集和待剪枝图像处理网络，对待剪枝图像处理网络进行预训练，得到预训练模型，其中预训练过程中所使用的图像集与图像数据集分布相似或相同；将预训练模型划分为特征提取量化模块和输出模块，其中特征提取量化模块作为剪枝对象；根据基于贪心策略的通道选择方法对特征提取量化模块进行反向的逐层通道剪枝，并在每层剪枝完成后进行预定轮次的微调；对进行反向的逐层通道剪枝后的预训练模型在图像数据集上进行一定轮次的微调，得到微调后的图像(56)对比文件吴雨茜;王俊丽;杨丽;余淼淼.代价敏感深度学习方法研究综述.计算机科学.2019,(05),全文.