+高级检索
网刊加载中。。。

使用Chrome浏览器效果最佳,继续浏览,你可能不会看到最佳的展示效果,

确定继续浏览么?

复制成功,请在其他浏览器进行阅读

一种道路裂缝检测的变尺度VS-UNet模型  PDF

  • 赵志宏 1,2
  • 何朋 2
  • 郝子晔 2
1. 石家庄铁道大学 省部共建交通工程结构力学行为与系统安全国家重点实验室,河北 石家庄 050043; 2. 石家庄铁道大学 信息科学与技术学院,河北 石家庄 050043

中图分类号: TP391.41

最近更新:2024-07-02

DOI: 10.16339/j.cnki.hdxbzkb.2024267

  • 全文
  • 图表
  • 参考文献
  • 作者
  • 出版信息
EN
目录contents

摘要

为解决目前现有的图像分割算法存在检测精度低、对裂缝检测缺乏针对性等问题,采用多尺度特征融合方法,提出一种扩展LG Block模块Extend-LG Block,其由多个并行不同膨胀率的空洞卷积组成.通过参数可调节分支数量和空洞卷积膨胀率,从而改变其感受野大小,进而提取和融合不同尺度的裂缝特征.对比在深层使用多尺度特征融合模块的网络以及使用固定尺度结构进行多尺度特征融合的网络的优劣,提出一种变尺度结构的UNet模型VS-UNet,使用多个不同参数的Extend-LG Block替换UNet网络中的基本卷积块.该结构在网络浅层进行多尺度特征融合,多尺度特征融合模块提取的尺度随网络层加深逐渐减少.此结构在加强图像的细节特征提取能力的同时保持原有的抽象特征提取能力,还可避免网络参数的增加.在DeepCrack数据集以及CFD数据集上进行实验验证,结果表明,相较于其他两种结构和方法,提出的变尺度结构的网络在有更高检测精度的同时,在可视化实验对比上对各种大小的裂缝有更好的分割效果.最后与其他图像分割算法进行对比,各项指标与UNet相比均有一定程度提升,证明了网络改进的有效性.研究结果可为进一步提升道路裂缝检测效果提供参考.

我国道路交通设施的快速发展为人们的出行提供了便利,但同时也出现了很多新的问题.我国道路多为混凝土结构,容易出现裂

1,随着时间的推移裂缝会逐渐加大,带来安全隐2.因此路面裂缝的及时检测是当前亟须解决的问题.

传统路面裂缝检测多采用人工检测的方

3,这种方法对于裂缝的检测会比较全面,但同时会消耗较大的人力与物力.基于阈值的分割算4对裂缝的识别率较高,然而在光照不均匀、背景图不一致的情况下识别率较低.基于边缘检测的分割算5对于边缘明显的裂缝检测效果较好,但是对于边缘信息较弱的裂缝以及背景噪点较多的裂缝检测效果较差.传统的图像分割算法对于复杂情况下的图像分割效果仍然不6.

近年来,基于深度学习的裂缝检测方法逐渐成为主流,取得了很好的效

7.Shelhamer8提出使用FCN(Fully Convolutional Networks)模型完成图像分割任务,在图像分割任务上具有不俗的效果.Yang9基于FCN网络完成了裂缝检测任务,实验结果表明FCN网络可应用于裂缝检测任务.Badrinarayanan10在FCN的基础上提出的SegNet网络使用编码器-解码器的结构,具有良好的分割性能.Ronneberger11提出的UNet网络使用特征拼接的方式将采样前后的特征进行融合,广泛应用在了医疗图像分割领域.

由于混凝土表面具有不平整、光照不均和裂缝背景较为复杂等特点,并且裂缝像素所占比例较

12,因此上述分割网络在裂缝检测任务上精度不佳.祝一帆13使用密集连接结构改进U-Net网络,结合残差块以及扩张卷积提出Crack UNet网络,在裂缝检测任务上具有良好的效果.朱立学14提出使用符合HDC(Hybrid Dilated Convolution)设计原则的多尺度串联空洞卷积组合,有效避免了稀疏的串联空洞组合会出现的网格效应,在香蕉图像分割任务上取得较好的效果.石甜甜15提出一种多尺度融合注意力模块,在特征融合部分增强网络获取不同尺度信息的能力.Chen16提出空洞卷积空间金字塔池化(ASPP)模块,使用不同扩张率的并行空洞卷积捕获多尺度信息.Song17提出一种多尺度的医学图像分割网络PLU-Net,设计一种新的LG Block (Local Guided Block)模块,通过并行的二分支空洞卷积结构来增加提取特征的尺度,实验结果表明,采用LG Block模块提高了分割的精度,增加了网络的性能.但是PLU-Net网络在浅层增加的尺度有限,模型的分割精度还有待提高.本文针对上述裂缝检测模型存在的问题,提出一种改进的道路裂缝检测模型,所做主要贡献有:

