基于Faster R-CNN的缝隙检测与提取算法

    Crack Detection and Extraction Based on Faster R-CNN

    • 摘要: 为了使快速区域卷积神经网络(faster region-based convolutional neural network,Faster R-CNN)适用于小尺寸结构缝隙目标检测的应用,提出了一种基于Faster R-CNN的缝隙检测与提取算法,保留了小尺寸结构目标的细节信息,并提升了检测准确率.该算法分为缝隙检测和缝隙提取2个阶段.首先,在faster R-CNN的目标检测框架下,选取ImageNet数据集上的视觉几何组(visual geometry group,VGG)网络预训练模型作为特征提取网络,调整网络模型使其适应具有小尺寸结构的缝隙目标,并通过缝隙检测网络的训练确定最优的网络超参数,获得缝隙目标边框.然后,根据对目标区域的分析,提出基于数学形态学算法的缝隙提取算法,将缝隙目标从背景中分割出来.最终通过去噪、断裂连接和细化操作提取单像素宽缝隙目标,通过统计单像素宽缝隙目标的像素点个数得到缝隙目标长度值.实验结果表明,该算法可准确且完整地提取缝隙目标,在铁轨裂缝数据集上平均准确率达到63.87%,在道路裂缝数据集上的F1-score指标达到65.6%.

       

      Abstract: To make the faster region-based convolutional neural network (Faster R-CNN) to fit the detection of crack objects with small-scale structures, a faster R-CNN-based crack detection and extraction method was proposed, which is able to preserve details of small objects and thus improve the performance of the detection. In the crack detection step, the faster R-CNN was considered as the framework of object detection, and the pre-trained visual geometry group model on ImageNet dataset was used as the feature extraction network. Then, the network model was adjusted to fit the small-scale crack object, and the optimal hyper-parameters was set by training to get object bounding-boxes of cracks. In the crack extraction step, according to the analysis of cracks, a crack extraction algorithm was proposed based on mathematical morphology to segment the object from the background, and one-pixel-wide cracks were finally extraced after denoising, linking and thinning. In this case, the length of the crack was be obtained by counting the number of pixels. Experimental results show that the proposed method can extract accurate and complete cracks, the average precision on the railroad crack dataset can achieve 63.87%, and the F1-score on the road crack dataset can achieve 65.6%.

       

    /

    返回文章
    返回