DD-Net: Dynamic Network Architecture for Optimized Curve Segmentation and Reduce Computational Redundancy
摘要
Curvilinear Structure segmentation has numerous applications in various fields, including providing a better understanding of defects such as cracks on roads or walls, thereby assessing the structural safety of buildings. Numerous curve segmentation methods based on Convolutional Neural Networks(CNN) have been developed in recent years. However, preserving feature integrity remains a challenge. To address this issue, this paper introduces a double dynamic network, called DD-Net, which consists of a dynamic structure for training and dynamic masking for inference. Firstly, to enhance the nonlinearity of the model, a paradigm to auto-adjust the CNN structure via a trainable module is formed for better fine-grained feature extraction. This dynamic structure (DS) paradigm enables a trade-off to extract or discard the feature data in the model training. Due to the data heterogeneity between the training and inference data, models may contain some useless nodes that are less effective during inference. The dynamic masking (DM) is proposed to omit the useless nodes based on the score difference between the train and inference feature statistics, thereby reducing redundant computations. To further improve the model’s performance on thin-curve segmentation and to preserve feature integrity, we introduce a non-curve suppression (NCS) module. This module focuses on background information while considering foreground prediction to address noisy conditions. The experimental results show that our DD-Net achieves promising results on three benchmark datasets and outperforms state-of-the-art curve segmentation models.