一种改进的基于YOLO V2的船舶目标检测方法技术

技术编号:20045694 阅读:19 留言:0更新日期:2019-01-09 04:24
一种改进的基于YOLO V2的船舶目标检测方法,包括以下步骤:(1)依据VOC数据集格式制作船舶数据集;(2)以改进的Darknet‑19网络作为基础网络,做YOLO V2的特征提取器和分类网络,添加额外的辅助网络层在改进的Darknet‑19上完成定位检测,构成完整的YOLO V2目标检测网络;(3)利用搭建好的网络结构训练模型;(4)利用训练出的模型进行船舶目标检测及模型评估。效果1:取代了传统的非成像技术和成像技术的船舶检测方法,避免了外界干扰对船舶检测带来的影响,并且基于YOLO V2的实时性和准确度能满足海洋监管系统的实时监测、准确检测的要求。效果2:针对小目标检测效果不好的问题,对网络做了改进,在很大程度上解决了目标检测领域小目标检测效果不好的问题。

【技术实现步骤摘要】
一种改进的基于YOLOV2的船舶目标检测方法
本专利技术涉及计算机视觉、深度学习技术等领域,具体设计一种改进的基于YOLOV2的船舶目标检测方法。
技术介绍
海洋与人类的生活紧密相连,息息相关,在军事上更是战略地位举足轻重,意义重大。近年来我国海上安全形势日趋复杂,别国侵犯我国海域,非法捕捞现象严重,海上安全问题频发,海洋监管和对入侵船舶的实时检测变得至关重要。但是传统方法的检测精度和速度都不能达到要求,所以迫切需要一种更快更准确的检测方法。现有的传统船舶检测方法主要有两种:非成像技术和成像技术检测。非成像技术检测譬如脉冲雷达、声呐等,通过主动、被动的工作方式获取船舶目标的辐射信号或者回波,利用信号处理技术和模式识别技术对目标信号进行分析检测,但由于水下环境的复杂性、不可预性,船舶自身系统的辐射场的干扰以及通信传输信道的繁杂,给检测带来很多困难。而红外、雷达和卫星遥感等成像技术又易受海面波浪噪声、浓雾的影响,且监测范围小。针对船舶目标检测,有一些相对较好的方案,如:基于FasterR-CNN的目标检测方法,分别训练一个RPN(RegionProposalNetwork)和一个fast-rcnn卷积神经网络,RPN直接预测出建议框,卷积网络和FastR-CNN部分共享,该方法检测准确率很高,但不能满足实时的要求。基于YOLO(YouOnlyLookOnce)的目标检测,直接回归了目标的位置和类别,无需RegionProposal,满足了实时性的要求,但检测精度不尽人意。基于SSD(SingleShotMultiboxDetector)的目标检测方法,同样直接回归了目标的位置和类别,检测速度和检测精度也差强人意,但对小目标的检测效果并不是很好。
技术实现思路
专利技术目的:本专利技术的目的在于克服上述现有技术的不足,提出了一种改进的基于YOLOV2的船舶目标检测方法,提高船舶目标检测的准确率,达到实时检测的要求。本专利技术所采用的的技术方案是:一种改进的基于YOLOV2的船舶目标检测方法,其特征在于包括以下步骤:(1)依据VOC数据集格式制作船舶数据集;(2)以改进的Darknet-19网络作为基础网络,做YOLOV2的特征提取器和分类网络,添加额外的辅助网络层在改进的Darknet-19上完成定位检测,构成完整的YOLOV2目标检测网络;(3)利用搭建好的网络结构训练模型;(4)利用训练出的模型进行船舶目标检测及模型评估。步骤(1)中建立用来存放数据集的文件夹Boat-detection,在Boat-detection文件夹下生成三个文件夹,命名分别为Annotations、ImageSets、JPEGImages文件夹,其中在ImageSets文件夹下生成一个子文件夹--Main文件夹;调整船舶图片数据格式为.jpg格式,并按照PASCALVOC官方命名方法将图片数据从000001.jpg开始统一重命名,将图片数据存放在JPEGImages文件夹中;对图片数据进行标注,即标出目标的类别和位置信息,将标注信息保存为同名的.xml格式文件,存放至Annotations文件夹中;将已有数据按比例生成训练样本集和测试样本集,生成train.txt和test.txt文件,里面存放训练样本集和测试样本集的绝对路径信息,将.txt文件放至ImageSets文件夹下的Main文件夹中。所述步骤(2)中改进的Darknet-19网络由8个卷积模块、5个池化模块、1个通过层(passthroughlayer)模块、1个均值池化模块、1个分类模块组成。所述检测图像作为第一卷积模块的输入;第一卷积模块、第一池化模块、第二卷积模块、第二池化模块、第三卷积模块、第三池化模块、第四卷积模块、第五卷积模块、第六卷积模块、第四池化模块、一个通过层模块、第七卷积模块、第五池化模块、第八卷积模块、一个均值池化模块、一个softmax分类模块依次级联后的输出来完成检测目标的特征提取和分类的。去掉第八卷积模块、一个均值池化模块、一个softmax分类模块,在后面加上新的第八卷积模块、第九卷积模块、一个通过层模块、第十卷积模块、第十一卷积模块依次重新级联作为额外的辅助网络层完成检测目标定位检测。步骤(3)中训练模型的过程如下:初始化网络待训练参数及超参数,向初始化后的网络中输入训练样本数据进行网络前向传播,得到实际的输出结果通过损失函数结合反向传播BP算法进行反馈调整网络参数,进行迭代训练,迭代至损失函数的损失值降至设定阈值或迭代次数达到最大迭代次数时结束训练,得到用于船舶目标检测的网络模型。通过损失值对模型进行评估反馈。若反馈良好,则模型设计合理;否则,返回步骤(2),利用反馈结果对网络结构和参数进行再次分析改进直至模型反馈达到预期要求。所述的训练样本数据包括待训练的图像样本及其目标位置、类别信息。步骤(4)利用已训练好的网络模型对测试样本进行测试验证,得到检测目标的位置和类别。优点效果:本专利技术有以下两种良好的效果:效果1:取代了传统的非成像技术和成像技术的船舶检测方法,避免了外界干扰对船舶检测带来的影响,并且基于YOLOV2的实时性和准确度能满足海洋监管系统的实时监测、准确检测的要求。效果2:针对小目标检测效果不好的问题,对网络做了改进,在很大程度上解决了目标检测领域小目标检测效果不好的问题。附图说明图1是本专利技术的步骤图;图2是实施例1中的数据制作存放的结构图;图3是实施例1中的改进的YOLOV2目标检测网络的结构图。具体实施方式一种改进的基于YOLOV2的船舶目标检测方法,该方法包括以下步骤:(1)依据VOC数据集格式制作船舶数据集;(2)以改进的Darknet-19网络作为基础网络,做YOLOV2的特征提取器和分类网络,添加额外的辅助网络层在改进的Darknet-19上完成定位检测,构成完整的YOLOV2目标检测网络;(3)利用搭建好的网络结构训练模型,对模型进行评估;(4)利用训练出的模型进行船舶目标检测及模型测试。以下将结合附图和具体实施方式,对本专利技术作进一步的详细说明。步骤(2)中改进的Darknet-19网络由8个卷积模块、5个池化模块、1个通过层(passthroughlayer)模块、1个均值池化模块、1个分类模块组成;所述检测图像作为第一卷积模块的输入;第一卷积模块、第一池化模块、第二卷积模块、第二池化模块、第三卷积模块、第三池化模块、第四卷积模块、第五卷积模块、第六卷积模块、第四池化模块、一个通过层模块、第七卷积模块、第五池化模块、第八卷积模块、一个均值池化模块、一个softmax分类模块依次级联后的输出来完成检测目标的特征提取和分类的;去掉第八卷积模块、一个均值池化模块、一个softmax分类模块,在后面加上新的第八卷积模块、第九卷积模块、一个通过层模块、第十卷积模块、第十一卷积模块依次重新级联作为额外的辅助网络层完成检测目标定位检测。步骤(3)中训练模型的过程如下:初始化网络待训练参数及超参数,向初始化后的网络中输入训练样本数据进行网络前向传播,得到实际的输出结果通过损失函数结合反向传播BP算法进行反馈调整网络参数,进行迭代训练,迭代至损失函数的损失值降至设定阈值或迭代次数达到最大迭代次数时结束训练,得到用于船舶目标检测的本文档来自技高网
...

