Infrared Image Detection and Recognition of Substation Electrical Equipment Based on Improved YOLOv8
To address the challenges associated with lightweight design and small object detection in infrared imaging for substation electrical equipment, this paper introduces an enhanced YOLOv8_Adv network model. This model builds on YOLOv8 through several strategic improvements. The backbone network incorp...
Saved in:
| Main Authors: | , , |
|---|---|
| Format: | Article |
| Language: | English |
| Published: |
MDPI AG
2024-12-01
|
| Series: | Applied Sciences |
| Subjects: | |
| Online Access: | https://www.mdpi.com/2076-3417/15/1/328 |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1841549419645763584 |
|---|---|
| author | Haotian Tao Agyemang Paul Zhefu Wu |
| author_facet | Haotian Tao Agyemang Paul Zhefu Wu |
| author_sort | Haotian Tao |
| collection | DOAJ |
| description | To address the challenges associated with lightweight design and small object detection in infrared imaging for substation electrical equipment, this paper introduces an enhanced YOLOv8_Adv network model. This model builds on YOLOv8 through several strategic improvements. The backbone network incorporates PConv and FasterNet modules to substantially reduce the computational load and memory usage, thereby achieving model lightweighting. In the neck layer, GSConv and VoVGSCSP modules are utilized for multi-stage, multi-feature map fusion, complemented by the integration of the EMA attention mechanism to improve feature extraction. Additionally, a specialized detection layer for small objects is added to the head of the network, enhancing the model’s performance in detecting small infrared targets. Experimental results demonstrate that YOLOv8_Adv achieves a 4.1% increase in mAP@0.5 compared to the baseline YOLOv8n. It also outperforms five existing baseline models, with the highest accuracy of 98.7%, and it reduces the computational complexity by 18.5%, thereby validating the effectiveness of the YOLOv8_Adv model. Furthermore, the effectiveness of the model in detecting small targets in infrared images makes it suitable for use in areas such as infrared surveillance, military target detection, and wildlife monitoring. |
| format | Article |
| id | doaj-art-dcbb29ccc3c74ed2962af5b7e93f4dff |
| institution | Kabale University |
| issn | 2076-3417 |
| language | English |
| publishDate | 2024-12-01 |
| publisher | MDPI AG |
| record_format | Article |
| series | Applied Sciences |
| spelling | doaj-art-dcbb29ccc3c74ed2962af5b7e93f4dff2025-01-10T13:15:10ZengMDPI AGApplied Sciences2076-34172024-12-0115132810.3390/app15010328Infrared Image Detection and Recognition of Substation Electrical Equipment Based on Improved YOLOv8Haotian Tao0Agyemang Paul1Zhefu Wu2School of Electrical Engineering, Dalian University of Technology, Dalian 116024, ChinaCollege of Information Engineering, Zhejiang University of Technology, Hangzhou 310000, ChinaCollege of Information Engineering, Zhejiang University of Technology, Hangzhou 310000, ChinaTo address the challenges associated with lightweight design and small object detection in infrared imaging for substation electrical equipment, this paper introduces an enhanced YOLOv8_Adv network model. This model builds on YOLOv8 through several strategic improvements. The backbone network incorporates PConv and FasterNet modules to substantially reduce the computational load and memory usage, thereby achieving model lightweighting. In the neck layer, GSConv and VoVGSCSP modules are utilized for multi-stage, multi-feature map fusion, complemented by the integration of the EMA attention mechanism to improve feature extraction. Additionally, a specialized detection layer for small objects is added to the head of the network, enhancing the model’s performance in detecting small infrared targets. Experimental results demonstrate that YOLOv8_Adv achieves a 4.1% increase in mAP@0.5 compared to the baseline YOLOv8n. It also outperforms five existing baseline models, with the highest accuracy of 98.7%, and it reduces the computational complexity by 18.5%, thereby validating the effectiveness of the YOLOv8_Adv model. Furthermore, the effectiveness of the model in detecting small targets in infrared images makes it suitable for use in areas such as infrared surveillance, military target detection, and wildlife monitoring.https://www.mdpi.com/2076-3417/15/1/328YOLOv8infrared imagetarget detectionlightweightdetection accuracy |
| spellingShingle | Haotian Tao Agyemang Paul Zhefu Wu Infrared Image Detection and Recognition of Substation Electrical Equipment Based on Improved YOLOv8 Applied Sciences YOLOv8 infrared image target detection lightweight detection accuracy |
| title | Infrared Image Detection and Recognition of Substation Electrical Equipment Based on Improved YOLOv8 |
| title_full | Infrared Image Detection and Recognition of Substation Electrical Equipment Based on Improved YOLOv8 |
| title_fullStr | Infrared Image Detection and Recognition of Substation Electrical Equipment Based on Improved YOLOv8 |
| title_full_unstemmed | Infrared Image Detection and Recognition of Substation Electrical Equipment Based on Improved YOLOv8 |
| title_short | Infrared Image Detection and Recognition of Substation Electrical Equipment Based on Improved YOLOv8 |
| title_sort | infrared image detection and recognition of substation electrical equipment based on improved yolov8 |
| topic | YOLOv8 infrared image target detection lightweight detection accuracy |
| url | https://www.mdpi.com/2076-3417/15/1/328 |
| work_keys_str_mv | AT haotiantao infraredimagedetectionandrecognitionofsubstationelectricalequipmentbasedonimprovedyolov8 AT agyemangpaul infraredimagedetectionandrecognitionofsubstationelectricalequipmentbasedonimprovedyolov8 AT zhefuwu infraredimagedetectionandrecognitionofsubstationelectricalequipmentbasedonimprovedyolov8 |