1)提出一种多尺度特征提取融合模块Extend-LG Block模块.针对裂缝细节特征提取比较困难, Extend-LG Block模块可以更好地提取裂缝特征.模块通过参数调节分支数量以及空洞卷积的扩张率,采用HDC设计原则,在一定程度上避免了空洞卷积堆叠导致的网格效应,可以提取更丰富的多尺度特征信息.

2)提出一种采用变尺度结构设计的U-Net模型.考虑到深度卷积神经网络浅层提取细节特征信息,深层提取抽象特征信息的特点,将固定尺度结构改进为变尺度结构,具体地,网络浅层用多分支的Extend-LG Block提取多尺度特征,网络深层用少分支的Extend-LG Block提取多尺度特征,在最底层用普通卷积来提取抽象特征.

1 方 法

1.1 空洞卷积

在裂缝检测任务中,往往存在大小不一的各种裂缝,这就要求网络需要从各个尺度提取裂缝中的信息,而大尺度的信息需要大感受野的卷积来提

18.传统卷积一般只能通过增加卷积核的大小或者增加网络的深度来增大感受野,这会增加网络的参数量.YU19提出的空洞卷积结构解决了该问题.空洞卷积如图1所示,空洞卷积在保持参数数量不变的情况下增大卷积核的感受野,使每个卷积输出都包含较大范围的信息;同时可以保证输出的特征映射的大小保持不变.

fig

图1  不同膨胀率的空洞卷积

Fig.1  Dilated convolutions with different dilation rates

(a)膨胀率为1 (b)膨胀率为2 (c)膨胀率为3

传统卷积与空洞卷积的卷积核大小的对应关系如下:

k'=d×(k-1)+1 (1)

空洞卷积感受野的计算公式如下:

F=2d-1×k-1+k (2)

式中:k表示输入的卷积核尺寸;d表示扩张系数;k'表示扩张后等效的卷积核尺寸.假设扩张系数为7,卷积核尺寸为3×3,通过上式(1)~(2)可以计算发现扩张之后等效的卷积核尺寸为15.可以看到通过扩张卷积仅需要与3×3的普通卷积同样的参数量就可以获得与15×15的普通卷积相同的感受野大小.

空洞卷积在有效增加网络感受野的同时,也存在网格效应(the Gridding Effect)问题.网格效应如图2所示,网格效应会导致输入信号当中的一些像素得不到利用,即局部信息会丢失.

fig

图2  网格效应

Fig.2  The Gridding Effect

(a)膨胀率为1 (b)膨胀率为2 (c)膨胀率为3

Wang

20针对空洞卷积存在的网格效应问题,提出HDC设计结构.基于HDC结构的空洞卷积组合可以有效减少网格效应的产生.HDC首先定义两个非零元素之间最大的距离公式,该公式如下所示:

Mi=MaxMi+1-2ri,Mi+1-2Mi+1-ri,ri (3)

式中:Mi表示第i层两个非零元素之间的最大距离;ri表示第i层的膨胀系数.HDC结构要求:1)假设卷积核大小为k×k,则M2k,即第二层的两个非零元素之间的最大距离小于或等于该层卷积核的大小.2)空洞卷积组合的扩张系数应为锯齿形状,如[1,3,5,1,3,5].3)空洞卷积组合的公约数不能大于1.采用上述设计原则的空洞卷积组合可以有效减少网格效应的发生.

1.2 Extend-LG Block

为进一步提升网络对于裂缝特征的提取能力,设计一种多分支卷积结构对UNet模型的基本卷积块进行替换.

UNet网络采用的基本卷积块如图3所示,由两个3×3普通卷积组成.这种结构的卷积块缺乏提取图像不同尺度特征的能力,不能同时很好地提取大裂缝与小裂缝的特征.

fig

图3  UNet网络的基本卷积块

Fig.3  The basic convolutional block of the UNet network

