Research on Water Garbage Detection Algorithm Based on GFL Network

Nengmin Yi; Weilin Luo

doi:10.54097/fcis.v3i1.6353

Authors

Nengmin Yi
Weilin Luo

DOI:

https://doi.org/10.54097/fcis.v3i1.6353

Keywords:

Garbage detection, HAM attention mechanism, GFL (Generalized Focal Loss)

Abstract

Aiming at the problem that existing water surface garbage detection algorithms cannot meet the tradeoff between real-time and accuracy in natural scenes, we propose a GFL (Generalized Focal Loss) based GFL_ HAM algorithm to improve the accuracy of water surface garbage detection, and better apply to water surface garbage detection application scenarios. We have added the HAM (Hybrid Attention Model) module to the feature extraction module (resnet50) of the GFL network to improve the feature extraction capability of the backbone network and the network representation capability. And use the self-constructed garbage detection datasets under natural scenes for training and testing. After analyzing the training process and test results, our network has higher reliability compared to the current mainstream single stage network. Our network can achieve 60.12% mAP on the water surface garbage detection datasets, which is 1.09%, 1.36%, and 1.61% higher than YOLOv3, SSD, and GFL, respectively.

Downloads

Download data is not yet available.

References

Duan, Ling. "Global Environment Outlook 6" was released after 5 years: The Earth has been severely damaged [J]. The world environment,2020(2):3.

Regulations of Beijing Municipality on the Administration of Domestic Waste [J]. Communiqu é of the Standing Committee of the Beijing Municipal People's Congress, 2020, 000(005):P.83-95.

Meinecke G, R atmeyer V, Renken J, HYBRID-ROV-Development of a new underwater vehicle for high-risk areas [C]/ /OCEANS11-MTS /IEEE Kona, 2011: 1-6.

Hou Z Q, Han C Z, A Background reconstruction algorithm based on pixel intensity classification [J], Journal of Software, 2005, 16( 9) : 1568-1576,

Huang, Tao, Research on Water Object Extraction Algorithms in Natural Scenes[J]. Communication Technology,2013(3) : 80-82.

YANG M, THUNG G. Classification of trash for recyclability status[R].Stanford CS229 Project Report,2016:832-841.

Wang, Xie, Yang, et al. A garbage classification and detection method based on deep learning [J] .Modern electronic technology, 2021, 44(21):4.

Xu, Xiong. An improved lightweight garbage target detection algorithm [J]. Computer Technology and Development, 2022(002):032.

Li, X., Wang, W., Wu, L., Chen, S., Hu, X., Li, J., Tang, J., & Yang, J. (2020). Generalized Focal Loss: Learning Qualified and Distributed Bounding Boxes for Dense Object Detection. ArXiv, abs/2006.04388.

He K , Zhang X , Ren S , et al. Deep Residual Learning for Image Recognition[J]. IEEE, 2016.

LIN T Y, DOLLARP, GIR SHICKR, et al, Feature pyramid networks for object detection[C]/ /2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR) , Piscataway: IEEE, 2017: 2117-2125,

Neubeck A , Gool LJV . Efficient Non-Maximum Suppression[C] // 18th International Conference on Pattern Recognition (ICPR 2006), 20-24 August 2006, Hong Kong, China. IEEE Computer Society, 2006.

Lin T Y , Goyal P , Girshick R , et al. Focal Loss for Dense Object Detection[J]. IEEE Transactions on Pattern Analysis & Machine Intelligence, 2017, PP(99):2999-3007.

Rezatofighi, Hamid, et al. "Generalized intersection over union: A metric and a loss for bounding box regression." Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition. 2019.

Woo, Sanghyun et al.“CBAM: Convolutional Block Attention Module.”European Conference on Computer Vision (2018).

NTY,DOLLARP, GIRSHICKR, et al, Feature pyramid networks for object detection [C]/ /2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), Piscataway : IEEE , 2017: 2117-2125.

Redmon J , Farhadi A . YOLOv3: An Incremental Improvement [J]. arXiv e-prints, 2018.

Berg AC , Fu CY , Szegedy C , et al. SSD: Single Shot Multi Box Detector:, 10.1007/978-3-319-46448-0_2[P]. 2015.