Research on small target detection in remote sensing images based on improved YOLOv7

Runjie Liu; Le Chen; Genhao Cai; Yiyang Liu

doi:10.1117/12.3011793

7 December 2023 Research on small target detection in remote sensing images based on improved YOLOv7

Runjie Liu, Le Chen, Genhao Cai, Yiyang Liu

Proceedings Volume 12941, International Conference on Algorithms, High Performance Computing, and Artificial Intelligence (AHPCAI 2023); 129413Y (2023) https://doi.org/10.1117/12.3011793
Event: Third International Conference on Algorithms, High Performance Computing, and Artificial Intelligence (AHPCAI 203), 2023, Yinchuan, China

Abstract

To solve the current target detection algorithms in remote sensing images, which have problems of small target detection accuracy is not high and easy to produce error detection and omission detection under the target dense and complex background, this paper introduces two enhancements rooted in the YOLOv7 algorithm. First, the ACmix attention mechanism is added after the backbone network of YOLOv7 algorithm, and second, drawing on the ideas of Soft-NMS algorithm and DIOU-NMS algorithm, CHSD-NMS algorithm is proposed, which improves the base NMS algorithm by softening the strategy of eliminating the candidate box and replacing the IOU with the DIOU to change the traditional NMS algorithm violent removal of candidate boxes. To evaluate the efficacy of the proposed algorithm in this study, experiments are conducted on the remote sensing image dataset DOTA and compared with the basic YOLOv7 algorithm. The experimental results indicate that the algorithm's average accuracy improves by 2.9% with the incorporation of the ACmix attention mechanism at IOU=0.5, a further enhancement of 3.2% is achieved by substituting the traditional NMS algorithm with the CHSD-NMS algorithm, and the overall average accuracy of the algorithm in this study increases by 5.2% with the implementation of both improvements compared to the conventional algorithm.

(2023) Published by SPIE. Downloading of the abstract is permitted for personal use only.

Citation Download Citation

Runjie Liu, Le Chen, Genhao Cai, and Yiyang Liu "Research on small target detection in remote sensing images based on improved YOLOv7", Proc. SPIE 12941, International Conference on Algorithms, High Performance Computing, and Artificial Intelligence (AHPCAI 2023), 129413Y (7 December 2023); https://doi.org/10.1117/12.3011793

ACCESS THE FULL ARTICLE

INSTITUTIONAL
Select your institution to access the SPIE Digital Library.

SELECT YOUR INSTITUTION

PERSONAL
Sign in with your SPIE account to access your personal subscriptions or to use specific features such as save to my library, sign up for alerts, save searches, etc.

PERSONAL SIGN IN

No SPIE Account? Create one

PURCHASE THIS CONTENT

SUBSCRIBE TO DIGITAL LIBRARY

50 downloads per 1-year subscription

Members: $195

Non-members: $335 ADD TO CART

25 downloads per 1 - year subscription

Members: $145

Non-members: $250 ADD TO CART

PURCHASE SINGLE ARTICLE

Includes PDF, HTML & Video, when available

Members: $17.00

Non-members: $21.00 ADD TO CART

PROCEEDINGS
8 PAGES

DOWNLOAD PAPER SAVE TO MY LIBRARY

GET CITATION

RIGHTS & PERMISSIONS

Get copyright permission Get copyright permission on Copyright Marketplace

KEYWORDS

Detection and tracking algorithms

Target detection

Small targets

Remote sensing

Ablation

Feature extraction

Convolution

Show All Keywords

Keywords/Phrases

Search In:

Publication Years