DBS-YOLO: A High-Precision Object Detection Algorithm for Hazardous Waste Images

被引:0
|
作者
Xiao, Zhenqi [1 ,2 ]
Yang, Guangxiang [2 ]
Wang, Xu [2 ]
Yu, Guangling [2 ]
Wang, Xiaoheng [2 ]
Yang, Baofeng [2 ]
Zhang, Delin [2 ]
机构
[1] Guangzhou Xinhua Univ, Sch Informat & Intelligent Engn, Dongguan 523133, Peoples R China
[2] Chongqing Technol & Business Univ, Sch Comp Sci & Informat Engn, Chongqing 400067, Peoples R China
关键词
Feature extraction; Accuracy; YOLO; Real-time systems; Convolutional neural networks; Computational modeling; Sorting; Convolution; Classification algorithms; Training; Bi-level routing attention (BRA); deformable convolutional networks version 3 (DCNv3); hazardous waste detection; soft non-maximum suppression (Soft-NMS); YOLOv8; GARBAGE; MODEL;
D O I
10.1109/TIM.2024.3476526
中图分类号
TM [电工技术]; TN [电子技术、通信技术];
学科分类号
0808 ; 0809 ;
摘要
The adverse effects of hazardous waste on the natural environment and human health are evident. In real-life scenarios, the dense distribution, mutual occlusion, and complex image backgrounds of hazardous waste present significant challenges for existing object detection algorithms to achieve a high detection accuracy. In order to enhance the efficiency of hazardous waste sorting, this article proposes an improved detection model, DBS-YOLO, based on the YOLOv8n network. This model strikes a balance between accuracy and lightweight design. First, we replace part of the convolution modules in the C2f module with deformable convolutional networks version 3 (DCNv3) modules, proposing the DC2f module. Incorporating this module into YOLOv8n not only achieves a lightweight network design but also enhances the model's adaptability to occluded hazardous waste. In addition, we introduce a bi-level routing attention (BRA) module to construct a global receptive field, capturing long-range dependencies and improving the model's focus on targets. Finally, we employ a soft non-maximum suppression (Soft-NMS) algorithm to reduce instances of false positives (FPs) and false negatives (FNs) caused by mutual occlusion, further enhancing model accuracy. Experimental results demonstrate that compared to the original YOLOv8n network, DBS-YOLO achieves improvements of 2.0% in precision, 1.0% in recall, 2.1% in mean average precision (mAP)@0.5, and 4.9% in mAP@0.5:0.95, reaching a mAP@0.5 of 95.4%. The model also mitigates instances of FPs and FNs. Comparative analyses with SDD, Faster R-CNN, and YOLO series networks confirm the effectiveness of the proposed model.
引用
收藏
页数:15
相关论文
共 50 条
  • [1] Lightweight and high-precision object detection algorithm based on YOLO framework
    Fan Xin-chuan
    Chen Chun-mei
    CHINESE JOURNAL OF LIQUID CRYSTALS AND DISPLAYS, 2023, 38 (07) : 945 - 954
  • [2] An MRS-YOLO Model for High-Precision Waste Detection and Classification
    Ren, Yuanming
    Li, Yizhe
    Gao, Xinya
    SENSORS, 2024, 24 (13)
  • [3] HP-YOLOv8: High-Precision Small Object Detection Algorithm for Remote Sensing Images
    Yao, Guangzhen
    Zhu, Sandong
    Zhang, Long
    Qi, Miao
    SENSORS, 2024, 24 (15)
  • [4] MS-YOLO: A Lightweight and High-Precision YOLO Model for Drowning Detection
    Song, Qi
    Yao, Bodan
    Xue, Yunlong
    Ji, Shude
    SENSORS, 2024, 24 (21)
  • [5] EBSE-YOLO: High Precision Recognition Algorithm for Small Target Foreign Object Detection
    Wang, Shuohe
    Wang, Yongda
    Chang, Yujian
    Zhao, Ruikang
    She, Yansong
    IEEE ACCESS, 2023, 11 : 57951 - 57964
  • [6] Task interleaving and orientation estimation for high-precision oriented object detection in aerial images
    Ming, Qi
    Miao, Lingjuan
    Zhou, Zhiqiang
    Song, Junjie
    Dong, Yunpeng
    Yang, Xue
    ISPRS JOURNAL OF PHOTOGRAMMETRY AND REMOTE SENSING, 2023, 196 : 241 - 255
  • [7] High-precision automatic detection method of illegal construction object images in complex scenes
    Liu, Wenjin
    Zhang, Shudong
    Zhou, Lijuan
    Xu, Min
    Ren, Zhongshan
    JOURNAL OF ELECTRONIC IMAGING, 2023, 32 (03)
  • [8] Trinity-Yolo: High-precision logo detection in the real world
    Mao, KeJi
    Jin, RunHui
    Chen, KaiYan
    Mao, JiaFa
    Dai, GuangLin
    IET IMAGE PROCESSING, 2023, 17 (07) : 2272 - 2283
  • [9] Object Detection in Shelf Images with YOLO
    Melek, Ceren Gulra
    Sonmez, Elena Battini
    Albayrak, Songul
    PROCEEDINGS OF 18TH INTERNATIONAL CONFERENCE ON SMART TECHNOLOGIES (IEEE EUROCON 2019), 2019,
  • [10] Fastand High-Precision Rectangle Detection Algorithm
    Lu Lu
    Yang Jiong
    Liang Jie
    Jiang Yulin
    LASER & OPTOELECTRONICS PROGRESS, 2020, 57 (18)