一种基于语义分割和霍夫变换结合的海天线检测方法技术

本发明专利技术提出了一种基于语义分割和霍夫变换结合的海天线检测方法，具体步骤为：S1：输入水上交通场景的可见光图像，制作水上交通图像的数据集，S2：建立水上交通图像Mo

【技术实现步骤摘要】
一种基于语义分割和霍夫变换结合的海天线检测方法


[0001]本专利技术涉及海天线检测
，特别涉及一种基于语义分割和霍夫变换结合的海天线检测方法。

技术介绍

[0002]海天线是将水上交通图像划分为天空区域和海面区域的边界线，是水上交通图像所具有的重要特征之一。在远距离水平观测的水上交通图像中，船舶目标大部分位于海天线附近，海天线检测有助于缩小船舶目标检测搜索范围，为船舶智能航行提供安全保障，在海上搜救、航线规划、海域监控等领域都具有重要意义。
[0003]目前许多文献提出了不同关于海天线检测的算法。其中，主要包括直线拟合法、梯度显著性、图像分割等方法，但是面对复杂水上交通场景，这些方法没有取得很好的效果。其主要原因在于水上交通图像中包含大量干扰海天线检测的因素，如条纹状的海浪与海天线具有一定的相似性、大雾天气下天空与海面较为相似、大量船只破坏海天线的连续性等。因此，在复杂水上交通场景精确检测存在一定的局限性。

技术实现思路