针对UNet基本卷积块存在的问题,LG Block使用两个并行的不同扩张率的空洞卷积,其结构如图4所示,通过不同扩张率的空洞卷积提取不同尺度的特征.与UNet网络的基本卷积块相比,这种结构的卷积块拥有提取多尺度特征的能力,并且与其他采用串行结构的空洞卷积块相比,由于其设计符合HDC结构,可以很好地避免网格效应问题.

fig

图4  LG Block结构

Fig.4  LG Block structure

LG Block由两个分支组成,每个分支有一个3×3的空洞卷积,其扩张率分别为1和3,然后将两个扩张卷积操作的结果连接起来,以增强特征传播.然后,采用1×1卷积运算,在不改变特征图大小的情况下,加入非线性特征,实现各个不同尺度特征的融合.

LG Block仅增加一个膨胀率为3的空洞卷积,虽然扩大了感受野,但增加的尺度有限,在浅层网络中提取到的特征仍不够丰富,故需要在浅层增加提取的尺度.而受制于UNet网络中特征图尺寸大小随着池化层减少的特点,在网络深层无须增加大感受野卷积来提取大尺度信息.因此网络需要多种不同感受野的卷积块调节网络不同层的提取特征的尺度.针对上述问题,对LG Block进行扩展,扩展之后的模块Extend-LG Block的结构如图5所示.

fig

图5  Extend-LG Block结构

Fig.5  Extend-LG Block structure

扩展之后的Extend-LG Block与原有固定的两分支不同,可通过参数增减分支,网络增加的扩张卷积的膨胀系数为2n-1.特别地,LG Block是n=2时的Extend-LG Block. Extend-LG Block对于输入图像的处理过程与LG-Block类似,可描述为:

F1=σ(BN(f3×31(I)))F2=σ(BN(f3×33(I)))Fn=σ(BN(f3×32n-1(I)))Fo=f1×1(Cat(F1,F2,,Fn)) (4)

式中:I表示输入的图像;Fn表示经过扩张率为n的空洞卷积提取到的特征;fk×kr(·)表示扩张率为r的,卷积核大小为k的空洞卷积操作;σ(·)表示ReLu激活函数;Cat(·)表示特征拼接操作;BN(·)表示归一化操作;f1×1(·)表示一个卷积核大小为1的普通卷积操作;Fo表示Extend-LG Block模块输出的特征.

为防止使用多个扩张卷积导致网络出现网格效应,Extend-LG Block的组合采用HDC设计结构.网络的扩张系数设计为r=[1,3,5,…,2n-1],符合锯齿形状的扩张系数的要求,其次,扩张系数的最大公约数不大于1.需证明网络的M2≤3,证明过程如下:

根据定义可知

Mn=rn=2n-1Mn-1=MaxMn-2rn,Mn-2Mn-rn,rn=          Max5-2n,2n-5,2n-3

n≤2时,即r=[1,3]或者r=[1]满足设计要求.

所以Mn-1=2n-3.

以下讨论n≥3的情况.

Mn-2=MaxMn-1-2rn-2,Mn-1-          2Mn-1-rn-2,rn-2=         Max7-2n,2n-7,2n-5

又因为n≥3,所以Mn-2=2n-5.

由上述可知:Mn-i=2n-2i-1.

i=n-2时,M2=2n-2i-1.

M2=3k=3.

故扩张系数组合符合HDC设计结构.

为验证不同取值的Extend-LG Block提取的效果,将n=2、3、4时Extend-LG Block提取的特征图进行对比,不同n取值的Extend-LG Block提取的特征图如图6所示.

fig

图6  不同n取值的Extend-LG Block提取特征图对比

Fig.6  Comparison of feature maps extracted from Extend-LG Block with different n values

通过不同n取值的Extend-LG Block提取的特征图对比,可以发现n取值越大,Extend-LG Block对于裂缝的边缘信息提取效果越好,同时对于一些背景噪声的过滤效果也得到加强.综上所述,扩展之后的结构相较于LG Block能够提取更多尺度特征,对于各种裂缝的分割均具有较好的效果.

1.3 VS-UNet

UNet起初应用于医疗图像分割领域.近几年有学者将其引入裂缝检测领域.UNet结构如图7所示,其采用下采样(编码器),上采样(解码器),跳跃连接的设计方法.浅层可以提取图像细节特征,如边界、颜色、纹理等,深层会提取图像抽象特

21.UNet采用跳跃连接的方法在特征传播阶段进行多尺度特征融合,缺乏在特征提取阶段进行多尺度特征融合的能力.

