High-altitude Nut Recognition Algorithm Based on Improved YOLOv5

Fang-fang MENG , Xiao-zhuang TIAN , Wei FANG , Dong-ying ZHANG , Yun ZAN , Chen ZHANG , Peng ZHAN

Science Technology and Engineering ›› 2025, Vol. 25 ›› Issue (1) : 262 -269.

PDF
Science Technology and Engineering ›› 2025, Vol. 25 ›› Issue (1) : 262-269 DOI: 10.12404/j.issn.1671-1815.2402469
research-article
High-altitude Nut Recognition Algorithm Based on Improved YOLOv5
Author information +
History +
PDF

Abstract

In order to improve the recognition accuracy of high-altitude nuts and reduce the false detection rate of bolts and nuts, a high-altitude nut recognition model based on improved YOLOv5 was proposed. Firstly, a new attention mechanism efficient multi-scale attention(EMA) was added to the backbone network to integrate more information. Secondly, in order to enhance the network's feature extraction capability, bidirectional feature pyramid network (BiFPN) was used to replace the PANet of the neck network. Finally, structured intersection over union(SIoU) was used to replace the original loss function complete intersection-over-union(CIoU) to accelerate the convergence of the model and improve its classification accuracy. The results show that the improved model has better performance than the original YOLOv5 model. The accuracy of the improved model increases by 0.92%. The recall increases by 0.16%. The average precision 1 (mAP_0.5:0.5) increases by 0.53%. And the average precision 2 (mAP:0.95) increases by 2.26%. An actual recognition comparison experiment between the improved model and the original YOLOv5 model was carried out. The experimental results show that the improved model has better recognition performance, which reduces the missed detection rate and the false detection rate, and improves the actual recognition rate. The improved model can well meet the recognition and image data acquisition of high-altitude nuts. And it also provide a data foundation for subsequent nut maintenance.

Keywords

nut / object detection / YOLOv5 / attention mechanism

Cite this article

Download citation ▾
Fang-fang MENG, Xiao-zhuang TIAN, Wei FANG, Dong-ying ZHANG, Yun ZAN, Chen ZHANG, Peng ZHAN. High-altitude Nut Recognition Algorithm Based on Improved YOLOv5. Science Technology and Engineering, 2025, 25(1): 262-269 DOI:10.12404/j.issn.1671-1815.2402469

登录浏览全文

4963

注册一个新账户 忘记密码

References

Funding

PDF (10108KB)

19

Accesses

0

Citation

Detail

Sections
Recommended

/

〈 〉