A Lightweight Method for Detecting Sewer Defects Based on Improved YOLOv5

被引:2
|
作者
Zhang, Xing [1 ]
Zhang, Jiawei [2 ]
Tian, Lei [2 ]
Liu, Xiang [2 ]
Wang, Shuohong [3 ,4 ]
机构
[1] Shanghai Univ Engn Sci, Sch Management, Shanghai 201620, Peoples R China
[2] Shanghai Univ Engn Sci, Sch Elect & Elect Engn, Shanghai 201620, Peoples R China
[3] Harvard Univ, Dept Mol & Cellular Biol, Cambridge, MA 02138 USA
[4] Harvard Univ, Ctr Brain Sci, Cambridge, MA 02138 USA
来源
APPLIED SCIENCES-BASEL | 2023年 / 13卷 / 15期
关键词
sewer defect detection; YOLOv5; GhostNet; lightweight; coordinate attention mechanism;
D O I
10.3390/app13158986
中图分类号
O6 [化学];
学科分类号
0703 ;
摘要
In response to the issues of the existing sewer defect detection models, which are not applicable to small computing platforms due to their complex structure and large computational volume, as well as the low detection accuracy, a lightweight detection model based on YOLOv5, named YOLOv5-GBC, is proposed. Firstly, to address the computational redundancy problem of the traditional convolutional approach, GhostNet, which is composed of Ghost modules, is used to replace the original backbone network. Secondly, aiming at the problem of low detection accuracy of small defects, more detailed spatial information is introduced by fusing shallow features in the neck network, and weighted feature fusion is used to improve the feature fusion efficiency. Finally, to improve the sensitivity of the model to key feature information, the coordinate attention mechanism is introduced into the Ghost module and replaced the traditional convolution approach in the neck network. Experimental results show that compared with the YOLOv5 model, the model size and floating point of operations (FLOPs) of YOLOv5-GBC are reduced by 74.01% and 74.78%, respectively; the mean average precision (MAP) and recall are improved by 0.88% and 1.51%, respectively; the detection speed is increased by 63.64%; and the model size and computational volume are significantly reduced under the premise of ensuring the detection accuracy, which can effectively meet the needs of sewer defect detection on small computing platforms.
引用
收藏
页数:16
相关论文
共 50 条
  • [1] A Sewer Pipeline Defect Detection Method Based on Improved YOLOv5
    Wang, Tong
    Li, Yuhang
    Zhai, Yidi
    Wang, Weihua
    Huang, Rongjie
    [J]. PROCESSES, 2023, 11 (08)
  • [2] A method for detecting uneaten feed based on improved YOLOv5
    Xu, Chen
    Wang, Zhiyong
    Du, Rongxiang
    Li, Yachao
    Li, Daoliang
    Chen, Yingyi
    Li, Wensheng
    Liu, Chunhong
    [J]. COMPUTERS AND ELECTRONICS IN AGRICULTURE, 2023, 212
  • [3] An Improved Lightweight YOLOv5 Algorithm for Detecting Strawberry Diseases
    Chen, Shunlong
    Liao, Yinghua
    Lin, Feng
    Huang, Bo
    [J]. IEEE ACCESS, 2023, 11 : 54080 - 54092
  • [4] An Intelligent Method for Detecting Surface Defects in Aluminium Profiles Based on the Improved YOLOv5 Algorithm
    Wang, Teng
    Su, Jianhuan
    Xu, Chuan
    Zhang, Yinguang
    [J]. ELECTRONICS, 2022, 11 (15)
  • [5] Lightweight Meter Pointer Recognition Method Based on Improved YOLOv5
    Zhang, Chi
    Wang, Kai
    Zhang, Jie
    Zhou, Fan
    Zou, Le
    [J]. SENSORS, 2024, 24 (05)
  • [6] Citrus Detection Method Based on Improved YOLOv5 Lightweight Network
    Gao, Xinyang
    Wei, Sheng
    Wen, Zhiqing
    Yu, Tianbiao
    [J]. Computer Engineering and Applications, 2023, 59 (11) : 212 - 221
  • [7] Detection of Surface Defects in Lightweight Insulators Using Improved YOLOv5
    Guo Yu
    Ma Meiling
    Li Dalin
    [J]. LASER & OPTOELECTRONICS PROGRESS, 2023, 60 (24)
  • [8] A method of citrus epidermis defects detection based on an improved YOLOv5
    Hu, WenXin
    Xiong, JunTao
    Liang, JunHao
    Xie, ZhiMing
    Liu, ZhiYu
    Huang, QiYin
    Yang, ZhenGang
    [J]. BIOSYSTEMS ENGINEERING, 2023, 227 : 19 - 35
  • [9] A lightweight method for apple-on-tree detection based on improved YOLOv5
    Li, Mei
    Zhang, Jiachuang
    Liu, Hubin
    Yuan, Yuhui
    Li, Junhui
    Zhao, Longlian
    [J]. SIGNAL IMAGE AND VIDEO PROCESSING, 2024, 18 (10) : 6713 - 6727
  • [10] Driver Abnormal Expression Detection Method Based on Improved Lightweight YOLOv5
    Yao, Keming
    Wang, Zhongzhou
    Guo, Fuao
    Li, Feng
    [J]. ELECTRONICS, 2024, 13 (06)