fig

图7  UNet结构图

Fig.7  Diagram of the UNet structure

为了进一步提高UNet网络的多尺度特征提取能力,李国燕

21在网络的编码器与解码器之间加入并行多尺度特征融合模块,如ASPP模块,增强网络多尺度特征提取能力,其结构如图8所示.在编码器与解码器之间加入多尺度特征融合模块虽然可以增强网络对于不同大小裂缝的提取能力,但是其放置位置在网络的深层中,仅针对网络提取的抽象特征进行多尺度特征融合,浅层提取的细节信息经过最大池化层会有一定损失,使得这种模块不能很好地融合网络浅层的特征.

fig

图8  加入多尺度特征融合模块的网络

Fig.8  Network with multi-scale feature fusion module

越来越多的研究者使用多尺度特征融合模块作为网络的基本网络单元替换UNet网络卷积块的方法,来提升网络多尺度特征提取的能力.Song

17使用固定尺度结构提取特征,如图9所示.该结构在网络的每层都使用相同大小和扩张率的扩张卷积块提取特征,这会同时保持网络对于浅层特征以及深层特征的多尺度特征提取能力, 并且由于UNet下采样机制,深层网络拥有较大的感受野可提取大尺度信息,但是会导致网络过于关注全局信息,从而影响小裂缝的分割效果,除此之外该结构由于增加卷积数量会导致网络参数量增加.

fig

图9  采用固定尺度结构设计的编码器与解码器

Fig. 9  The encoder and decoder designed with fixed scale structure

本文在使用多尺度特征融合模块作为网络的基本卷积块替换原有网络的基本卷积块的基础上,将固定尺度的多尺度特征融合模块替换为可变尺度的多尺度特征融合模块,进而提出一种变尺度的多尺度特征融合结构,如图10所示.具体表现为在浅层使用大尺度卷积块提取细节特征,在深层使用小尺度卷积提取抽象特征.这种变尺度结构可以提升网络对于细节特征的提取能力,保持网络对于抽象特征的提取能力,在避免网络参数量增加的情况下,仍保持有较高的精度.除此之外,同时兼顾提取到的全局信息与局部信息,避免网络偏向全局信息而影响小裂缝的检测.

fig

图10  采用变尺度结构设计的编码器与解码器

Fig.10  The encoder and decoder designed with variable scale structure

本文设计一种采用变尺度结构的多尺度裂缝检测网络VS-UNet(Variable Scale-UNet),如图11所示,由n=4,3,2,1的Extend-LG Block组合而成,n根据下采样的深度依次递减.在浅层使用较大的数值n可以提取到更多尺度的细节特征,在深层使用较小数值的n可以避免网络尺度过大导致的对细小裂缝分割能力减弱的问题,还会减少网络的参数量增加.值得注意的是为保持深层网络提取抽象特征的能力,在最低层保持了UNet中的两个3×3的普通卷积.上采样的过程与下采样相反,Extend-LG Block依据网络的深度n依次增加.

fig

图11  VS-UNet网络结构模型图

Fig.11  Diagram of the VS-UNet network structure model

使用上述的结构,网络一定程度上增加了每层的宽度,相较于UNet层数没有增加,还可以避免因层数增加而导致的梯度消失问题.

VS-UNet网络主要分为编码器与解码器两个部分.分别为4次下采样与4次上采样.每次下采样前特征图会经过不同参数n的Extend-LG Block模块,此时模块会进一步提取出n个尺度的特征进行融合,此外会使特征图的通道数翻倍.使用最大池化层对模型进行下采样,采样之后特征图的大小会缩小到原来的1/2.解码器部分与编码器部分相反,特征图先由转置卷积进行上采样,之后与下采样提取到的相同大小特征图进行拼接操作,融合浅层提取到的特征,再由Extend-LG Block模块进一步进行特征提取.最后由一个1×1的卷积对像素进行预测.

由于网络输出的结果为单个像素的概率预测,故需要对网络预测结果进行后处理.本文使用阈值来进行分割结果的二值

23,即通过设定的阈值来划分目标对象(白色)和背景对象(黑色),从而得到最终的分割图像.

2 实验与分析

2.1 数据集与参数设置

为验证改进的有效性,实验采用DeepCrack数据

