图像处理模型的剪枝方法、装置及电子设备制造方法及图纸

技术编号:40196730 阅读:19 留言:0更新日期:2024-01-27 00:00
本申请提供一种图像处理模型的剪枝方法、装置及电子设备,涉及软件测试技术领域,所述方法包括:获取样本,样本包括样本图像以及与样本图像对应的参考图像;在已训练的图像处理模型的多个卷积层的多个通道中确定一目标通道,获取在目标通道应用多种不同通道参数的情况下,图像处理模型对样本图像进行处理得到的多个处理结果图像;根据多个处理结果图像和参考图像之间的差异程度,确定目标通道的第一贡献度;根据多个目标通道的第一贡献度,确定每个通道对图像处理模型的重要性,从而对图像处理模型中的通道进行剪枝处理,得到目标图像处理模型,如此,可以在确保明显降低图像处理模型的参数量和计算量的同时保证图像处理模型的图像处理效果。

【技术实现步骤摘要】

本申请涉及图像处理,具体而言,涉及一种图像处理模型的剪枝方法、装置及电子设备


技术介绍

1、随着科技的发展,用于实现图像识别、目标检测等图像处理任务的神经网络模型已在各个领域广泛应用,通常,这些执行图像处理任务的神经网络模型参数量巨大,在计算时也会造成大量计算资源的消耗。

2、目前,通常采用通道剪枝的方法对图像处理模型进行处理,以降低模型的参数量,但这种剪枝方法难以保证剪枝后的图像处理模型的图像处理效果,可能会导致图像处理的效果不好,无法满足图像处理的要求。


技术实现思路

1、为了至少克服现有技术中的上述不足,本申请的目的在于提供一种图像处理模型的剪枝方法、装置及电子设备。

2、第一方面,本申请实施例提供一种图像处理模型的剪枝方法,所述图像处理模型的剪枝方法包括:

3、获取样本,所述样本包括样本图像以及与所述样本图像对应的参考图像;

4、在已训练的图像处理模型的多个卷积层的多个通道中确定一目标通道,获取在所述目标通道应用多种不同通道参数的情况下,所述图像处理模本文档来自技高网...

【技术保护点】

1.一种图像处理模型的剪枝方法,其特征在于,所述方法包括:

2.根据权利要求1所述的图像处理模型的剪枝方法,其特征在于,所述根据多个所述目标通道的所述第一贡献度对所述图像处理模型的通道进行剪枝处理的步骤,包括:

3.根据权利要求2所述的图像处理模型的剪枝方法,其特征在于,所述根据多个所述目标通道的所述第一贡献度对所述图像处理模型的通道进行剪枝处理,得到目标图像处理模型的步骤,包括:

4.根据权利要求2所述的图像处理模型的剪枝方法,其特征在于,所述根据多个所述目标通道的所述第一贡献度对所述图像处理模型的通道进行剪枝处理,得到目标图像处理模型的步骤,包括:...

【技术特征摘要】

1.一种图像处理模型的剪枝方法,其特征在于,所述方法包括:

2.根据权利要求1所述的图像处理模型的剪枝方法,其特征在于,所述根据多个所述目标通道的所述第一贡献度对所述图像处理模型的通道进行剪枝处理的步骤,包括:

3.根据权利要求2所述的图像处理模型的剪枝方法,其特征在于,所述根据多个所述目标通道的所述第一贡献度对所述图像处理模型的通道进行剪枝处理,得到目标图像处理模型的步骤,包括:

4.根据权利要求2所述的图像处理模型的剪枝方法,其特征在于,所述根据多个所述目标通道的所述第一贡献度对所述图像处理模型的通道进行剪枝处理,得到目标图像处理模型的步骤,包括:

5.根据权利要求1所述的图像处理模型的剪枝方法,其特征在于,所述处理结果图像和所述参考图像之间的差异程度l(θ...

【专利技术属性】
技术研发人员:周凡
申请(专利权)人:广州虎牙科技有限公司
类型:发明
国别省市:

网友询问留言 已有0条评论
  • 还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。

1