首页 | 本学科首页   官方微博 | 高级检索  
     

采用改进YOLOv3的高分辨率遥感图像目标检测
引用本文:夏英,黄秉坤. 采用改进YOLOv3的高分辨率遥感图像目标检测[J]. 重庆邮电大学学报(自然科学版), 2022, 34(3): 383-392
作者姓名:夏英  黄秉坤
作者单位:重庆邮电大学 计算机科学与技术学院,重庆 400065
基金项目:国家自然科学基金(41871226);重庆市基础与前沿研究计划项目(cstc2019jcyjmsxmX0131)
摘    要:由于高分辨率遥感图像存在目标排列密集、尺寸差别大等情况,传统算法难以准确地对其进行目标检测。在YOLOv3算法的基础上,提出一种改进的高分辨率遥感图像目标检测算法(remote sensing-YOLO,RS-YOLO)。利用K-means聚类算法对数据集进行聚类,重新设计适合遥感图像的先验框; 引入高斯模型计算预测框的不确定度,以提高网络对预测框坐标的准确度; 使用弱化的非极大值抑制算法(soft non-aximum suppression,Soft-NMS)对预测框进行处理,增强算法对密集排列目标的检测能力。实验结果表明,改进后的算法能够对高分辨率遥感图像进行有效的目标检测,以NWPU VHR-10数据集为例,RS-YOLO的平均检测精度达到了87.97%。

关 键 词:深度学习  目标检测  高分辨率遥感图像  YOLOv3算法  高斯模型
收稿时间:2020-10-12
修稿时间:2022-03-10

Object detection of high resolution remote sensing images based on improved YOLOv3
XIA Ying,HUANG Bingkun. Object detection of high resolution remote sensing images based on improved YOLOv3[J]. Journal of Chongqing University of Posts and Telecommunications, 2022, 34(3): 383-392
Authors:XIA Ying  HUANG Bingkun
Affiliation:School of Computer Science and Technology, Chongqing University of Posts and Telecommunications, Chongqing 400065, P.R.China
Abstract:Due to the dense arrangement of targets and the big difference of target sizes, it is difficult to detect objectives accurately in high-resolution remote sensing images by traditional algorithms. An improved high-resolution remote sensing image object detection method (RS-YOLO) based on YOLOv3 algorithm is proposed. K-means algorithm is used to cluster data set and redesign the prior anchor box suitable for remote sensing image. Gaussian model is introduced to calculate the uncertainty information of bounding box to improve the accuracy of coordinates from the network. Soft-NMS algorithm is used to process the bounding box which enhance the detection ability for densely arranged targets. Experimental results show that the improved algorithm is effective in high-resolution remote sensing images object detection. In NWPU VHR-10 data set, the average detection precision of RS-YOLO reached 87.97%.
Keywords:deep learning  object detection  high resolution remote sensing image  YOLOv3 algorithm  Gaussian model
点击此处可从《重庆邮电大学学报(自然科学版)》浏览原始摘要信息
点击此处可从《重庆邮电大学学报(自然科学版)》下载全文
设为首页 | 免责声明 | 关于勤云 | 加入收藏

Copyright©北京勤云科技发展有限公司  京ICP备09084417号