首页 | 本学科首页   官方微博 | 高级检索  
     

ERCUnet:一种基于U-Net改进的道路裂缝检测模型
引用本文:刘宇翔,佘维,沈占峰,谭帅. ERCUnet:一种基于U-Net改进的道路裂缝检测模型[J]. 计算机与现代化, 2022, 0(7): 33-39
作者姓名:刘宇翔  佘维  沈占峰  谭帅
基金项目:郑州大学大学生创新创业训练计划资助项目(202110459167); 国家自然科学基金资助项目(41971375); 国家重点研发计划项目(2018YFB0505000)
摘    要:针对传统的道路裂缝检测方法存在灵活度不高、普适性不强等问题,本文参考ResNet中的残差设计和U-Net模型的U形编码解码结构,设计一种基于U-Net改进的道路裂缝检测模型——ERCUnet。该模型以残差块为主体,针对裂缝检测优化不同深度卷积层的卷积核数量,模型中所有的残差块结构相同,模型整体结构更加整齐和简单,具有可塑性好、结构性强,残差结构不仅使特征融合更加充分,也避免了深层卷积神经网络梯度消失的问题。实验在CrackForest数据集上进行,将CrackForest的118张含标注图片按照5〖DK〗∶1的比例划分训练集和测试集,通过一系列数据增广方法,有效缓解了训练数据过少的问题。损失函数融合了交叉熵和F1分数,缓解了正负样本不均衡的问题,最终的实验结果显示ERCUnet模型参数量仅为U-Net(BN)模型的13.30%,在测试集上的查全率、查准率、F1值均达70%以上,噪声率、准确率分别为29.05%、99.01%。为证实ERCUnet的可塑性,通过修改模型参数得到ERCUnet-tiny模型,其参数量仅为U-Net(BN)模型的2.39%,在测试集上取得了与U-Net(BN)相近的效果。

关 键 词:道路裂缝检测   U-Net   残差结构   数据增广  
收稿时间:2022-07-25

ERCUnet: An Improved Road Crack Detection Model Based on U-Net
Abstract:Aiming at the problems of traditional road  crack  detection methods,  such as low flexibility and poor universality, refering to the residual design in ResNet and the U-shaped encoding and decoding structure of U-Net model, an improved road crack detection model based on U-Net, named ERCUnet,  is designed. The model takes residual blocks as the main body, and optimizes the number of convolution cores of convolution layers at different depths for crack detection. All residual blocks in the model have the same structure, the overall structure of the model is more neat and simple, with good elasticity and strong structure. The residual structure not only makes the feature fusion more sufficient but also avoids the problem of gradient disappearance of deep convolution neural network. The experiment is conducted on the CrackForest dataset. The 118 labeled pictures of CrackForest are divided into training set and testing set according to the ratio of 5〖DK(〗∶〖DK)〗1. Through a series of data expansion methods, the problem of too little training data is effectively alleviated. The loss function combines cross entropy and F1 score to alleviate the imbalance between positive and negative samples. The final experimental results show that the number of parameters of ERCUnet model is only 13.30% of that of U-Net model, the recall, precision, and F1 are all greater than 70%, and noise rate and accuracy are 29.05% and 99.01% on testing set. ERCUnet-tiny model is obtained by modifying model parameters to confirm the plasticity of ERCUnet, and the number of its parameters is only 2.39% of that of U-Net model, similar effect to U-Net is achieved  on testing set.
Keywords:road crack detection   U-Net   residual structure   data expansion  
点击此处可从《计算机与现代化》浏览原始摘要信息
点击此处可从《计算机与现代化》下载全文
设为首页 | 免责声明 | 关于勤云 | 加入收藏

Copyright©北京勤云科技发展有限公司  京ICP备09084417号