Cargando…
MR-YOLO: An Improved YOLOv5 Network for Detecting Magnetic Ring Surface Defects
Magnetic rings are widely used in automotive, home appliances, and consumer electronics. Due to the materials used, processing techniques, and other factors, there will be top cracks, internal cracks, adhesion, and other defects on individual magnetic rings during the manufacturing process. To find...
Autores principales: | , , , , |
---|---|
Formato: | Online Artículo Texto |
Lenguaje: | English |
Publicado: |
MDPI
2022
|
Materias: | |
Acceso en línea: | https://www.ncbi.nlm.nih.gov/pmc/articles/PMC9781278/ https://www.ncbi.nlm.nih.gov/pubmed/36560265 http://dx.doi.org/10.3390/s22249897 |
_version_ | 1784857034685939712 |
---|---|
author | Lang, Xianli Ren, Zhijie Wan, Dahang Zhang, Yuzhong Shu, Shuangbao |
author_facet | Lang, Xianli Ren, Zhijie Wan, Dahang Zhang, Yuzhong Shu, Shuangbao |
author_sort | Lang, Xianli |
collection | PubMed |
description | Magnetic rings are widely used in automotive, home appliances, and consumer electronics. Due to the materials used, processing techniques, and other factors, there will be top cracks, internal cracks, adhesion, and other defects on individual magnetic rings during the manufacturing process. To find such defects, the most sophisticated YOLOv5 target identification algorithm is frequently utilized. However, it has problems such as high computation, sluggish detection, and a large model size. This work suggests an enhanced lightweight YOLOv5 (MR-YOLO) approach for the identification of magnetic ring surface defects to address these issues. To decrease the floating-point operation (FLOP) in the feature channel fusion process and enhance the performance of feature expression, the YOLOv5 neck network was added to the Mobilenetv3 module. To improve the robustness of the algorithm, a Mosaic data enhancement technique was applied. Moreover, in order to increase the network’s interest in minor defects, the SE attention module is inserted into the backbone network to replace the SPPF module with substantially more calculations. Finally, to further increase the new network’s accuracy and training speed, we substituted the original CIoU-Ioss for SIoU-Loss. According to the test, the FLOP and Params of the modified network model decreased by 59.4% and 47.9%, respectively; the reasoning speed increased by 16.6%, the model’s size decreased by 48.1%, and the mAP only lost by 0.3%. The effectiveness and superiority of this method are proved by an analysis and comparison of examples. |
format | Online Article Text |
id | pubmed-9781278 |
institution | National Center for Biotechnology Information |
language | English |
publishDate | 2022 |
publisher | MDPI |
record_format | MEDLINE/PubMed |
spelling | pubmed-97812782022-12-24 MR-YOLO: An Improved YOLOv5 Network for Detecting Magnetic Ring Surface Defects Lang, Xianli Ren, Zhijie Wan, Dahang Zhang, Yuzhong Shu, Shuangbao Sensors (Basel) Article Magnetic rings are widely used in automotive, home appliances, and consumer electronics. Due to the materials used, processing techniques, and other factors, there will be top cracks, internal cracks, adhesion, and other defects on individual magnetic rings during the manufacturing process. To find such defects, the most sophisticated YOLOv5 target identification algorithm is frequently utilized. However, it has problems such as high computation, sluggish detection, and a large model size. This work suggests an enhanced lightweight YOLOv5 (MR-YOLO) approach for the identification of magnetic ring surface defects to address these issues. To decrease the floating-point operation (FLOP) in the feature channel fusion process and enhance the performance of feature expression, the YOLOv5 neck network was added to the Mobilenetv3 module. To improve the robustness of the algorithm, a Mosaic data enhancement technique was applied. Moreover, in order to increase the network’s interest in minor defects, the SE attention module is inserted into the backbone network to replace the SPPF module with substantially more calculations. Finally, to further increase the new network’s accuracy and training speed, we substituted the original CIoU-Ioss for SIoU-Loss. According to the test, the FLOP and Params of the modified network model decreased by 59.4% and 47.9%, respectively; the reasoning speed increased by 16.6%, the model’s size decreased by 48.1%, and the mAP only lost by 0.3%. The effectiveness and superiority of this method are proved by an analysis and comparison of examples. MDPI 2022-12-15 /pmc/articles/PMC9781278/ /pubmed/36560265 http://dx.doi.org/10.3390/s22249897 Text en © 2022 by the authors. https://creativecommons.org/licenses/by/4.0/Licensee MDPI, Basel, Switzerland. This article is an open access article distributed under the terms and conditions of the Creative Commons Attribution (CC BY) license (https://creativecommons.org/licenses/by/4.0/). |
spellingShingle | Article Lang, Xianli Ren, Zhijie Wan, Dahang Zhang, Yuzhong Shu, Shuangbao MR-YOLO: An Improved YOLOv5 Network for Detecting Magnetic Ring Surface Defects |
title | MR-YOLO: An Improved YOLOv5 Network for Detecting Magnetic Ring Surface Defects |
title_full | MR-YOLO: An Improved YOLOv5 Network for Detecting Magnetic Ring Surface Defects |
title_fullStr | MR-YOLO: An Improved YOLOv5 Network for Detecting Magnetic Ring Surface Defects |
title_full_unstemmed | MR-YOLO: An Improved YOLOv5 Network for Detecting Magnetic Ring Surface Defects |
title_short | MR-YOLO: An Improved YOLOv5 Network for Detecting Magnetic Ring Surface Defects |
title_sort | mr-yolo: an improved yolov5 network for detecting magnetic ring surface defects |
topic | Article |
url | https://www.ncbi.nlm.nih.gov/pmc/articles/PMC9781278/ https://www.ncbi.nlm.nih.gov/pubmed/36560265 http://dx.doi.org/10.3390/s22249897 |
work_keys_str_mv | AT langxianli mryoloanimprovedyolov5networkfordetectingmagneticringsurfacedefects AT renzhijie mryoloanimprovedyolov5networkfordetectingmagneticringsurfacedefects AT wandahang mryoloanimprovedyolov5networkfordetectingmagneticringsurfacedefects AT zhangyuzhong mryoloanimprovedyolov5networkfordetectingmagneticringsurfacedefects AT shushuangbao mryoloanimprovedyolov5networkfordetectingmagneticringsurfacedefects |