Paper
11 September 2024 Improved YOLOv5m UAV viewpoint fight behavior detection algorithm
Baixuan Han, Yueping Peng, Hexiang Hao, Wenji Yin
Author Affiliations +
Proceedings Volume 13253, Fourth International Conference on Signal Image Processing and Communication (ICSIPC 2024); 1325317 (2024) https://doi.org/10.1117/12.3041508
Event: 4th International Conference on Signal Image Processing and Communication, 2024, Xi'an, China
Abstract
An improved YOLOv5m fighting behavior detection algorithm is proposed to address the problem of insufficient accuracy and real-time performance of behavior detection algorithms for aerial viewpoints. Three UAVs located at different heights and angles are used to shoot and construct the fight behavior dataset, after replacing the convolution module and introducing maximum feature pooling, a lightweight network is designed and the original backbone network is replaced to achieve the lightweight effect; an attention mechanism is added to the detection head to improve the detection accuracy of the model, and the effects of the two attention mechanisms on the model's detection accuracy are tested; bi-directional features are added to the neck The pyramid structure is added to the neck to strengthen the feature extraction ability; the results show that compared with the original network, the detection accuracy is improved from 88.3% to 93.4%, the number of parameters is reduced by 83.37%, the weight of the model is reduced from 17.02MB to 11.39MB and the number of detected frames per second reaches 48, which meets the requirements of lightweight and real-time detection.
(2024) Published by SPIE. Downloading of the abstract is permitted for personal use only.
Baixuan Han, Yueping Peng, Hexiang Hao, and Wenji Yin "Improved YOLOv5m UAV viewpoint fight behavior detection algorithm", Proc. SPIE 13253, Fourth International Conference on Signal Image Processing and Communication (ICSIPC 2024), 1325317 (11 September 2024); https://doi.org/10.1117/12.3041508
Advertisement
Advertisement
RIGHTS & PERMISSIONS
Get copyright permission  Get copyright permission on Copyright Marketplace
KEYWORDS
Convolution

Object detection

Detection and tracking algorithms

Unmanned aerial vehicles

Feature fusion

Feature extraction

Data modeling

Back to Top