基于一次性自动化网络剪枝的分类方法及系统技术方案

技术编号:43472025 阅读:18 留言:0更新日期:2024-11-27 13:10
基于一次性自动化网络剪枝的分类方法及系统,涉及人工智能技术领域,针对现有技术对图像数据进行分类,存在效率低的问题,本申请通过引入数据驱动的初始化剪枝方法,可以在图像分类任务中有效减少神经网络的参数和计算需求,提升网络在资源受限环境中的应用性能。本申请能够在保持高识别准确性的同时,提升图像分类处理的效率。

【技术实现步骤摘要】

本专利技术涉及人工智能,具体为基于一次性自动化网络剪枝的分类方法及系统


技术介绍

1、图像识别是神经网络在现实世界中应用的一个重要领域,通过对图像进行分类、检测和分割,广泛应用于自动驾驶、医疗诊断、安防监控等领域。然而,现有图像的分辨率越来越高,数据量越来越庞大,因此,对图像数据的分类越来越困难,存在效率低的问题。


技术实现思路

1、本专利技术的目的是:针对现有技术对图像数据进行分类,存在效率低的问题,提出基于一次性自动化网络剪枝的分类方法及系统。

2、本专利技术为了解决上述技术问题采取的技术方案是:

3、基于一次性自动化网络剪枝的分类方法,所述方法包括以下步骤:

4、获取待分类的多类别图像,并将多类别图像输入训练好的分类网络,得到分类结果,所述训练好的分类网络通过以下步骤得到:

5、步骤一:利用训练好的剪枝网络对分类网络进行参数重要性预测,设定稀疏度程度,并以设定的稀疏度程度为阈值,剪除预测重要性分数低于该阈值的参数,保留重要性不低于该阈值的参数,进而得到剪本文档来自技高网...

【技术保护点】

1.基于一次性自动化网络剪枝的分类方法,其特征在于所述方法包括以下步骤:

2.根据权利要求1所述的面向图像识别网络的一次性自动化网络剪枝方法,其特征在于所述步骤二中多类别图像为标准化处理后的图像,所述标准化处理包括缩放、裁剪、灰度化、归一化以及数据增强。

3.根据权利要求1所述的面向图像识别网络的一次性自动化网络剪枝方法,其特征在于所述分类网络为ResNet、VGGNet或MobileNet。

4.根据权利要求1所述的面向图像识别网络的一次性自动化网络剪枝方法,其特征在于所述迭代剪枝通过SNIP方法进行。

5.根据权利要求4所述的面向图像识...

【技术特征摘要】

1.基于一次性自动化网络剪枝的分类方法,其特征在于所述方法包括以下步骤:

2.根据权利要求1所述的面向图像识别网络的一次性自动化网络剪枝方法,其特征在于所述步骤二中多类别图像为标准化处理后的图像,所述标准化处理包括缩放、裁剪、灰度化、归一化以及数据增强。

3.根据权利要求1所述的面向图像识别网络的一次性自动化网络剪枝方法,其特征在于所述分类网络为resnet、vggnet或mobilenet。

4.根据权利要求1所述的面向图像识别网络的一次性自动化网络剪枝方法,其特征在于所述迭代剪枝通过snip方法进行。

5.根据权利要求4所述的面向图像识别网络的一次性自动化网络剪枝方法,其特征在于所述迭代剪枝具体为:...

【专利技术属性】
技术研发人员:查富生刘晟凯程耀锋张森郭伟孙立宁王鹏飞
申请(专利权)人:哈工大苏州研究院
类型:发明
国别省市:

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

1