Abstract:In recent years, convolutional-neural-network based stereo matching methods have achieved significant gains compared to conventional methods in terms of both speed and accuracy. Current state-of-the-art disparity estimation algorithms require many parameters and large amounts of computational resources and are not suited for applications on edge devices. In this paper, we propose an end-to-end light-weight network (LWNet) for fast stereo matching, which consists of an efficient backbone with multi-scale feature fusion for feature extraction, a 3D U-Net aggregation architecture for disparity computation, and color guidance in a 2D convolutional neural network (CNN) for disparity refinement. We adopt MobileNetV2 as an efficient backbone in feature extraction. The channel attention module is applied to improve the representational capacity of features and multi-resolution information is adaptively incorporated into the cost volume via cross-scale connections. In addition, instead of using regular 3D convolutions, we utilize pseudo 3D convolutions in the 3D U-Net architecture to aggregate the cost volume for a better balance between computational cost and accuracy. Further, we introduce a left-right consistency check and color guidance and design a robust disparity refinement network with skip connections and dilated convolutions to capture global context information and further improve disparity-estimation accuracy with little computational cost and memory space. A depth-wise separable convolution is proposed to replace all the standard convolutions in the section of disparity refinement, which can decrease computational complexity and the number of parameters without significant accuracy reduction. Extensive experiments on Scene Flow, KITTI 2015, and KITTI 2012 benchmarks demonstrate that the proposed LWNet achieves competitive accuracy when compared with state-of-the-art stereo matching methods.

A Multi-Correlation Stereo Matching Network for Disparity Estimation

Disparity Estimation Using Multilevel and Global Information

MSDC-Net: Multi-Scale Dense and Contextual Networks for Stereo Matching.

MSDC-Net: Multi-Scale Dense and Contextual Networks for Automated Disparity Map for Stereo Matching

A stereo matching network with a cascade spatial pyramid pooling (CSPP) substructure

MULTI-SCALE CASCADE DISPARITY REFINEMENT STEREO NETWORK

Practical Stereo Matching via Cascaded Recurrent Network with Adaptive Correlation

Disparity Estimation Disparity Refinement Disparity Convolution Correlation Initial Disparity Residual Deconvolution Refined Disparity Reconstruction Error Shared Weights Skip Connection

Multi-path Feature Mining Network for Stereo Matching

A Light-Weight Stereo Matching Network Based on Multi-Scale Features Fusion and Robust Disparity Refinement

An efficient and accurate multi-level cascaded recurrent network for stereo matching

A Light-Weight Network with Multi-Scale Features Fusion and Color Guidance for Stereo Matching

DMCNet: Towards Lightweight Volumetric Stereo Matching

A Dual Branch Multiscale Stereo Matching Network for High-Resolution Satellite Remote Sensing Images

Multi-Dimensional Cooperative Network for Stereo Matching

A Joint 2D-3D Complementary Network for Stereo Matching

Stereo Matching Method with Integrated Geometric Encoding for Disparity Refinement

Real-Time Stereo Matching Networks Based on Multi-Modal and Sparse-Dense Fusion

Multilevel Disparity Reconstruction Network for Real-Time Stereo Matching

EHSS: an Efficient Hybrid-supervised Symmetric Stereo Matching Network

Multi-scale Cross-form Pyramid Network for Stereo Matching