[0004]针对水上交通场景复杂，海天线检测干扰因素繁多，现有海天线检测技术在复杂水上交通场景下检测效果不佳的问题，本专利技术提出了一种基于语义分割和霍夫变换结合的海天线检测方法，采用Mo
‑
SegNet网络模型作为语义分割模型并霍夫变换拟合海天线，实现了复杂水上交通场景对海天线的精准检测。
[0005]技术方案如下所述：
[0006]一种基于语义分割和霍夫变换结合的海天线检测，包括以下步骤：
[0007]S1：输入水上交通场景的可见光图像，制作水上交通图像的数据集，所述数据集包括训练集和测试集，所述训练集经过扩充之后得到扩充训练集；
[0008]S2：建立水上交通图像Mo
‑
SegNet网络模型作为语义分割模型，利用S1中扩充训练集对所述Mo
‑
SegNet网络模型进行训练得到语义分割权重，利用训练好的Mo
‑
SegNet网络模型对输入的S1中测试集的水上交通图像进行逐像素分割输出语义分割图像；
[0009]S3：采用Canny检测算法对S2输出的语义分割的图像进行边缘检测，输出仅包含海天线边缘轮廓及背景的二值图像；
[0010]S4：在二值图像上进行霍夫变换，对海天线边缘轮廓的边缘像素进行拟合，得到海天线检测结果。
[0011]优选地，S1中，针对输入所述水上交通场景的可见光图像，将图像由标注软件逐像素进行标注制作数据集，将水上交通图像按照像素点分类为天空区域、海面区域和其他区域，并将所述数据集按8:2的比例分为训练集和测试集，对训练集进行旋转、镜像图像处理对训练集进行扩充，最终得到扩充训练集；所述其他区域包括海面船舶、漂浮物和遮挡物像素。
[0012]优选地，S2中，所述Mo
‑
SegNet网络模型由编码网络和解码网络构成；
[0013]所述编码网络由一次标准卷积和三次深度可分离卷积组成，每一次卷积均增加了一层BN层，对输出的特征图结果进行归一化；通过多层卷积实现对水上交通图像中天空、海面、船舶等像素特征的提取，得到相应的特征层；
[0014]所述解码网络采用三次重复的补零、卷积、批标准化、激活和上采样操作，将水上交通图像中的每一个像素赋予一个类别标签；在最后一层解码器输出高维特征表示向量，作为可训练Softmax分类器中函数的输入。
[0015]优选地，所述Softmax分类器中Softmax函数为：
[0016][0017]式中，i为第i个节点输出，z
i
为第i个节点的输出值，C为输出节点的个数，即分类的类别个数。
[0018]优选地，所述Softmax分类器中应用交叉熵作为损失函数，公式为：
[0019][0020]式中，L
ce
为损失值，M为类别数，T
i
为真实概率，F
i
为预测概率。
[0021]优选地，S3中，选用所述Canny检测算法对语义分割图像进行边缘检测的具体过程为：通过代表边缘检测的垂直和水平方向的算子，如水平方向的G
x
和垂直方向的G
y
所得出一阶导数值，推测出像素点的方向和像素点的梯度G，计算公式为：
[0022][0023][0024]式中，G代表梯度强度，G
x
代表x方向的梯度幅值，G
y
代表y方向的梯度幅值，代表像素点的方向，arctan为反正切函数。
[0025]优选地，S4中，采用所述霍夫变换拟合海天界限的边缘像素的方法是：
[0026]在图像空间中海天线边缘轮廓像素坐标内的直线设定为预测海天线；
[0027]在xy平面坐标系中，将任意一条预测海天线上每个点的像素坐标(x，y)通过参数变换到霍夫空间中，得到ρθ平面坐标系的一组单元格H(θ，ρ)，对所述单元格进行投票累加，得到该条预测海天线的累加值，选择累加值最大的预测海天线作为海天线的检测结果。
[0028]优选地，所述参数变换的公式为：ρ＝xcosθ+ysinθ
[0029]式中，ρ表示直线到原点的距离，θ为该直线与正x轴的夹角。
[0030]本专利技术提出了一种基于语义分割和霍夫变换结合的海天线检测方法，对经典语义分割网络SegNet网络进行改进，采用Mobilenet网络作为编码网络，并对解码网络进行精简，提出Mo
‑
SegNet网络，对输入的S1中测试集的水上交通图像进行逐像素分割，并且减少了网络参数。同时，该Mo
‑
SegNet网络对训练集进行扩充进行深度学习训练，扩充数据集，以增强网络的鲁棒性。第三，在S3中选用Canny检测算法对语义分割图像进行边缘检测，将语义分割图像中海天界限以及船舶边缘检测出来，有效排除天空中云层、海面上海浪等线性干扰因素的影响。第四，本专利技术采用霍夫变换拟合海天线，有效的解决了海天界限受船舶影响破坏连续性的问题。综上所述，本专利技术用Mo
‑
SegNet网络模型作为语义分割模型并霍夫变
换拟合海天线，实现了复杂水上交通场景对海天线的精准检测。
附图说明
[0031]图1为实施例中一种基于语义分割和霍夫变换结合的海天线检测方法流程图。
[0032]图2为实施例中一种水上交通场景的可见光图像。
[0033]图3为实施例中包含海天线边缘轮廓及背景的二值图像。
[0034]图4为实施例中基于语义分割和霍夫变换结合的海天线检测方法输出的海天线检测效果图。
具体实施方式
[0035]下面结合附图，对本专利技术的具体实施例做进一步说明。
[0036]实施例一：
[0037]如图1所示，一种基于语义分割和霍夫变换结合的海天线检测方法，包括以下步骤：
[0038]S1：输入水上交通场景的可见光图像，制作水上交通图像的数据集，所述数据集包括训练集和测试集，所述训练集经过本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种基于语义分割和霍夫变换结合的海天线检测，其特征在于，包括以下步骤：S1：输入水上交通场景的可见光图像，制作水上交通图像的数据集；所述数据集包括训练集和测试集，所述训练集经过扩充之后得到扩充训练集；S2：建立水上交通图像Mo
‑
SegNet网络模型作为语义分割模型，利用S1中扩充训练集对所述Mo
‑
SegNet网络模型进行训练得到语义分割权重，利用训练好的Mo
‑
SegNet网络模型对输入的S1中测试集的水上交通图像进行逐像素分割输出语义分割图像；S3：采用Canny检测算法对S2输出的语义分割的图像进行边缘检测，输出仅包含海天线边缘轮廓及背景的二值图像；S4：在二值图像上进行霍夫变换，对海天线边缘轮廓的边缘像素进行拟合，得到海天线检测结果。2.如权利要求1所述的基于语义分割和霍夫变换结合的海天线检测方法，其特征在于，S1中，针对输入所述水上交通场景的可见光图像，将图像由标注软件逐像素进行标注制作数据集，将水上交通图像按照像素点分类为天空区域、海面区域和其他区域，并将所述数据集按8:2的比例分为训练集和测试集，对训练集进行旋转、镜像图像处理对训练集进行扩充，最终得到扩充训练集；所述其他区域包括海面船舶、漂浮物和遮挡物像素。3.如权利要求1所述的基于语义分割和霍夫变换结合的海天线检测方法，其特征在于，S2中，所述Mo
‑
SegNet网络模型由编码网络和解码网络构成；所述编码网络由一次标准卷积和三次深度可分离卷积组成，每一次卷积均增加了一层BN层，对输出的特征图结果进行归一化；通过多层卷积实现对水上交通图像中天空、海面、船舶等像素特征的提取，得到相应的特征层；所述解码网络采用三次重复的补零、卷积、批标准化、激活和上采样操作，将水上交通图像中的每一个像素赋予一个类别标签；在最后一层解码器输出高维特征表示向量，作为可训练Softmax分类器中函数的输入。4.如...

【专利技术属性】
技术研发人员：陈信强，韩冰，吴中岱，陈建慧，段载誉，
申请(专利权)人：上海海事大学，
类型：发明
国别省市：