- · 《湖北农机化》栏目设置[06/29]
- · 《湖北农机化》投稿方式[06/29]
- · 《湖北农机化》征稿要求[06/29]
- · 《湖北农机化》刊物宗旨[06/29]
基于改进空洞卷积神经网络的丘陵山区田间道路(5)
作者:网站采编关键词:
摘要:本文参考轩永仓[10]和Bengio等[28]的思路,采用预训练初始化参数的同时,对模型采用了两阶段训练(two-stage training)[31]的方法。具体步骤是: 1)使用Ima
本文参考轩永仓[10]和Bengio等[28]的思路,采用预训练初始化参数的同时,对模型采用了两阶段训练(two-stage training)[31]的方法。具体步骤是:
1)使用ImageNet上训练好的VGG-16模型参数初始化需要训练的DCNN网络。
2)人工选取一些特征明显、包含对象类别少的简单图像 500张先单独对模型进行训练。经过多次试验,确定权值参数的学习率为 10-4,mini-batch size为 14,momentum为0.9,weight decay为0.000 5,采用随机梯度下降法进行训练,等到模型收敛之后,将模型参数保存。由于图像简单,模型收敛速度快。
3)在全部训练集上对上一步保存的模型进行再次训练,利用上一步中所得到的参数来初始化模型,减小学习率为10-5,通过训练更新所有的网络权值和参数。
4 田间道路图像语义分割试验
4.1 试验设计
为验证构建的基于空洞卷积神经网络的丘陵山区田间道路图像语义分割模型的效果,对实际道路场景的语义分割进行了测试。
1)模型改进效果测试
首先测试前端模块,只采用本文构建的前端模块(下文简称为Front-end)进行训练和测试,验证对于VGG-16的修改是否有效。
然后测试上下文模块与前端模块的组合,分别将构建的Basic和Large 2种结构的上下文模块插入前端模块(下文分别简称为Front-end + Basic和Front-end + Large),设置学习率为10-5,迭代次数4 000,对上下文模块进行初始化。由于上下文网络的感受野为67×67,因此使用宽度为33的缓冲区来填充输入特征图。
网络模型都采用两阶段训练方法进行训练。在各网络模型的测试过程中,图像的读取操作通过调用 Python第三方库函数实现。
2)不同模型对比试验
测试文献[10]中效果最好FCN-8s模型,用以对比本文构建的语义分割模型的预测精度。采用CAFFE提供的基于VGG-16的FCN-8s模型进行训练,训练策略与参数设置和前端模块相同。
3)阴影道路对比测试
针对田间道路上阴影突出的问题,测试 Front-end +Large网络模型对阴影道路的语义分割效果。在测试集中选取了100张道路被阴影明显覆盖的图片、100张路面无阴影的图片,分别应用Front-end + Large进行语义分割测试,对比2种情况下的语义分割效果。
4)导航线提取精度测试
田间道路图像语义分割的主要应用之一就是提取道路上的导航线。采用文献[1]和文献[14]中对二值化道路图像提取导航线的方法对本文语义分割后的图像提取导航线,具体方法是:1)采用Front-end + Large模型对田间道路图像进行语义分割;2)语义分割识别出道路区域后,对道路区域分块求解形心点;3)对这些形心点采用最小二乘法进行拟合得到道路的虚拟中线,此虚拟中线即为智能农机在田间道路上自主行驶的局部路径导航线。为验证本文构建的语义分割模型对田间道路识别的实际效果,测试了此虚拟中线与田间道路实际中线的偏差。偏差求取方法是:将获取的虚拟中线通过相机标定将其像素坐标转换为实际平面坐标,沿纵向方向等距离取点计算其横向坐标值与实测的道路中线上对应点横向坐标值的相对偏差。选取与文献[1]和文献[14]所选道路类似的普通水泥路、小障碍物遮挡道路以及杂草、水渍等覆盖道路等3种田间道路进行测试。
4.2 语义分割评判指标
将网络模型分割的结果与人工标注图像(采用Adobe Photoshop CC 2018)进行比较以分析各网络模型语义分割的精度。把人工标注结果视为真实图像分割结果,以此为标准,通过统计各网络模型语义分割像素误差来评判模型分割效果。使用目前普遍采用的评价标准进行评判[32]。
假设njm表示属于j类语义而被识别为m类的像素点个数;N表示语义类别总数,本文N=11,评判指标定义如下:
1)类别像素准确率wj:属于j类语义且被正确分割为j类的第j类像素准确率为
2)统计像素准确率PA(pixel accuracy):标记正确的像素占总像素的比例,计算式为
3)类别平均准确率MPA(mean pixel accuracy):计算每个类被正确分类像素数的比例,然后求取所有类的平均值,计算式为
4)平均区域重合度 MIoU(mean intersection over union):预测像素正确的交集除以预测像素与原来像素的并集。具体计算过程为在每个类上计算区域重合度,然后求取所有类的平均值,计算式为
以上度量标准中,像素准确率反映了在图像中具体类别被正确分割的概率,而MIoU由于简洁、代表性强而成为最常用的度量标准。
文章来源:《湖北农机化》 网址: http://www.hbnjhzz.cn/qikandaodu/2021/0709/1515.html