【技术保护点】
1.一种改进的基于YOLO V2的船舶目标检测方法,其特征在于:该方法包括以下步骤:(1)依据VOC数据集格式制作船舶数据集;(2)以改进的Darknet‑19网络作为基础网络,做YOLO V2的特征提取器和分类网络,添加额外的辅助网络层在改进的Darknet‑19上完成定位检测,构成完整的YOLO V2目标检测网络;(3)利用搭建好的网络结构训练模型,对模型进行评估;(4)利用训练出的模型进行船舶目标检测及模型测试。

【技术特征摘要】
2018.06.15 CN 20181061768671.一种改进的基于YOLOV2的船舶目标检测方法,其特征在于:该方法包括以下步骤:(1)依据VOC数据集格式制作船舶数据集;(2)以改进的Darknet-19网络作为基础网络,做YOLOV2的特征提取器和分类网络,添加额外的辅助网络层在改进的Darknet-19上完成定位检测,构成完整的YOLOV2目标检测网络;(3)利用搭建好的网络结构训练模型,对模型进行评估;(4)利用训练出的模型进行船舶目标检测及模型测试。2.根据权利要求书1所述的一种改进的基于YOLOV2的船舶目标检测方法,其特征在于:步骤(1)中建立用来存放数据集的文件夹Boat-detection,在Boat-detection文件夹下生成三个文件夹,命名分别为Annotations、ImageSets、JPEGImages文件夹,其中在ImageSets文件夹下生成一个子文件夹--Main文件夹;调整船舶图片数据格式为.jpg格式,并按照PASCALVOC官方命名方法将图片数据从000001.jpg开始统一重命名,将图片数据存放在JPEGImages文件夹中;对图片数据进行标注,即标出目标的类别和位置信息,将标注信息保存为同名的.xml格式文件,存放至Annotations文件夹中;将已有数据按比例生成训练样本集和测试样本集,生成train.txt和test.txt文件,里面存放训练样本集和测试样本集的绝对路径信息,将.txt文件放至ImageSets文件夹下的Main文件夹中。3.根据权利要求书1所述的一种改进的基于YOLOV2的船舶目标检测方法,其特征在于:步骤(2)中改进的Darknet-19网络由8个卷积模块、5个池...

【专利技术属性】
技术研发人员:陈亮刘韵婷于洋宋建辉李世杰
申请(专利权)人:沈阳理工大学
类型:发明
国别省市:辽宁,21

网友询问留言 已有0条评论
  • 还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。

1