1. Object Detection Based on Improved YOLOv7 for UAV Aerial Image.
- Author
-
CUI Liqun and CAO Huawei
- Subjects
OBJECT recognition (Computer vision) ,FEATURE extraction ,AERIAL photography ,IMAGE reconstruction ,PROBLEM solving - Abstract
An improved YOLOv7 aerial image object detection algorithm is proposed to solve the problems of low detection accuracy caused by mesoscale changes, small targets and dense occlusion in UAV aerial images. Firstly, a weighted sampling module with joint dynamic convolution is designed to capture features from multiple dimensions and improve the feature extraction ability of the model. Secondly, add a shallow feature detection head to retain more detailed information and enhance the ability to utilize small target features. Then, a multi- scale feature aggregation module (C2-Res2Block) with residual structure is constructed in the feature fusion part to make the model fuse rich multi-scale information. Finally, the MPDIoU measure is used to replace the traditional IOU to calculate the boundary regression loss and improve the localization ability of the model to the densely occluding target. Experiments on UAV aerial photography data set VisDrone2019 show that the improved algorithm is 4.3 percentage points higher than the original model on mAP@0.5, 2.4 percentage points on mAP@0.5: 0.95, the number of parameters is reduced by 6.81x10
6 , and the detection accuracy is higher than the current mainstream object detection algorithms. It effectively improves the detection accuracy of UAV aerial images, and obviously improves the false detection and missing detection of aerial objects. [ABSTRACT FROM AUTHOR]- Published
- 2024
- Full Text
- View/download PDF