Qunbiao Wu, Tao Liang, Haifeng Fang, Yangyang Wei, Mingqiang Wang, Defang He
{"title":"一种轻量级的可回收生活垃圾多目标检测深度学习算法","authors":"Qunbiao Wu, Tao Liang, Haifeng Fang, Yangyang Wei, Mingqiang Wang, Defang He","doi":"10.1089/ees.2023.0138","DOIUrl":null,"url":null,"abstract":"In light of the rapid development of human society, there has been a notable surge in waste production, which has resulted in environmental pollution and degradation. This is a pervasive issue that requires attention. To address the environmental problems caused by waste generation and advance the development of recyclable domestic waste detection, this article proposes waste classification as a solution. Traditional waste sorting methods have proven to be inefficient and prone to errors, hence the need for a more effective approach. A multiobjective recyclable domestic waste detection and classification method based on improved You Only Look Once v5s (YOLOv5s) is proposed in this study. In this study, the network structure is enhanced through the implementation of the Bidirectional Pyramid Network (BiFPN). The coordinate attention mechanism is then incorporated to elevate the accuracy of the model. Additionally, the loss function is refined by adopting the Efficient Intersection Over Union Loss (EIOU_Loss) metric to further optimize network performance. Finally, the introduction of the Ghost convolution module reduces parameter count and significantly improves the real-time detection speed. The waste dataset named Multi-classified Recyclable Domestic Trash Identification Dataset (MULTI-TRASH), which is composed of machine shooting, web crawler, and artificial photography, is used for verification due to its good generalization and representativeness. The mean Average Precision at a threshold of 0.5 ([email protected]) value of 94.8% is achieved by the improved model, which is a 30.72% reduction in the number of parameters and a 1.2% improvement in the [email protected] value compared with YOLOv5s. The effectiveness of the proposed algorithm is proved by a comparison with other target detection algorithms. This study aims to provide technical references for the development of a recyclable domestic waste detection system.","PeriodicalId":11777,"journal":{"name":"Environmental Engineering Science","volume":"19 1","pages":"0"},"PeriodicalIF":1.8000,"publicationDate":"2023-10-10","publicationTypes":"Journal Article","fieldsOfStudy":null,"isOpenAccess":false,"openAccessPdf":"","citationCount":"0","resultStr":"{\"title\":\"A Lightweight Deep Learning Algorithm for Multi-Objective Detection of Recyclable Domestic Waste\",\"authors\":\"Qunbiao Wu, Tao Liang, Haifeng Fang, Yangyang Wei, Mingqiang Wang, Defang He\",\"doi\":\"10.1089/ees.2023.0138\",\"DOIUrl\":null,\"url\":null,\"abstract\":\"In light of the rapid development of human society, there has been a notable surge in waste production, which has resulted in environmental pollution and degradation. This is a pervasive issue that requires attention. To address the environmental problems caused by waste generation and advance the development of recyclable domestic waste detection, this article proposes waste classification as a solution. Traditional waste sorting methods have proven to be inefficient and prone to errors, hence the need for a more effective approach. A multiobjective recyclable domestic waste detection and classification method based on improved You Only Look Once v5s (YOLOv5s) is proposed in this study. In this study, the network structure is enhanced through the implementation of the Bidirectional Pyramid Network (BiFPN). The coordinate attention mechanism is then incorporated to elevate the accuracy of the model. Additionally, the loss function is refined by adopting the Efficient Intersection Over Union Loss (EIOU_Loss) metric to further optimize network performance. Finally, the introduction of the Ghost convolution module reduces parameter count and significantly improves the real-time detection speed. The waste dataset named Multi-classified Recyclable Domestic Trash Identification Dataset (MULTI-TRASH), which is composed of machine shooting, web crawler, and artificial photography, is used for verification due to its good generalization and representativeness. The mean Average Precision at a threshold of 0.5 ([email protected]) value of 94.8% is achieved by the improved model, which is a 30.72% reduction in the number of parameters and a 1.2% improvement in the [email protected] value compared with YOLOv5s. The effectiveness of the proposed algorithm is proved by a comparison with other target detection algorithms. This study aims to provide technical references for the development of a recyclable domestic waste detection system.\",\"PeriodicalId\":11777,\"journal\":{\"name\":\"Environmental Engineering Science\",\"volume\":\"19 1\",\"pages\":\"0\"},\"PeriodicalIF\":1.8000,\"publicationDate\":\"2023-10-10\",\"publicationTypes\":\"Journal Article\",\"fieldsOfStudy\":null,\"isOpenAccess\":false,\"openAccessPdf\":\"\",\"citationCount\":\"0\",\"resultStr\":null,\"platform\":\"Semanticscholar\",\"paperid\":null,\"PeriodicalName\":\"Environmental Engineering Science\",\"FirstCategoryId\":\"1085\",\"ListUrlMain\":\"https://doi.org/10.1089/ees.2023.0138\",\"RegionNum\":4,\"RegionCategory\":\"环境科学与生态学\",\"ArticlePicture\":[],\"TitleCN\":null,\"AbstractTextCN\":null,\"PMCID\":null,\"EPubDate\":\"\",\"PubModel\":\"\",\"JCR\":\"Q4\",\"JCRName\":\"ENGINEERING, ENVIRONMENTAL\",\"Score\":null,\"Total\":0}","platform":"Semanticscholar","paperid":null,"PeriodicalName":"Environmental Engineering Science","FirstCategoryId":"1085","ListUrlMain":"https://doi.org/10.1089/ees.2023.0138","RegionNum":4,"RegionCategory":"环境科学与生态学","ArticlePicture":[],"TitleCN":null,"AbstractTextCN":null,"PMCID":null,"EPubDate":"","PubModel":"","JCR":"Q4","JCRName":"ENGINEERING, ENVIRONMENTAL","Score":null,"Total":0}
引用次数: 0
摘要
随着人类社会的快速发展,垃圾的产生急剧增加,造成了环境的污染和退化。这是一个普遍存在的问题,需要引起注意。为了解决垃圾产生带来的环境问题,促进可回收生活垃圾检测的发展,本文提出了垃圾分类作为解决方案。传统的废物分类方法效率低下,容易出错,因此需要一种更有效的方法。本研究提出了一种基于改进You Only Look Once v5s (YOLOv5s)的多目标可回收生活垃圾检测分类方法。在本研究中,通过实施双向金字塔网络(Bidirectional Pyramid network, BiFPN)来增强网络结构。为了提高模型的准确性,引入了坐标注意机制。此外,通过采用EIOU_Loss (Efficient Intersection Over Union loss)度量对损失函数进行细化,进一步优化网络性能。最后,Ghost卷积模块的引入减少了参数数量,显著提高了实时检测速度。由于具有较好的泛化和代表性,我们使用了多分类可回收生活垃圾识别数据集(MULTI-TRASH)进行验证,该数据集由机器拍摄、网络爬虫和人工摄影组成。改进的模型在阈值为0.5 ([email protected])时的平均平均精度为94.8%,与YOLOv5s相比,参数数量减少了30.72%,[email protected]值提高了1.2%。通过与其他目标检测算法的比较,证明了该算法的有效性。本研究旨在为可回收生活垃圾检测系统的开发提供技术参考。
A Lightweight Deep Learning Algorithm for Multi-Objective Detection of Recyclable Domestic Waste
In light of the rapid development of human society, there has been a notable surge in waste production, which has resulted in environmental pollution and degradation. This is a pervasive issue that requires attention. To address the environmental problems caused by waste generation and advance the development of recyclable domestic waste detection, this article proposes waste classification as a solution. Traditional waste sorting methods have proven to be inefficient and prone to errors, hence the need for a more effective approach. A multiobjective recyclable domestic waste detection and classification method based on improved You Only Look Once v5s (YOLOv5s) is proposed in this study. In this study, the network structure is enhanced through the implementation of the Bidirectional Pyramid Network (BiFPN). The coordinate attention mechanism is then incorporated to elevate the accuracy of the model. Additionally, the loss function is refined by adopting the Efficient Intersection Over Union Loss (EIOU_Loss) metric to further optimize network performance. Finally, the introduction of the Ghost convolution module reduces parameter count and significantly improves the real-time detection speed. The waste dataset named Multi-classified Recyclable Domestic Trash Identification Dataset (MULTI-TRASH), which is composed of machine shooting, web crawler, and artificial photography, is used for verification due to its good generalization and representativeness. The mean Average Precision at a threshold of 0.5 ([email protected]) value of 94.8% is achieved by the improved model, which is a 30.72% reduction in the number of parameters and a 1.2% improvement in the [email protected] value compared with YOLOv5s. The effectiveness of the proposed algorithm is proved by a comparison with other target detection algorithms. This study aims to provide technical references for the development of a recyclable domestic waste detection system.
期刊介绍:
Environmental Engineering Science explores innovative solutions to problems in air, water, and land contamination and waste disposal, with coverage of climate change, environmental risk assessment and management, green technologies, sustainability, and environmental policy. Published monthly online, the Journal features applications of environmental engineering and scientific discoveries, policy issues, environmental economics, and sustainable development.