DMS-YOLOv5: A Decoupled Multi-Scale YOLOv5 Method for Small Object Detection

被引:4
|
作者
Gao, Tianyu [1 ]
Wushouer, Mairidan [1 ]
Tuerhong, Gulanbaier [1 ]
机构
[1] Xinjiang Univ, Coll Informat Sci & Engn, Xinjiang Multilingual Informat Technol Res Ctr, Xinjiang Multilingual Informat Technol Lab, Urumqi 830017, Peoples R China
来源
APPLIED SCIENCES-BASEL | 2023年 / 13卷 / 10期
关键词
small object detection; YOLOv5; convolutional neural network; decoupled network; attention mechanisms;
D O I
10.3390/app13106124
中图分类号
O6 [化学];
学科分类号
0703 ;
摘要
Small objects detection is a challenging task in computer vision due to the limited semantic information that can be extracted and the susceptibility to background interference. In this paper, we propose a decoupled multi-scale small object detection algorithm named DMS-YOLOv5. The algorithm incorporates a receptive field module into the feature extraction network for better focus on low-resolution small objects. The coordinate attention mechanism, which combines spatial and channel attention information, is introduced to reduce interference from background information and enhance the network's attention to object information. A detection layer tailored to small-sized objects is added to compensate for the loss of small object information in multiple downsampling operations, greatly improving the detection capability of small objects. Next, The decoupled network is introduced into the detection head network for branch processing of classification and bounding box regression tasks. Finally, the bounding box loss function is improved to alleviate missed detection problems caused by the concentration of small objects and mutual occlusion between objects. The improved method achieved a mean average precision improvement of 12.1% on VisDrone2019-DET dataset compared to the original method. In comparison experiments with similar methods, our proposed method also demonstrated good performance, validating its effectiveness.
引用
收藏
页数:19
相关论文
共 50 条
  • [1] A Decoupled YOLOv5 with Deformable Convolution and Multi-scale Attention
    Yuan, Gui
    Liu, Gang
    Chen, Jian
    [J]. KNOWLEDGE SCIENCE, ENGINEERING AND MANAGEMENT, PT I, 2022, 13368 : 3 - 14
  • [2] A Decoupled YOLOv5 with Deformable Convolution and Multi-scale Attention
    Yuan, Gui
    Liu, Gang
    Chen, Jian
    [J]. Lecture Notes in Computer Science (including subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics), 2022, 13368 LNAI : 3 - 14
  • [3] Enhanced Multi-scale Target Detection Method Based on YOLOv5
    Hui K.
    Yang W.
    Liu H.
    Zhang Z.
    Zheng J.
    Bai X.
    [J]. Binggong Xuebao/Acta Armamentarii, 2023, 44 (09): : 2600 - 2610
  • [4] Small Object Detection Method based on Improved YOLOv5
    Gao, Tianyu
    Wushouer, Mairidan
    Tuerhong, Gulanbaier
    [J]. 2022 INTERNATIONAL CONFERENCE ON VIRTUAL REALITY, HUMAN-COMPUTER INTERACTION AND ARTIFICIAL INTELLIGENCE, VRHCIAI, 2022, : 144 - 149
  • [5] A Kind of Water Surface Multi-Scale Object Detection Method Based on Improved YOLOv5 Network
    Ma, Zhongli
    Wan, Yi
    Liu, Jiajia
    An, Ruojin
    Wu, Lili
    [J]. MATHEMATICS, 2023, 11 (13)
  • [6] MCF-YOLOv5: A Small Target Detection Algorithm Based on Multi-Scale Feature Fusion Improved YOLOv5
    Gao, Song
    Gao, Mingwang
    Wei, Zhihui
    [J]. INFORMATION, 2024, 15 (05)
  • [7] Improved YOLOv5 for Small Object Detection Algorithm
    Yu, Jun
    Jia, Yinshan
    [J]. Computer Engineering and Applications, 2023, 59 (12) : 201 - 207
  • [8] A Multi-Scale Traffic Object Detection Algorithm for Road Scenes Based on Improved YOLOv5
    Li, Ang
    Sun, Shijie
    Zhang, Zhaoyang
    Feng, Mingtao
    Wu, Chengzhong
    Li, Wang
    [J]. ELECTRONICS, 2023, 12 (04)
  • [9] A multi-scale cucumber disease detection method in natural scenes based on YOLOv5
    Li, Shufei
    Li, Kaiyu
    Qiao, Yan
    Zhang, Lingxian
    [J]. COMPUTERS AND ELECTRONICS IN AGRICULTURE, 2022, 202
  • [10] MTD-YOLOv5: Enhancing marine target detection with multi-scale feature fusion in YOLOv5 model
    Lian-suo, W. E. I.
    Shen-hao, Huang
    Long-yu, Ma
    [J]. HELIYON, 2024, 10 (04)