Computer and Modernization ›› 2022, Vol. 0 ›› Issue (07): 33-39.

Previous Articles     Next Articles

ERCUnet: An Improved Road Crack Detection Model Based on U-Net

  

  1. (1. School of Software Technology, Zhengzhou University, Zhengzhou 450001, China;
    2. Aerospace Information Research Institute, Chinese Academy of Sciences, Beijing 100101, China)
  • Online:2022-07-25 Published:2022-07-25

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.

Key words: road crack detection, U-Net, residual structure, data expansion