24以及CFD(Crack Forest Dataset)数据25.其中,DeepCrack数据集包含573张大小为544×384的混凝土路面图像,CFD数据集包含118张分辨率为480×320的已标注裂缝图片.实验中以8∶2的比例划分训练集与测试集,同时在训练集中再以8∶2的比例划分出验证集.考虑到CFD数据集数据量较少,使用翻转、镜像等数据增强技术进行扩充.

实验环境使用Windows11操作系统,运行内存16GB,处理器为Intel Core I5 8400,GPU为NVIDIA GTX1060,编程语言为Python3.8,模型基于Pytorch1.13.1深度学习框架实现.

训练过程中选用Adam优化器作为模型的优化器,损失函数为均方误差损失函数,BatchSize大小设置为2.学习率设置为3×10-3.训练的轮数为50轮.为避免网络陷入局部最优解,实验采用多组随机初始化参数.使用阈值0.5对预测结果进行二值化处理.

2.2 评价指标

实验采用查准率P(Precision)、查全率R(Recall)、F1F1_Score)作为评价指标.P是正确检测的裂缝像素与原本正确像素的比值.R表示正确检测的裂缝像素占所有裂缝像素的百分比.F1F1_Score)表示同时衡量PR的值,尽可能保证在裂缝像素遗漏少的前提下误判也少.以上几个评价指标均与模型的性能成正比.上述指标计算公式如下所示:

P=TPTP+FP (5)
R=TPTP+FN (6)
F1=2×P×RP+R (7)

式中:TP(True Positive)表示被正确分为裂缝像素的数量;FP(False Positive)为错误分为裂缝像素的数量;FN(False Negative)表示目标被错误分割为背景的像素数量.

2.3 Extend-LG Block不同放置位置实验

为验证Extend-LG Block放置位置对网络的影响,分别设计仅放置在解码器部分的网络模型、仅放置在编码器部分的网络模型以及同时放置在编码器与解码器部分的网络模型,实验结果如表1所示.

表1  Extend-LG Block放置位置对比结果
Tab.1  Comparison results of Extend-LG Block placement positions
数据集下采样上采样P/%R/%F1/%
DeepCrack 85.12 74.18 79.27
78.51 79.93 79.21
88.06 78.02 82.74
CFD 73.28 59.09 65.43
77.99 58.36 66.76
82.71 64.83 72.68

在DeepCrack数据集中,放置在上采样中的P值比放置在下采样过程中的P值低6.61%,而放置在上采样中的R值比放置在下采样过程中的R值高5.75%.在F1指标对比上,下采样比上采样高0.06%.在CFD数据集中,放置在上采样中的P值比放置在下采样过程中的P值高4.71%,而放置在上采样中的R值比放置在下采样过程中的R值低0.73%.在F1指标的对比上,上采样比下采样高出1.33%.

根据上述分析可知同时放置在下采样与上采样过程中的F1指标在两个数据集上均优于放置在单一位置上的F1指标,故同时放置的效果比任意单一放置的效果更好.

2.4 Extend-LG Block不同参数取值对网络性能的影响

为验证使用不同尺度的Extend-LG Block对网络的影响,分别对只采用n=4、3、2的Extend-LG Block的U-Net网络以及VS-UNet进行实验,实验结果如表2所示.

表 2  使用不同参数的Extend-LG Block实验对比结果
Tab.2  Comparison results of Extend-LG Block experiments with different parameters
数据集方法nP/%R/%F1/%Parameters/M
DeepCrack Extend-LG Block 4 88.58 79.19 83.62 51.03
Extend-LG Block 3 87.95 80.87 84.26 42.51
Extend-LG Block 2 87.61 77.38 82.18 33.99
VS-UNet 88.06 78.02 82.74 28.13
CFD Extend-LG Block 4 78.14 63.81 70.25 51.03
Extend-LG Block 3 78.29 65.25 71.18 42.51
Extend-LG Block 2 81.86 61.60 70.30 33.99
VS-UNet 82.71 64.83 72.68 28.13

从表中可以看到,在DeepCrack数据集中,n=3时Extend-LG Block的F1指标与R指标最高,F1指标比n=4时高出0.64%,参数量低8.52M,F1指标比n=2时高出2.08%,比VS-UNet高出1.52%.通过n=4与 n=3的对比实验发现,提升n的取值并不一定会取得更好的效果,反而会增加参数量.VS-UNet相比于n=2的Extend-LG Block,在F1指标上高出0.56%,虽然在F1指标上略低于n=4,n=3的情况,但是参数量明显低于其他几种方法.在CFD数据集中可以发现,VS-UNet的F1指标比n=4的Extend-LG Block高出2.43%的Extend-LG Block,比n=3高出1.5%,比n= 2高出2.43%.注意到Extend-LG Block n=3模型在DeepCrack上优于VS-UNet而VS-UNet在CFD数据集上优于Extend-LG Block,这是因为CFD数据集多为细小裂缝,深层网络感受野的过度增加使得网络过度关注于全局特征,从而对小裂缝的分割效果产生了负作用.

