第40卷第1期2023年2月河北省科学院学报JournaloftheHebeiAcademyofSciencesVol.40No.1Feb.2023收稿日期:2022-10-10基金项目:河北省高等学校科学研究计划项目(QN2022061)作者简介:王彦雅(1997-),男,河北廊坊人,硕士研究生,研究方向:深度学习、目标检测.文章编号:1001-9383(2023)01-0001-06基于密集编码标签的遥感图像旋转目标检测算法王彦雅,李卫东,张伟娜(河北经贸大学信息技术学院,河北石家庄050061)摘要:针对现有目标检测算法在遥感图像中检测精度低,容易漏检等问题,提出了一种遥感图像旋转目标检测算法,使用YOLOv5m作为基本框架。首先,使用环形平滑标签CSL(CircularSmoothLabel,CSL)将角度回归预测转变为角度分类预测,解决回归预测中的角度周期性和边界可交换性的问题,提升检测精度。其次,使用密集编码标签(DenselyCodedLabel,DCL)替换稀疏编码标签,大幅减少预测层厚度,提升训练速度。实验表明,使用改进后的算法较基准算法mAP提升4.88%,模型训练速度与原模型速度基本相同,证明了算法的有效性。关键词:旋转目标检测;遥感图像;环形平滑标签;密集编码标签中图分类号:TP391.4文献标识码:ARotatingtargetdetectionalgorithminremotesensingimagebasedondenselycodedlabelWANGYan-ya,LIWei-dong,ZHANGWei-na(CollegeofInformationTechnology,HebeiUniversityofEconomicsandBusiness,ShijiazhuangHebei050061,China)Abstract:Aimingattheproblemsoflowdetectionaccuracyandeasytomissdetectionofexistingtargetdetectionalgorithmsinremotesensingimages,aremotesensingimagerotationtargetdetectionalgorithmisproposed,usingYOLOv5masthebasicframework.Firstly,thecircularsmoothlabelisusedtotransformtheangleregressionpredictionintoangleclassificationprediction,inordertosolvetheproblemofangleperiodicityandboundaryinterchangeabilityintheregressionpredictionandimprovethedetectionaccuracy.Secondly,denselycodedlabelisusedtoreplacesparsecodinglabel,whichgreatlyreducesthethicknessofpredictionlayerandimprovetrainingspeed.Theexperimentshowsthattheimprovedalgorithmis4.88%higherthanthebenchmarkalgorithmmAP,andthemodeltrainingspeedisbasicallythesameastheoriginalmodelspeed,whichprovestheeffectivenessofthealgorithm.DOI:10.16191/j.cnki.hbkx.2023.01.010河北省科学院学报2023年第40卷Keywords:Rotatingtargetde...
