[发明专利]一种船舶检测方法、系统及计算机存储介质

说明书

本发明公开一种船舶检测方法、系统及计算机存储介质。其中，该方法包括：将原始训练集进行复制，并将复制好的图片从不同种类的数据增强中随机选择一种进行数据增强，将上述图片进行模型训练，得到第一更新船舶检测模型；对原始训练集进行不同种类的数据增强，得到增强训练集；将原始训练集、增强训练集通过第一更新船舶检测模型进行训练，计算得到全部目标的差异值，根据差异值将目标进行裁剪并粘贴到对应的原始图片中并进行融合操作，以使目标与原始图片的边缘虚化，中心突出；本发明利用原始迭代图片对增强迭代图片的引导，使模型不仅能学习原始迭代图片特征，还能学习各种场景下的增强迭代图片特征，提高不同环境下的准确率和降低错误率。

技术领域

本发明涉及目标检测技术领域，具体而言，涉及一种船舶检测方法、系统及计算机存储介质。

背景技术

随着水上交通的不断发展，导致水上交通指挥工作人员的作业量急剧上升。其中船舶检查是水上交通的重要工作之一，为了减少水上交通工作人员监视船舶的工作量，船舶检测在水上交通被广泛使用。但水上交通环境背景复杂且受到白天晚上光照影响，以及人为因素在训练集标定所犯的人为错误，目前的船舶检测方法在船舶检测中存在以下问题：

（1）水上交通复杂，数据采集无法全面采集，对环境拟合存在问题，误检偏多；

（2）水上交通实时监测，受到一年四季和一天光线变换影响；

（3）拷贝粘贴的目标存在比较明显的图像分界线；

（4）因人工出现错误标定结果。

针对上述问题，目前尚未提出有效的解决方案。

发明内容

本发明实施例中提供一种船舶检测方法、系统及计算机存储介质，以解决现有技术中水上交通复杂，数据采集无法全面采集，对环境拟合存在问题，误检偏多；水上交通实时监测，受到一年四季和一天光线变换影响；拷贝粘贴的目标存在比较明显的图像分界线；因人工出现错误标定结果的问题。

为达到上述目的，一方面，本发明提供了一种船舶检测方法，其中，该方法包括：S1，从原始训练集中抽取原始迭代图片；对每张原始迭代图片分别进行复制，将复制好的每张原始迭代图片从不同种类的数据增强中随机选择一种进行数据增强，得到增强迭代图片；S2，将每张原始迭代图片输入到所述初始船舶检测模型中进行模型训练，得到原始目标预测结果；以及将每张增强迭代图片输入到所述初始船舶检测模型中进行模型训练，得到增强目标预测结果；根据所述原始目标预测结果和所述增强目标预测结果计算得到迭代总损失值；根据所述迭代总损失值对所述初始船舶检测模型进行更新，得到当前迭代船舶检测模型；将原始训练集全部图片训练完，得到第一更新船舶检测模型以及第一轮总损失值；S3，将原始训练集输入到所述第一更新船舶检测模型中进行模型训练，得到每个目标的原始损失值；将原始训练集中每张图片分别进行不同种类的数据增强得到增强训练集，将所述增强训练集输入到所述第一更新船舶检测模型中进行模型训练，得到每个目标的增强损失值；根据所述原始损失值、所述增强损失值计算得到增强训练集中每个目标的差异值；将全部的所述差异值从大到小排列，并将排列后的第一预设范围内的差异值对应的目标进行裁剪；将裁剪的目标粘贴到对应的原始图片中并进行融合操作，以使目标与所述原始图片的边缘虚化，中心突出，得到第一粘贴图片集；S4，将所述第一粘贴图片集进行S1、S2操作，得到第二更新船舶检测模型以及第二轮总损失值；S5，将所述原始训练集输入到所述第二更新船舶检测模型，重复S3操作，得到第二粘贴图片集；S6，重复所述S4、S5，直至第二轮总损失值在第二预设范围内波动，停止模型训练，得到目标船舶检测模型；S7，将待检测图片输入到所述目标船舶检测模型中进行检测，得到目标位置和类别。

可选的，所述融合操作通过以下公式计算：

；

；

；

；