改进损失函数的地基云状目标检测算法  

Ground Nephogram Object Detection Algorithm Based on Improved Loss Function

作　　者：王胜春 陈阳 WANG Shengchun;CHEN Yang(College of Information Science and Engineering,Hunan Normal University,Changsha 410081,China)

机构地区：[1]湖南师范大学信息科学与工程学院,长沙410081

出　　处：《计算机工程与应用》2022年第2期169-175,共7页Computer Engineering and Applications

基　　金：湖南省重点领域研发计划(2019SK2161);湖南省重点研发计划(2016SK2017)。

摘　　要：云是一种常见的天气现象,云状是天气预测的关键特征。目前,地基云图像中的云状观测主要依赖于气象观测员的目视观测,十分依赖观测员的经验,实时性和效率较低。针对这一问题,提出使用深度学习的方法进行地基云状检测识别。设计了一种新的目标检测边界框损失函数UIoU,将其应用于YOLOv3算法上。并且使用了K-means聚类算法重新设计了适用于地基云状数据集的先验框尺寸,使得边界框回归更加精确和稳定。实验结果表明UIoU-YOLOv3相比于原算法精度得到了有效提升,在VOC数据集和地基云状数据集上mAP数值分别提升了3.4个百分点和2.56个百分点。Cloud is a common weather phenomenon,and cloud shape is a key feature of weather prediction.At present,cloud observations in ground nephogram images mainly rely on the visual observations of meteorological observers,and rely heavily on the experience of the observers,with low real-time performance and low efficiency.In response to this problem,the application of deep learning is proposed for ground nephogram detection and recognition.Firstly,a new object detection bounding box loss function UIoU is designed and applied to the YOLOv3 algorithm.And the K-means clustering algorithm is used to redesign the anchor box size suitable for ground nephogram datasets,so that the bounding box returns more precisely and stably.Experimental results show that the accuracy of UIoU-YOLOv3 has been effectively improved in comparision with original algorithm,and the mAP values on the VOC datasets and the ground nephogram datasets have been increased by 3.4 percentage points and 2.56 percentage points,respectively.

关 键 词：目标检测 地基云状识别 YOLOv3 DIoU 

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