A Fine-Grained Object Detection Model for Aerial Images Based on YOLOv5 Deep Neural Network  

作　　者：ZHANG Rui XIE Cong DENG Liwei 

机构地区：[1]Heilongjiang Provincial Key Laboratory of Complex Intelligent System and Integration,School of Automation,Harbin University of Science and Technology,Harbin 150080,China

出　　处：《Chinese Journal of Electronics》2023年第1期51-63,共13页电子学报（英文版）

基　　金：supported by the National Science Foundation of Heilongjiang Province(LH2019F024);the Key R&D Program Guidance Projects of Heilongjiang Province(GZ20210065)。

摘　　要：Many advanced object detection algorithms are mainly based on natural scenes object and rarely dedicated to fine-grained objects.This seriously limits the application of these advanced detection algorithms in remote sensing object detection.How to apply horizontal detection in remote sensing images has important research significance.The mainstream remote sensing object detection algorithms achieve this task by angle regression,but the periodicity of angle leads to very large losses in this regression method,which increases the difficulty of model learning.Circular smooth label(CSL)solved this problem well by transforming the regression of angle into a classification form.YOLOv5 combines many excellent modules and methods in recent years,which greatly improves the detection accuracy of small objects.We use YOLOv5 as a baseline and combine the CSL method to learn the angle of arbitrarily oriented targets,and distinguish the fine-grained between instance classes by adding an attention mechanism module to accomplish the fine-grained target detection task for remote sensing images.Our improved model achieves an average category accuracy of 39.2%on the FAIR1M dataset.Although our method does not achieve satisfactory results,this approach is very efficient and simple,reducing the hardware requirements of the model.

关 键 词：Fine-grain object detection High-resolution aerial images Oriented object detection YOLOv5 

分 类 号：TP391.41[自动化与计算机技术—计算机应用技术]