Improved YOLOv7-based steel surface defect detection algorithm
In response to the limited detection ability and low model generalization ability of the YOLOv7 algorithm for small targets, this paper proposes a detection algorithm based on the improved YOLOv7 algorithm for steel surface defect detection. First, the Transformer-InceptionDWConvolution (TI) module...
Main Authors: | , , , |
---|---|
Format: | Article |
Language: | English |
Published: |
AIMS Press
2024-01-01
|
Series: | Mathematical Biosciences and Engineering |
Subjects: | |
Online Access: | https://www.aimspress.com/article/doi/10.3934/mbe.2024016?viewType=HTML |
_version_ | 1797349562202980352 |
---|---|
author | Yinghong Xie Biao Yin Xiaowei Han Yan Hao |
author_facet | Yinghong Xie Biao Yin Xiaowei Han Yan Hao |
author_sort | Yinghong Xie |
collection | DOAJ |
description | In response to the limited detection ability and low model generalization ability of the YOLOv7 algorithm for small targets, this paper proposes a detection algorithm based on the improved YOLOv7 algorithm for steel surface defect detection. First, the Transformer-InceptionDWConvolution (TI) module is designed, which combines the Transformer module and InceptionDWConvolution to increase the network's ability to detect small objects. Second, the spatial pyramid pooling fast cross-stage partial channel (SPPFCSPC) structure is introduced to enhance the network training performance. Third, a global attention mechanism (GAM) attention mechanism is designed to optimize the network structure, weaken the irrelevant information in the defect image, and increase the algorithm's ability to detect small defects. Meanwhile, the Mish function is used as the activation function of the feature extraction network to improve the model's generalization ability and feature extraction ability. Finally, a minimum partial distance intersection over union (MPDIoU) loss function is designed to locate the loss and solve the mismatch problem between the complete intersection over union (CIoU) prediction box and the real box directions. The experimental results show that on the Northeastern University Defect Detection (NEU-DET) dataset, the improved YOLOv7 network model improves the mean Average precision (mAP) performance by 6% when compared to the original algorithm, while on the VOC2012 dataset, the mAP performance improves by 2.6%. These results indicate that the proposed algorithm can effectively improve the small defect detection performance on steel surface defects. |
first_indexed | 2024-03-08T12:32:05Z |
format | Article |
id | doaj.art-2ad381724fd547efa6b2ed670ebf4f5f |
institution | Directory Open Access Journal |
issn | 1551-0018 |
language | English |
last_indexed | 2024-03-08T12:32:05Z |
publishDate | 2024-01-01 |
publisher | AIMS Press |
record_format | Article |
series | Mathematical Biosciences and Engineering |
spelling | doaj.art-2ad381724fd547efa6b2ed670ebf4f5f2024-01-22T01:31:47ZengAIMS PressMathematical Biosciences and Engineering1551-00182024-01-0121134636810.3934/mbe.2024016Improved YOLOv7-based steel surface defect detection algorithmYinghong Xie0Biao Yin1Xiaowei Han2 Yan Hao31. School of Information Engineering, Shenyang University, Shenyang 110003, China1. School of Information Engineering, Shenyang University, Shenyang 110003, China2. Institute for Science, Technology and Innovation, Shenyang University, Shenyang 110003, China1. School of Information Engineering, Shenyang University, Shenyang 110003, ChinaIn response to the limited detection ability and low model generalization ability of the YOLOv7 algorithm for small targets, this paper proposes a detection algorithm based on the improved YOLOv7 algorithm for steel surface defect detection. First, the Transformer-InceptionDWConvolution (TI) module is designed, which combines the Transformer module and InceptionDWConvolution to increase the network's ability to detect small objects. Second, the spatial pyramid pooling fast cross-stage partial channel (SPPFCSPC) structure is introduced to enhance the network training performance. Third, a global attention mechanism (GAM) attention mechanism is designed to optimize the network structure, weaken the irrelevant information in the defect image, and increase the algorithm's ability to detect small defects. Meanwhile, the Mish function is used as the activation function of the feature extraction network to improve the model's generalization ability and feature extraction ability. Finally, a minimum partial distance intersection over union (MPDIoU) loss function is designed to locate the loss and solve the mismatch problem between the complete intersection over union (CIoU) prediction box and the real box directions. The experimental results show that on the Northeastern University Defect Detection (NEU-DET) dataset, the improved YOLOv7 network model improves the mean Average precision (mAP) performance by 6% when compared to the original algorithm, while on the VOC2012 dataset, the mAP performance improves by 2.6%. These results indicate that the proposed algorithm can effectively improve the small defect detection performance on steel surface defects.https://www.aimspress.com/article/doi/10.3934/mbe.2024016?viewType=HTMLyolov7transformerattention mechanismsppfcspcdefect detection |
spellingShingle | Yinghong Xie Biao Yin Xiaowei Han Yan Hao Improved YOLOv7-based steel surface defect detection algorithm Mathematical Biosciences and Engineering yolov7 transformer attention mechanism sppfcspc defect detection |
title | Improved YOLOv7-based steel surface defect detection algorithm |
title_full | Improved YOLOv7-based steel surface defect detection algorithm |
title_fullStr | Improved YOLOv7-based steel surface defect detection algorithm |
title_full_unstemmed | Improved YOLOv7-based steel surface defect detection algorithm |
title_short | Improved YOLOv7-based steel surface defect detection algorithm |
title_sort | improved yolov7 based steel surface defect detection algorithm |
topic | yolov7 transformer attention mechanism sppfcspc defect detection |
url | https://www.aimspress.com/article/doi/10.3934/mbe.2024016?viewType=HTML |
work_keys_str_mv | AT yinghongxie improvedyolov7basedsteelsurfacedefectdetectionalgorithm AT biaoyin improvedyolov7basedsteelsurfacedefectdetectionalgorithm AT xiaoweihan improvedyolov7basedsteelsurfacedefectdetectionalgorithm AT yanhao improvedyolov7basedsteelsurfacedefectdetectionalgorithm |