上述结果表明,增加Extend-LG Block的尺度可以提升网络的特征提取效果,但过多增加尺度会使得网络偏向全局特征,降低了网络的分割效果并增加网络参数量,故网络的整体尺度不应过大.同时,与采用固定尺度的Extend-LG Block的网络相比,变尺度结构的网络在有更小参数量的同时,对于细小裂缝数据集分割效果更好.

2.5 采用不同多尺度特征提取融合方法的网络对比

为验证本文提出的变尺度结构的有效性,本节分别设计了两个网络与VS-UNet进行对比.ASPP-UNet为在编码器与解码器之间加入并行多尺度特征融合模块的UNet网络,将其编码器与解码器之间的普通卷积块替换为ASPP模块.FS-UNet(Fixed scale-UNet)为采用固定尺度结构的UNet网络,将其基本卷积块全部替换为LG Block.

采用不同多尺度特征提取方法的网络对比结果如表3所示,在P指标和R指标的对比上,3种多尺度网络结构在不同数据集上各有优劣,而在综合P指标与R指标的F1指标的对比上,本文提出的VS-UNet在两个数据集中均取得三者之中最高值.在DeepCrack数据集中,VS-UNet的F1指标比ASPP-UNet与FS-UNet分别高出3.68%,2.34%.在CFD数据集中,VS-UNet的F1指标比ASPP-UNet与FS-UNet分别高出1.20%,3.51%.在参数的对比上,VS-UNet比ASPP-UNet高1.02 M,比FS-UNet低3.24 M.

表3  采用不同多尺度特征提取方法的网络对比结果
Tab.3  Comparison results using different Multi-Scale feature extraction methods
数据集模型P/%R/%F1/%Parameters/M
DeepCrack ASPP-UNet 90.06 70.45 79.06 27.11
FS-UNet 91.37 71.78 80.40 31.37
VS-UNet 88.06 78.02 82.74 28.13
CFD ASPP-UNet 78.02 65.95 71.48 27.11
FS-UNet 80.72 60.51 69.17 31.37
VS-UNet 82.71 64.83 72.68 28.13

本文提出的变尺度结构在裂缝检测的效果上优于其他两种方法,在参数的对比上接近使用ASPP模块的UNet网络,优于使用固定尺度结构的UNet网络.实验结果表明了本文提出的变尺度结构的有效性.

2.6 可视化分析

网络分割结果的可视化是验证改进算法有效性的途径之一,本节对UNet、ASPP-UNet、FS-UNet以及VS-UNet分别进行实验,将实验结果进行可视化对比.

在DeepCrack数据集以及CFD数据集上的可视化对比结果如图12所示,通过与其他算法对比可以发现,采用多尺度特征提取方法设计的UNet网络在裂缝识别的完整性上均优于UNet网络.在第一列中,ASPP-UNet与VS-UNet均完整分割出裂缝.在第二列中,ASPP-UNet表现了较高的误检率,FS-UNet在裂缝边缘的分割上表现较差.在CFD数据集中,VS-UNet对于细小裂缝分割效果优于其他两种方法.

fig

图12  可视化分析对比结果

Fig. 12  Visual analysis and comparison results

通过上述分析可知,VS-UNet对于大小裂缝均具有较好的分割效果,对于裂缝分割的完整性上优于其他几种方法.

2.7 与其他网络模型对比

为进一步验证本文网络的优越性,与SegNet网络、DeepCrack网

23、RCF网26、UNet网络进行对比,实验结果如表4所示.在DeepCrack数据集中,本文提出的VS-UNet相较于上述几个网络,在F1指标上分别提高5.7%、28.2%、17.5%、15.4%. 在CFD数据集的对比之中,VS-UNet相较于上述几个网络在F1指标上分别提高23.2%、19.7%、27.5%、17.8%.

