[发明专利]基于原型网络的小样本垃圾图像分类方法

说明书

本发明公开了一种基于原型网络的小样本垃圾图像分类方法，该方法需要少量垃圾图像样本就能完成对垃圾图像的自动分类识别，能够提高垃圾处理过程中的自动分拣效率。包括：首先使用摄像头采集若干垃圾图像构建垃圾分类数据集并使用伽马校正方法对图像进行预处理，然后使用四层卷积神经网络构建原型网络，接着计算查询样本类别概率输出值与真实标签值之间的损失函数对原型网络进行训练，最后固定原型网络中的参数对垃圾图像进行分类测试。

技术领域

本发明涉及计算机视觉领域，特别涉及基于原型网络的小样本垃圾图像分类方法。

背景技术

随着经济的快速增长，城市生活垃圾总量也随之逐年的大量增加。对这些垃圾进行及时有效地快速处理对于创造优美的生活环境，节约地球资源，促进社会的可持续发展具有重要意义。垃圾的自动分拣是实现垃圾分类处理的核心环节，而垃圾图像的分类识别是自动分拣环节的关键，其识别的速度直接制约整个垃圾处理生产线的效率。

随着大规模标注数据的出现以及高性能图形处理器的飞速发展，深度卷积神经网络在诸如图像分类、目标检测以及语义分割等各个计算机视觉领域都取得了巨大的成功。研究人员同样将其应用于垃圾图像的自动识别领域，例如，刘国栋等(刘国栋,冯立辉,陈子健,李亿俍,卢继华.一种基于DMD和YOLOV5的光电智能垃圾分拣方法.申请号：202110758716.8)公开了一种基于DMD和YOLOV5的光电智能垃圾分拣方法，该方法将构建的垃圾图像数据集输入YOLOV5网络进行训练，得到训练好的YOLOV5网络作为垃圾图像识别模型；翟懿奎等(翟懿奎,余翠琳,柯琪锐,周文略,甘俊英,应自炉,曾军英.水底视觉垃圾清理机器人及其运作方法，申请号：202010176992.9.)公开了一种水底视觉垃圾清理机器人及其运作方法，该方法使用宽度学习网络中的自动编码器提取垃圾图像特征，利用代价敏感型分类方法得到权重，并进行分类。袁靖等(袁靖,周翔,张标,邓永财.垃圾分选系统及垃圾分选方法.申请号：201910287785.)公开的一种垃圾分选系统及垃圾分选方法，该方法中使用Faster R-CNN模型中的区域卷积特征模块提取目标垃圾图像的卷积特征并对其进行分类识别。

然而，上述方法往往依赖于大量的标注图像才能完成对模型的训练，并且模型中的参数量巨大，在对垃圾图像进行实时分类识别时会产生时间滞后，从而影响整个垃圾处理生成线的分拣效率。

发明内容

发明目的：为了解决现有技术中的不足，本发明提供了一种基于原型网络的小样本垃圾图像分类方法，该方法需要少量垃圾图像样本就能完成对垃圾图像的自动分类识别，能够提高垃圾处理过程中的自动分拣效率。

技术方案：本发明所述的基于原型网络的小样本垃圾图像分类方法，包括如下主要步骤：步骤1、图像的采集与预处理：使用摄像头采集若干垃圾图像构建垃圾分类数据集I，并将数据集I分为训练集I_train和测试集I_test；对于图像数据集I中的每一幅图像利用伽玛校正方法对进行预处理，增加图像的亮度，便于后续的分类识别；

步骤2、原型网络的构建：该网络主要是由四个模块组成，分别为主干网络，原型计算模块，相似性计算模块以及Softmax层；

步骤3、原型网络的输入与输出：采用片段式训练方式，在每个训练片段构建N-way-K-shot分类任务，每个分类任务利用训练集I_train建立支持样本数据集和查询样本数据集将和中的图像输入到原型网络中，依次经过网络的四个模块，得到关于查询样本的类别概率输出值；

步骤4、原型网络的训练：计算查询样本类别概率输出值与真实标签值之间的损失函数，利用梯度下降方法对网络中的参数进行优化；

步骤5、垃圾图像的分类测试：固定原型网络中的参数，利用测试数据集I_test建立支持样本数据集和测试样本数据集，并将其输入到原型网络中，依次经过网络的三个模块，就得到关于查询样本的类别输出值，从而完成对垃圾图像的分类测试。