火力与指挥控制2024,Vol.49Issue(9):76-82,89,8.DOI:10.3969/j.issn.1002-0640.2024.09.011
基于改进RetinaNet的轻量化航拍目标检测方法
Lightweight Aerial Object Detection Method Based on Improved RetinaNet
摘要
Abstract
Aiming at the problems of smaller targets,different scales and large number of model parameters of target detection algorithm in UAV aerial images,a lightweight target detection algorithm model,Retinanet-S,based on RetinaNet is proposed.Firstly,GhostNetV2 is used as the lightweight back-bone to enhance the characteristic information of the targets.Secondly,while the lightweight modules GSConv and VoV-GSCSP are used in the neck to ensure the detection accuracy and to reduce the number of model parameters.Finally,the Loss function is improved into Focal SIoU Loss to further improve the accuracy of the model.Through the experiment on VisDrone2019 dataset,the parameter amount of RetinaNet-S is only 5.52 M,the detection speed is increased by 9.55 FPS.关键词
航拍图像/目标检测/RetinaNet/轻量化Key words
aerial images/object detection/RetinaNet/lightweight分类
信息技术与安全科学引用本文复制引用
刘砚菊,王雪梅,宋建辉,刘晓阳,蒲家鹏..基于改进RetinaNet的轻量化航拍目标检测方法[J].火力与指挥控制,2024,49(9):76-82,89,8.基金项目
辽宁省教育厅高等学校基本科研项目(LJKZ0275) (LJKZ0275)
沈阳市中青年科技创新人才支持计划项目(RC210247) (RC210247)
辽宁省属本科高校基本科研业务费专项资金资助项目 ()