表4  与其他网络模型对比
Tab.4  Comparison with other network models
数据集模型P/%R/%F1/%
DeepCrack SegNet6 73.2 81.2 77.0
DeepCrack23 53.5 55.5 54.5
RCF26 60.1 71.3 65.2
UNet6 64.2 70.6 67.3
VS-UNet 88.0 78.0 82.7
CFD SegNet6 42.0 60.2 49.5
DeepCrack23 46.7 61.5 53.0
RCF26 41.5 49.5 45.2
UNet6 61.7 54.9 54.9
VS-UNet 82.7 64.8 72.7

根据上述对比可以证明本文提出VS-UNet网络在裂缝检测上均优于其他几个同类型网络.

3 结 论

本文针对目前图像分割网络存在的问题,对UNet网络结构进行改进,得出以下的结论:

1)本文首先提出一种扩展的LG-Block模块,使用多分支结构代替原有的二分支结构,增大模型的感受野,增加特征提取的尺度.其次提出一种变尺度结构的U-Net模型,该模型由多个增强后的LG-Block模块组成,相较于U-Net拥有更大的感受野,对裂缝的分割效果更好.

2)采用变尺度结构可有效地提取裂缝特征.在可视化对比实验中,VS-UNet可有效地分割出大小不一的裂缝.与其他几个网络模型进行对比,结果表明VS-UNet拥有更高的F1-Score,其在DeepCrack数据集以及CFD数据集上分别较UNet提升15.4%,证明其对于裂缝的分割效果更好.

目前注意力机制使用相对广泛,可以提高网络的分割效果,可以作为网络改进的下一步研究方向.

参考文献

1

蒋文波罗秋容张晓华基于数字图像的混凝土道路裂缝检测方法综述[J].西华大学学报(自然科学版)2018371):75-84 [百度学术] 

JIANG W BLUO Q RZHANG X HA review of concrete roads crack detection methods based on digital image[J].Journal of Xihua University (Natural Science Edition)2018371):75-84(in Chinese) [百度学术] 

2

史梦圆高俊钗改进U-Net算法的路面裂缝检测研究[J].自动化与仪表20223710):52-55 [百度学术] 

SHI M YGAO J CResearch on pavement crack detection based on improved U-Net algorithm[J]. Automation & Instrumentation20223710):52-55(in Chinese) [百度学术] 

3

邓露褚鸿鹄龙砺芝基于深度学习的土木基础设施裂缝检测综述[J].中国公路学报2023362):1-21 [百度学术] 

DENG LCHU H HLONG L Zet alReview of deep learning-based crack detection for civil infrastructures[J].China Journal of Highway and Transport2023362):1-21(in Chinese) [百度学术] 

4

KIRSCHKE K RVELINSKY S A. Histogram-based approach for automated pavement-crack sensing[J]. Journal of Transportation Engineering19921185):700-710 [百度学术] 

5

张爱华余胜生周敬利一种基于边缘检测的局部阈值分割算法[J].小型微型计算机系统2003244):661-663 [百度学术] 

ZHANG A HYU S SZHOU J LA local-threshold segment algorithm based on edge-detection[J]. Mini-micro Systems2003244):661-663(in Chinese) [百度学术] 

6

ZHANG L XSHEN J KZHU B J. A research on an improved Unet-based concrete crack detection algorithm[J].Structural Health Monitoring2021204):1864-1879 [百度学术] 

7

黄鹏郑淇梁超图像分割方法综述[J].武汉大学学报(理学版)2020666):519-531 [百度学术] 

HUANG PZHENG QLIANG COverview of image segmentation methods[J].Journal of Wuhan University (Natural Science Edition)2020666):519-531(in Chinese) [百度学术] 

8

SHELHAMER ELONG JDARRELL TFully convolutional networks for semantic segmentation[C]//IEEE Transactions on Pattern Analysis and Machine IntelligenceIEEE2017640-651 [百度学术] 

9

YANG X CLI HYU Y Tet alAutomatic pixel‐level crack detection and measurement using fully convolutional network[J].Computer-Aided Civil and Infrastructure Engineering20183312):1090-1109 [百度学术] 

10

BADRINARAYANAN VKENDALL ACIPOLLA RSegNet:a deep convolutional encoder-decoder architecture for image segmentation[J].IEEE Transactions on Pattern Analysis and Machine Intelligence20173912):2481-2495 [百度学术] 

11

RONNEBERGER OFISCHER PBROX T. U-Net:convolutional networks for biomedical image segmentation[M]//Lecture Notes in Computer ScienceChamSpringer International Publishing2015234-241 [百度学术] 

