| 轻量化网络LW-GCNet在垃圾分类中的应用 |
| |
| 引用本文： | 夏景明,徐子峰,谈玲. 轻量化网络LW-GCNet在垃圾分类中的应用[J]. 环境工程, 2023, 41(2): 173-180. DOI: 10.13205/j.hjgc.202302023 |
| |
| 作者姓名： | 夏景明 徐子峰 谈玲 |
| |
| 作者单位： | 南京信息工程大学人工智能学院,南京210044;南京信息工程大学数字取证教育部工程研究中心,南京210044 |
| |
| 基金项目： | 国家自然科学基金项目(61871032)国家重点研发计划(2021YFB2700910)江苏省高等学校基金项目(20KJB510036) |
| |
| 摘 要： | 垃圾分类是构建绿色城市的重要途径。传统的垃圾分类是由人工进行,分类不彻底,工作强度大,不利于环境保护与资源再利用。为提高垃圾分类的准确性,提出了一种基于VGG16网络的轻量化网络模型LW-GCNet (light weight garbage classify network)。该网络模型通过引入深度可分离卷积和SE(squeeze-and-excitation)模块来进行特征提取,并将垃圾图像的浅层和深层特征有机融合,在减少计算量的同时,增强了待分类垃圾图像通道之间的依赖关系,为分类提供多层次的语义信息。此外,LW-GCNet模型采用自适应最大池化和全局平均池化取代VGG16网络中的全连接层,有效降低了参数量。利用由4类垃圾图像构成的数据集GRAB125对LW-GCNet性能进行验证。实验结果表明：该方法在保证识别速度的前提下,识别平均准确率达到77.17%,参数量为3.15M,易于在户外的嵌入式系统中进行部署。
|
| 关 键 词： | 垃圾分类 轻量化网络 SE模块 特征融合 |
| 收稿时间： | 2022-05-02 |
APPLICATION RESEARCH OF LIGHTWEIGHT NETWORK LW-GCNet IN GARBAGE CLASSIFICATION |
| |
| Affiliation: | 1. School of Artificial Intelligence, Nanjing University of Information Science & Technology, Nanjing 210044, China;2. Engineering Research Center of Digital Forensics Ministry of Education, Nanjing University of Information Science & Technology, Nanjing 210044, China |
| |
| Abstract: | Garbage classification is an important way to build a green city. The traditional garbage classification is commonly carried out manually, the classification is not thorough, and the labor intensity classification is high, which is not conducive to environmental protection and resource reuse. In order to improve the accuracy of garbage classification, this paper proposed a lightweight network model LW-GCNet (light weight garbage classify network) based on VGG16. The network model performed feature extraction by introducing depthwise separable convolution and SE (squeeze-and-excitation) modules, and organically fused the shallow and deep features of junk images. These modules enhanced the dependencies between channels of garbage images to be classified while reducing the computational complexity of the model and providing multi-level semantic information for accurate classification. In addition, the LW-GCNet model adopted adaptive max pooling and global average pooling to replace the fully connected layer in the VGG16 network, which effectively reduced the number of parameters. The performance of LW-GCNet was validated using the dataset GRAB125 consisting of four types of garbage images. The experimental results showed that, on the premise of ensuring the recognition speed, the average recognition accuracy rate of this method reached 77.17%, and the parameter quantity was 3.15 M, making it easy to be deployed in outdoor embedded systems. |
| |
| Keywords: | |
| 本文献已被 万方数据 等数据库收录！ |
| 点击此处可从《环境工程》浏览原始摘要信息 |
|
点击此处可从《环境工程》下载全文 |
|
