首页 | 官方网站   微博 | 高级检索  
     

基于改进YOLOv2和迁移学习的管道巡检航拍图像第三方施工目标检测
引用本文:谌贵辉,易欣,李忠兵,钱济人,陈伍.基于改进YOLOv2和迁移学习的管道巡检航拍图像第三方施工目标检测[J].计算机应用,2020,40(4):1062-1068.
作者姓名:谌贵辉  易欣  李忠兵  钱济人  陈伍
作者单位:1. 西南石油大学 电气信息学院, 成都 610500;2. 浙江浙能天然气运行有限公司 科创中心, 杭州 310000
基金项目:南充市市校科技战略合作项目(18SXHZ0041)。
摘    要:针对传统目标检测算法应用在无人机航拍图像上第三方施工目标检测和违章占压建筑检测的数据集少、检测率低等问题,提出基于Aerial-YOLOv2和迁移学习的航拍图像目标检测算法。首先,利用结合数据增强的迁移学习策略训练的网络来扩大数据集规模,并利用K均值聚类分析得到符合所提数据集特点的锚点框数量和尺寸;其次,通过自适应对比度增强的方法对图像进行预处理;最后,提出改进卷积模块替代YOLOv2中的卷积块并结合特征融合的多尺度预测方式进行目标检测。用不同的算法和训练策略在无人机航拍图像上进行对比实验,实验结果表明,Aerial-YOLOv2算法结合多种训练策略后,其准确率、召回率分别能达到95%、91%,每张图像检测时间为14 ms。由此可知,该算法适用于无人机航拍图像第三方施工目标及违章占压建筑的智能检测。

关 键 词:管道巡检  深度学习  航拍图像  小目标检测  数据增强  迁移学习  
收稿时间:2019-09-05
修稿时间:2019-11-11

Third-party construction target detection in aerial images of pipeline inspection based on improved YOLOv2 and transfer learning
CHEN Guihui,YI Xin,LI Zhongbing,QIAN Jiren,CHEN Wu.Third-party construction target detection in aerial images of pipeline inspection based on improved YOLOv2 and transfer learning[J].journal of Computer Applications,2020,40(4):1062-1068.
Authors:CHEN Guihui  YI Xin  LI Zhongbing  QIAN Jiren  CHEN Wu
Affiliation:1. College of Electrical Engineering and Information, Southwest Petroleum University, Chengdu Sichuan 610500, China;2. Technology Innovation Center, Zhejiang ZheNeng Natural Gas Operation Company Limited, Hangzhou Zhejiang 310000, China
Abstract:Aiming at the few datasets and low detection rate when the traditional target detection algorithm applying to third-party construction target detection and illegally occupied building detection in the aerial images of drone,an aerial image target detection algorithm based on Aerial-YOLOv2 and transfer learning was proposed. Firstly,the trained network combining with data enhancement and transfer learning strategy was used to expand the dataset size,and K-means clustering analysis was used to obtain the number and size of anchor blocks that meet the characteristics of the proposed dataset. Secondly,the adaptive contrast enhancement was used to pre-process the image. Finally,the improved convolution module was proposed to replace the convolution block in YOLOv2,and the feature fusion multi-scale prediction method was combined for target detection. The comparison experiments of different algorithms and training strategies on the aerial images of drone were carried out. Results show that the accuracy and recall rate of the Aerial-YOLOv2 algorithm combined with various training strategies can respectively reach 95% and 91%,and the detection time per image is 14 ms. It can be seen that the algorithm is suitable for the intelligent detection of third-party construction targets and illegally occupied buildings in the aerial images of drone.
Keywords:pipeline inspection  deep learning  aerial image  small-object detection  data enhancement  transfer learning
本文献已被 维普 万方数据 等数据库收录!
点击此处可从《计算机应用》浏览原始摘要信息
点击此处可从《计算机应用》下载全文
设为首页 | 免责声明 | 关于勤云 | 加入收藏

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

京公网安备 11010802026262号