12

甘霖谢爱荣燕阳基于改进U-Net网络的混凝土表面裂缝分割[J].重庆邮电大学学报(自然科学版)2021334):645-652 [百度学术] 

GAN LXIE A RYAN Yet alCrack segmentation of concrete surface based on improved U-Net[J].Journal of Chongqing University of Posts and Telecommunications (Natural Science Edition)2021334):645-652(in Chinese) [百度学术] 

13

祝一帆王海涛李可一种高精度路面裂缝检测网络结构:Crack U-Net[J].计算机科学2022491):204-211 [百度学术] 

ZHU Y FWANG H TLI Ket alCrack U-Net:towards high quality pavement crack detection[J].Computer Science2022491):204-211(in Chinese) [百度学术] 

14

朱立学伍荣达付根平基于多尺度串联空洞卷积的轻量化UNet香蕉图像分割[J].农业工程学报20223813):194-201 [百度学术] 

ZHU L XWU R DFU G Pet al. Segmenting banana images using the lightweight UNet of multi-scale serial dilated convolution[J]. Transactions of the Chinese Society of Agricultural Engineering20223813):194-201(in Chinese) [百度学术] 

15

石甜甜郭中华闫翔基于多尺度融合注意力改进UNet的遥感图像水体分割[J].液晶与显示2023383):397-408 [百度学术] 

SHI T TGUO Z HYAN Xet alWater body segmentation in remote sensing images based on multi-scale fusion attention module improved UNet[J].Chinese Journal of Liquid Crystals and Displays2023383):397-408(in Chinese) [百度学术] 

16

CHEN L CPAPANDREOU GSCHROFF Fet al. Rethinking atrous convolution for semantic image segmentation[J]. arXiv prepriut arXiv:1706.055872017. [百度学术] 

17

SONG W. PLU-Net: Extraction of multi-scale feature fusion[J]. 2023. [百度学术] 

18

ARAUJO ANORRIS WSIM JComputing receptive fields of convolutional neural networks[J].Distill2019411):e21. [百度学术] 

19

VAN HASSELT HGUEZ ASILVER D. Deep reinforcement learning with double Q-learning[J].Proceedings of the AAAI Conference on Artificial Intelligence2016301):129-144 [百度学术] 

20

WANG P QCHEN P FYUAN Yet alUnderstanding convolution for semantic segmentation[C]//2018 IEEE Winter Conference on Applications of Computer Vision (WACV). Lake Tahoe,NV,USAIEEE20181451-1460 [百度学术] 

21

李国燕梁家栋刘毅MFC-DeepLabV3+:一种多特征级联融合裂缝缺陷检测网络模型[J].铁道科学与工程学报2023204):1370-1381 [百度学术] 

LI G YLIANG J DLIU Yet al. MFC-DeepLabV3+: a multi feature cascade fusion crack defect detection network model[J].Journal of Railway Science and Engineering2023204):1370-1381(in Chinese) [百度学术] 

22

许薛军张肖宁基于数字图像的混凝土桥梁裂缝检测技术[J].湖南大学学报(自然科学版)2013407):34-40 [百度学术] 

XU X JZHANG X NCrack detection of concrete bridges based digital image[J].Journal of Hunan University (Natural Sciences)2013407):34-40(in Chinese) [百度学术] 

23

LIU Y HYAO JLU X Het alDeepCrack:a deep hierarchical feature learning architecture for crack segmentation[J].Neurocomputing201933803):139-153 [百度学术] 

24

SHI YCUI L MQI Z Qet alAutomatic road crack detection using random structured forests[J].IEEE Transactions on Intelligent Transportation Systems20161712):3434-3445 [百度学术] 

25

LIU YCHENG M MHU X Wet alRicher convolutional features for edge detection[C]//2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR)Honolulu,HI,USAIEEE20175872-5881 [百度学术] 

26

JING PYU H YHUA Z Het alRoad crack detection using deep neural network based on attention mechanism and residual structure[J].IEEE Access202211919-929 [百度学术] 

作者稿件一经被我刊录用,如无特别声明,即视作同意授予我刊论文整体的全部复制传播的权利,包括但不限于复制权、发行权、信息网络传播权、广播权、表演权、翻译权、汇编权、改编权等著作使用权转让给我刊,我刊有权根据工作需要,允许合作的数据库、新媒体平台及其他数字平台进行数字传播和国际传播等。特此声明。
关闭