Effective Small Object Detection in Remote Sensing Images Based on Improved YOLOv8 Network
Year: 2024
Type: article
Abstract: Small object detection has long been a challenging and prominent research areain computer vision. Driven by deep learning, small object detection has made majorbreakthroughs and has been successfully applied in fields such as national defense security,intelligent transportation, and industrial automation. In our research, we conduct acomprehensive analysis and improvement of the YOLOv8-n algorithm for object detection,and propose two new algorithms. The first algorithm adds an SE Attention module and adetection head for small objects in YOLOv8. Improvements include the following points: a)adding a detection head for small objects to enhance detection capabilities for small objects,b) adding an SE Attention module to the network to improve the detection capability ofthe model. The resulting algorithm improves the performance of YOLOv8 in small objectdetection on the remote sensing image DOTA-v2.0 dataset and reduces the interference ofnoisy data to a certain extent. The second model incorporates the CA attention mechanisminto the YOLOv8n model and uses the SEResNeXtBottleneck detection header instead of theYOLOv8n header. Through experimental validation on the DOTAv1 dataset, our improvedmodel demonstrates a more accurate ability to detect small objects, and the experimentalresults show that by adding the CA attention mechanism, the model is able to focus moreeffectively on key regions of the image, thereby improving detection accuracy.
Authors: Zhiyuan, Li, Hao, Wang, Ma, Guangdi, Weichen, Yang, Ablameyko, Sergey
Language: English
Subfield: Computer Vision and Pattern Recognition
Field: Computer Science
Domain: Physical Sciences
Open Access status: green
OpenAlex ID: W6911916732