本文已被:浏览 1857次 下载 3178次
Received:February 08, 2020 Revised:March 03, 2020
Received:February 08, 2020 Revised:March 03, 2020
中文摘要: 垃圾分类作为资源回收利用的重要环节之一, 可以有效地提高资源回收利用效率, 进一步减轻环境污染带来的危害. 随着现代工业逐步智能化, 传统的图像分类算法已经不能满足垃圾分拣设备的要求. 本文提出一种基于卷积神经网络的垃圾图像分类模型(Garbage Classification Network, GCNet). 通过构建注意力机制, 模型完成局部和全局的特征提取, 能够获取到更加完善、有效的特征信息; 同时, 通过特征融合机制, 将不同层级、尺寸的特征进行融合, 更加有效地利用特征, 避免梯度消失现象. 实验结果证明, GCNet在相关垃圾分类数据集上取得了优异的结果, 能够有效地提高垃圾识别精度.
Abstract:Garbage classification, as one of the important links of resource recycling, can effectively improve the efficiency of resource recycling and further reduce the harm caused by environmental pollution. With the development of modern industry, traditional image classification algorithm cannot meet the requirements of garbage sorting equipment. This study proposes a garbage classification model based on convolutional neural networks (Garbage Classification Network, GCNet). By constructing the attention mechanism, the model completes extracting the local and global features and can obtain perfect and effective feature information. At the same time, the feature fusion mechanism is used to fuse features at different levels and sizes, which can effectively use features and prevent gradient from vanishing. The experimental results prove that GCNet has achieved excellent results on garbage classification datasets and can effectively improve the accuracy of garbage classification.
keywords: garbage classification convolutional neural network image classification attention mechanism feature fusion
文章编号: 中图分类号: 文献标志码:
基金项目:
引用文本:
董子源,韩卫光.基于卷积神经网络的垃圾图像分类算法.计算机系统应用,2020,29(8):199-204
DONG Zi-Yuan,HAN Wei-Guang.Classification Algorithm of Garbage Images Based on Convolutional Neural Network.COMPUTER SYSTEMS APPLICATIONS,2020,29(8):199-204
董子源,韩卫光.基于卷积神经网络的垃圾图像分类算法.计算机系统应用,2020,29(8):199-204
DONG Zi-Yuan,HAN Wei-Guang.Classification Algorithm of Garbage Images Based on Convolutional Neural Network.COMPUTER SYSTEMS APPLICATIONS,2020,29(8):199-204