Deep learning algorithms have achieved promising results in the field of pavement damage segmentation. Since the texture features of map crack damage are complex, existing segmentation methods may experience the problem of edge information loss for the damaged area. In this study, we therefore propose Road-Seg-CapsNet, a feature fusion model combined with StyleGAN, to segment multiple complex forms of damage to asphalt pavements. Firstly, a data enhancement algorithm based on StyleGAN is developed to amplify the dataset. Next, padding convolution is used in the convolution layer of Road-Seg-CapsNet to retain more image edge information, which effectively solves the problem of information loss at the edges of areas of damage with map cracks. The feature map in the upsampling layer is fused with the features of the convolutional layer to improve the segmentation accuracy of the model. An optimised dynamic routing algorithm is then applied, which reduces the model parameters and improves the efficiency of model training. Our experimental results show that the segmentation mAP of the proposed model can reach 0.942. To verify the superiority of the proposed method, a comparative experiment with FCN, Mask R-CNN and optimised Mask R-CNN is carried out. Based on the segmentation results, a quantification and measurement experiment is carried out, and the minimum accuracy for the damaged areas is found to be 0.903.
Read full abstract