收录:
摘要:
本发明公开了基于轻量化神经网络的图像分类模型构建和小样本图像分类方法,利用构建的轻量化神经网络用于小样本图像分类,运用训练得到的图像分类模型可以迁移学习到其他小样本分类任务。建立一种轻量化神经网络结构,包括:DWConv、倒残差结构、GConv、ShuffleNetUnit;使用小样本数据集,对搭建的轻量化神经网络进行训练,生成图像分类模型,以此模型来完成图像分类任务。本发明搭建的轻量化神经网络可以在保证分类精度的前提下,获得好的特征图以及分类参数,降低卷积神经网络的计算量。
关键词:
通讯作者信息:
电子邮件地址: