Loader Bucket Working Angle Identification Method Based on YOLOv5s and EMA Attention Mechanism
In response to the issues of low recognition efficiency and large errors encountered in the process of identifying the working angle of the bucket during current automated loader construction operations, a method based on YOLOv5s and the EMA attention mechanism for loader bucket working angle identi...
Saved in:
Main Authors: | , , , |
---|---|
Format: | Article |
Language: | English |
Published: |
IEEE
2024-01-01
|
Series: | IEEE Access |
Subjects: | |
Online Access: | https://ieeexplore.ieee.org/document/10613619/ |
Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
_version_ | 1841533418545872896 |
---|---|
author | Xuedong Zhang Bo Cui Zhaoxu Wang Wangting Zeng |
author_facet | Xuedong Zhang Bo Cui Zhaoxu Wang Wangting Zeng |
author_sort | Xuedong Zhang |
collection | DOAJ |
description | In response to the issues of low recognition efficiency and large errors encountered in the process of identifying the working angle of the bucket during current automated loader construction operations, a method based on YOLOv5s and the EMA attention mechanism for loader bucket working angle identification is proposed. Initially, a small target detection head, utilizing YOLOv5s, was designed to enhance sensitivity towards target recognition. The EMA attention mechanism was introduced to increase the recognition rate of the target area and the positioning accuracy of the target frame, effectively differentiating the background area from the target area. The Focal-EIOU Loss function was added to address the slow convergence speed of YOLOv5. Subsequently, Depth Separable Convolution was employed to replace the standard convolution in the C3 module of the Backbone, improving the model’s accuracy in identifying target deformation caused by changes in the bucket angle, reducing the computational load, and enhancing the model’s operational speed. Experimental results demonstrate that the model’s mean Average Precision (mAP) value reached 99.3%, a 3.0% increase over the benchmark model YOLOv5s. The GFLOPs reached 58.5, an increase of 42, with a growth rate of 254.55%. This method effectively enhances the precision and intelligence of loader construction operations. |
format | Article |
id | doaj-art-80b3e32555cb4a9d9e3ede6559ed3d7c |
institution | Kabale University |
issn | 2169-3536 |
language | English |
publishDate | 2024-01-01 |
publisher | IEEE |
record_format | Article |
series | IEEE Access |
spelling | doaj-art-80b3e32555cb4a9d9e3ede6559ed3d7c2025-01-16T00:01:17ZengIEEEIEEE Access2169-35362024-01-011210548810549610.1109/ACCESS.2024.343514610613619Loader Bucket Working Angle Identification Method Based on YOLOv5s and EMA Attention MechanismXuedong Zhang0https://orcid.org/0000-0003-2018-9827Bo Cui1https://orcid.org/0009-0002-8933-9168Zhaoxu Wang2Wangting Zeng3https://orcid.org/0009-0004-7617-1508College of Artificial Intelligence, North China University of Science and Technology, Tangshan, ChinaCollege of Artificial Intelligence, North China University of Science and Technology, Tangshan, ChinaCollege of Artificial Intelligence, North China University of Science and Technology, Tangshan, ChinaCollege of Artificial Intelligence, North China University of Science and Technology, Tangshan, ChinaIn response to the issues of low recognition efficiency and large errors encountered in the process of identifying the working angle of the bucket during current automated loader construction operations, a method based on YOLOv5s and the EMA attention mechanism for loader bucket working angle identification is proposed. Initially, a small target detection head, utilizing YOLOv5s, was designed to enhance sensitivity towards target recognition. The EMA attention mechanism was introduced to increase the recognition rate of the target area and the positioning accuracy of the target frame, effectively differentiating the background area from the target area. The Focal-EIOU Loss function was added to address the slow convergence speed of YOLOv5. Subsequently, Depth Separable Convolution was employed to replace the standard convolution in the C3 module of the Backbone, improving the model’s accuracy in identifying target deformation caused by changes in the bucket angle, reducing the computational load, and enhancing the model’s operational speed. Experimental results demonstrate that the model’s mean Average Precision (mAP) value reached 99.3%, a 3.0% increase over the benchmark model YOLOv5s. The GFLOPs reached 58.5, an increase of 42, with a growth rate of 254.55%. This method effectively enhances the precision and intelligence of loader construction operations.https://ieeexplore.ieee.org/document/10613619/Angle identificationYOLOv5ssmall object detectionattention mechanismdepth separable convolution |
spellingShingle | Xuedong Zhang Bo Cui Zhaoxu Wang Wangting Zeng Loader Bucket Working Angle Identification Method Based on YOLOv5s and EMA Attention Mechanism IEEE Access Angle identification YOLOv5s small object detection attention mechanism depth separable convolution |
title | Loader Bucket Working Angle Identification Method Based on YOLOv5s and EMA Attention Mechanism |
title_full | Loader Bucket Working Angle Identification Method Based on YOLOv5s and EMA Attention Mechanism |
title_fullStr | Loader Bucket Working Angle Identification Method Based on YOLOv5s and EMA Attention Mechanism |
title_full_unstemmed | Loader Bucket Working Angle Identification Method Based on YOLOv5s and EMA Attention Mechanism |
title_short | Loader Bucket Working Angle Identification Method Based on YOLOv5s and EMA Attention Mechanism |
title_sort | loader bucket working angle identification method based on yolov5s and ema attention mechanism |
topic | Angle identification YOLOv5s small object detection attention mechanism depth separable convolution |
url | https://ieeexplore.ieee.org/document/10613619/ |
work_keys_str_mv | AT xuedongzhang loaderbucketworkingangleidentificationmethodbasedonyolov5sandemaattentionmechanism AT bocui loaderbucketworkingangleidentificationmethodbasedonyolov5sandemaattentionmechanism AT zhaoxuwang loaderbucketworkingangleidentificationmethodbasedonyolov5sandemaattentionmechanism AT wangtingzeng loaderbucketworkingangleidentificationmethodbasedonyolov5sandemaattentionmechanism |