Shape-aware speckle matching network for cross-domain 3D reconstruction

Abstract

The learning-based binocular 3D reconstruction method outperforms traditional vision algorithms in terms of precision and efficiency. However, in some cross-domain scenes, the precision of the well-trained network drastically decreases when handling samples in diverse contexts. This paper proposes an end-to-end shape-aware speckle matching network (SSMNet) that combines shape-mask information to achieve improved precision and completeness of disparity calculation in cross-domain applications. The cascade attention mechanism is inserted in the feature extraction stage to concentrate on valuable regions. The shape-aware module is designed to learn additional shape contour information, and multiscale features are integrated simultaneously to construct the cost-volume for the subsequent lightweight 3D aggregation. In addition, instance normalization is adopted to guarantee style migration, and a hybrid loss function is used to supervise the learning process. Furthermore, a high-precision binocular speckle dataset is built, including training and testing sets in different distributions. Extensive quantitative and qualitative experiments demonstrate that SSMNet enhances cross-domain capability and achieves state-of-the-art performance. Measurement precision evaluation illustrates that the proposed method can realize the desired highly precise 3D shape measurement in a real industrial scenario.