Matchability and Uncertainty-Aware Iterative Disparity Refinement for Stereo Matching
Source abstract
Abstract
After significant progress in stereo matching, the pursuit of robust and efficient ill-posed-region disparity refinement methods remains challenging. To further improve the performance of disparity refinement, in this paper, we propose the matchability and uncertainty-aware iterative disparity refinement neural network. Firstly, a new matchability and uncertainty decoder (MUD) is proposed to decode the matchability mask and disparity uncertainties, which are used to evaluate the reliability of feature matching and estimated disparity, thereby reducing the susceptibility to mismatched pixels. Then, based on the proposed MUD, we present two modules: the uncertainty-preferred disparity field initialization (UFI) and the masked hidden state global aggregation (MGA) modules. In the UFI, a multi-disparity window scan-and-select method is employed to provide a further initialized disparity field and more accurate initial disparity. In the MGA, the adaptive masked disparity field hidden state is globally aggregated to extend the propagation range per iteration, improving the refinement efficiency. Finally, the experimental results on public datasets show that the proposed model achieves a reduction up to 17.9% in disparity average error and 16.9% in occluded outlier proportion, respectively, demonstrating its more practical handling of ill-posed regions.
Source: crossref · Quality: baseline
Research context
OpenAlex topics: Advanced Vision and Imaging · Medical Image Segmentation Techniques · Advanced Image Fusion Techniques
SciVX discovery tags are terminology-based. OpenAlex topics are source-provided classifications.
Affiliations
Engineering Research Center of Optical Instrument and System, Ministry of Education, Shanghai Key Lab of Modern Optical System, School of Optical-Electrical and Computer Engineering, University of Shanghai for Science and Technology, 516 Jungong Rd, Shanghai 200093, China
Funding
Science & Technology Commission of Shanghai Municipality (STCSM)
References (55)
Show source-provided references
1. Stereo Processing by Semiglobal Matching and Mutual Information10.1109/TPAMI.2007.1166 ↗
2. Bleyer, M., Rhemann, C., and Rother, C. (November, January 29). PatchMatch Stereo-Stereo Matching with Slanted Support Windows. Proceedings of the British Machine Vision Conference 2011, Dundee, Scotland.10.5244/C.25.14 ↗
3. A Taxonomy and Evaluation of Dense Two-Frame Stereo Correspondence Algorithms10.1023/A:1014573219977 ↗
4. Chang, J.-R., and Chen, Y.-S. (2018, January 18–23). Pyramid Stereo Matching Network. Proceedings of the 2018 IEEE/CVF Conference on Computer Vision and Pattern Recognition, Salt Lake City, UT, USA.10.1109/CVPR.2018.00567 ↗
5. Guo, X., Yang, K., Yang, W., Wang, X., and Li, H. (2019, January 15–20). Group-Wise Correlation Stereo Network. Proceedings of the 2019 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), Long Beach, CA, USA.10.1109/CVPR.2019.00339 ↗
6. Zhang, F., Prisacariu, V., Yang, R., and Torr, P.H.S. (2019, January 16–20). GA-Net: Guided Aggregation Net for End-To-End Stereo Matching. Proceedings of the 2019 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), Long Beach, CA, USA.10.1109/CVPR.2019.00027 ↗
7. StereoNet: Guided Hierarchical Refinement for Real-Time Edge-Aware Depth Prediction10.1007/978-3-030-01267-0_35 ↗
8. A Survey on Deep Learning Techniques for Stereo-Based Depth Estimation10.1109/TPAMI.2020.3032602 ↗
9. Stereo Matching with Color-Weighted Correlation, Hierarchical Belief Propagation, and Occlusion Handling10.1109/TPAMI.2008.99 ↗
10. Occlusion Filling in Stereo: Theory and Experiments10.1016/j.cviu.2013.01.008 ↗
11. Accurate Image-Guided Stereo Matching with Efficient Matching Cost and Disparity Refinement10.1109/TCSVT.2015.2473375 ↗
12. Lipson, L., Teed, Z., and Deng, J. (2021, January 1–3). RAFT-Stereo: Multilevel Recurrent Field Transforms for Stereo Matching. Proceedings of the 2021 International Conference on 3D Vision (3DV), London, UK.10.1109/3DV53792.2021.00032 ↗
13. Li, J., Wang, P., Xiong, P., Cai, T., Yan, Z., Yang, L., Liu, J., Fan, H., and Liu, S. (2022, January 18–24). Practical Stereo Matching via Cascaded Recurrent Network with Adaptive Correlation. Proceedings of the 2022 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR).10.1109/CVPR52688.2022.01578 ↗
14. Zhao, H., Zhou, H., Zhang, Y., Zhao, Y., Yang, Y., and Ouyang, T. (2022, January 4–8). EAI-Stereo: Error Aware Iterative Network for Stereo Matching. Proceedings of the 16th Asian Conference on Computer Vision (ACCV), Macao, China.10.1007/978-3-031-26319-4_1 ↗
15. Meister, S., Hur, J., and Roth, S. (2018, January 2–7). UnFlow: Unsupervised Learning of Optical Flow with a Bidirectional Census Loss. Proceedings of the Thirty-Second AAAI Conference on Artificial Intelligence, New Orleans, LA, USA.10.1609/aaai.v32i1.12276 ↗
16. Unifying Flow, Stereo and Depth Estimation10.1109/TPAMI.2023.3298645 ↗
17. Kendall, A., and Gal, Y. (2017, January 4–9). What Uncertainties Do We Need in Bayesian Deep Learning for Computer Vision?. Proceedings of the 31st Annual Conference on Neural Information Processing Systems (NIPS), Long Beach, CA, USA.
18. Stereo Matching by Training a Convolutional Neural Network to Compare Image Patches
19. Zhang, Y., Chen, Y., Bai, X., Yu, S., Yu, K., Li, Z., and Yang, K. (2020, January 7–12). Adaptive Unimodal Cost Volume Filtering for Deep Stereo Matching. Proceedings of the Thirty-Fourth AAAI Conference on Artificial Intelligence, New York, NY, USA.10.1609/aaai.v34i07.6991 ↗
20. Kendall, A., Martirosyan, H., Dasgupta, S., Henry, P., Kennedy, R., Bachrach, A., and Bry, A. (2017, January 22–29). End-to-End Learning of Geometry and Context for Deep Stereo Regression. Proceedings of the 2017 IEEE International Conference on Computer Vision (ICCV), Venice, Italy.10.1109/ICCV.2017.17 ↗
21. Mayer, N., Ilg, E., Hausser, P., Fischer, P., Cremers, D., Dosovitskiy, A., and Brox, T. (2016, January 27–30). A Large Dataset to Train Convolutional Networks for Disparity, Optical Flow, and Scene Flow Estimation. Proceedings of the 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), Las Vegas, NV, USA.10.1109/CVPR.2016.438 ↗
22. Bangunharcana, A., Cho, J.W., Lee, S., Kweon, I.S., Kim, K.-S., and Kim, S. (October, January 27). Correlate-and-Excite: Real-Time Stereo Matching via Guided Cost Volume Excitation. Proceedings of the 2021 IEEE/RSJ International Conference on Intelligent Robots and Systems (IROS), ELECTR NETWORK, Prague, Czech Republic.10.1109/IROS51168.2021.9635909 ↗
23. Geiger, A., Lenz, P., and Urtasun, R. (2012, January 16–21). Are We Ready for Autonomous Driving? The KITTI Vision Benchmark Suite. Proceedings of the Conference on Computer Vision and Pattern Recognition (CVPR), Providence, RI, USA.10.1109/CVPR.2012.6248074 ↗
24. Joint 3D Estimation of Vehicles and Scene Flow10.5194/isprsannals-II-3-W5-427-2015 ↗
25. High-Resolution Stereo Datasets with Subpixel-Accurate Ground Truth10.1007/978-3-319-11752-2_3 ↗
26. Schöps, T., Schönberger, J.L., Galliani, S., Sattler, T., Schindler, K., Pollefeys, M., and Geiger, A. (2017, January 21–26). A Multi-View Stereo Benchmark with High-Resolution Images and Multi-Camera Videos. Proceedings of the Conference on Computer Vision and Pattern Recognition (CVPR), Honolulu, HI, USA.10.1109/CVPR.2017.272 ↗
27. Duggal, S., Wang, S., Ma, W.-C., Hu, R., and Urtasun, R. (November, January 27). DeepPruner: Learning Efficient Stereo Matching via Differentiable PatchMatch. Proceedings of the 2019 IEEE/CVF International Conference on Computer Vision (ICCV), Seoul, Republic of Korea.10.1109/ICCV.2019.00448 ↗
28. Li, Z., Liu, X., Drenkow, N., Ding, A., Creighton, F.X., Taylor, R.H., and Unberath, M. (2021, January 11–17). Revisiting Stereo Depth Estimation from a Sequence-to-Sequence Perspective with Transformers. Proceedings of the 2021 IEEE/CVF International Conference on Computer Vision (ICCV), ELECTR NETWORK, Montreal, QC, Canada.10.1109/ICCV48922.2021.00614 ↗
29. Teed, Z., and Deng, J. (2020, January 23–28). RAFT: Recurrent All-Pairs Field Transforms for Optical Flow. Computer Vision-ECCV 2020. Proceedings of the 16th European Conference; Proceedings. Lecture Notes in Computer Science (LNCS 12347), Glasgow, UK.10.1007/978-3-030-58536-5_24 ↗
30. Cho, K., van Merrienboer, B., Bahdanau, D., and Bengio, Y. (2014). On the Properties of Neural Machine Translation: Encoder-Decoder Approaches. arXiv.10.3115/v1/W14-4012 ↗
31. Zhao, H., Zhou, H., Zhang, Y., Chen, J., Yang, Y., and Zhao, Y. (2023, January 17–24). High-Frequency Stereo Matching Network. Proceedings of the 2023 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), Vancouver, BC, Canada.10.1109/CVPR52729.2023.00134 ↗
32. Xu, G., Wang, X., Ding, X., and Yang, X. (2023, January 17–24). Iterative Geometry Encoding Volume for Stereo Matching. Proceedings of the 2023 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), Vancouver, BC, Canada.10.1109/CVPR52729.2023.02099 ↗
33. Sun, J., Shen, Z., Wang, Y., Bao, H., and Zhou, X. (2021, January 19–25). LoFTR: Detector-Free Local Feature Matching with Transformers. Proceedings of the 2021 IEEE/CVF Conference on Computer Vision and Pattern Recognition, CVPR, ELECTR NETWORK, Virtual.10.1109/CVPR46437.2021.00881 ↗
34. Poggi, M., and Mattoccia, S. (2016, January 19–22). Learning from Scratch a Confidence Measure. Proceedings of the British Machine Vision Conference 2016, York, UK.10.5244/C.30.46 ↗
35. Seki, A., and Pollefeys, M. (2016, January 19–22). Patch Based Confidence Prediction for Dense Disparity Map. Proceedings of the British Machine Vision Conference 2016, York, UK.10.5244/C.30.23 ↗
36. Poggi, M., and Mattoccia, S. (2017, January 12–26). Learning to Predict Stereo Reliability Enforcing Local Consistency of Confidence Maps. Proceedings of the Conference on Computer Vision and Pattern Recognition (CVPR), Honolulu, HI, USA.10.1109/CVPR.2017.483 ↗
37. Beyond Local Reasoning for Stereo Confidence Estimation with Deep Learning10.1007/978-3-030-01231-1_20 ↗
38. Kim, S., Kim, S., Min, D., and Sohn, K. (2019, January 16–20). LAF-Net: Locally Adaptive Fusion Networks for Stereo Confidence Estimation. Proceedings of the 2019 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), Long Beach, CA, USA.10.1109/CVPR.2019.00029 ↗
39. PDC-Net+: Enhanced Probabilistic Dense Correspondence Network10.1109/TPAMI.2023.3249225 ↗
40. Eldesokey, A., Felsberg, M., Holmquist, K., and Persson, M. (2020, January 13–19). Uncertainty-Aware CNNs for Depth Completion: Uncertainty from Beginning to End. Proceedings of the 2020 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), Seattle, WA, USA.10.1109/CVPR42600.2020.01203 ↗
41. Zhang, J., Yao, Y., Luo, Z., Li, S., Shen, T., Fang, T., and Quan, L. (2021, January 10–15). Learning Stereo Matchability in Disparity Regression Networks. Proceedings of the 25th International Conference on Pattern Recognition (ICPR), ELECTR NETWORK, Milan, Italy.10.1109/ICPR48806.2021.9412342 ↗
42. Aleatoric Uncertainty Estimation for Dense Stereo Matching via CNN-Based Cost Volume Analysis10.1016/j.isprsjprs.2020.11.003 ↗
43. Joint Estimation of Depth and Its Uncertainty from Stereo Images Using Bayesian Deep Learning10.5194/isprs-annals-V-2-2022-69-2022 ↗
44. Chen, L., Wang, W., and Mordohai, P. (2023, January 17–24). Learning the Distribution of Errors in Stereo Matching for Joint Disparity and Uncertainty Estimation. Proceedings of the 2023 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), Vancouver, BC, Canada.10.1109/CVPR52729.2023.01653 ↗
45. Ghiasi, G., Lin, T.-Y., and Le, Q.V. (2018, January 2–8). DropBlock: A Regularization Method for Convolutional Networks. Proceedings of the Advances in Neural Information Processing Systems 31 (NIPS), Montreal, Canada.
46. Dosovitskiy, A., Beyer, L., Kolesnikov, A., Weissenborn, D., Zhai, X., Unterthiner, T., Dehghani, M., Minderer, M., Heigold, G., and Gelly, S. (2021, January 3–7). An Image Is Worth 16x16 Words: Transformers for Image Recognition at Scale. Proceedings of the International Conference on Learning Representations (ICLR), Virtual Event, Austria.
47. MaxViT: Multi-Axis Vision Transformer10.1007/978-3-031-20053-3_27 ↗
48. Jiang, S., Campbell, D., Lu, Y., Li, H., and Hartley, R. (2021, January 11–17). Learning to Estimate Hidden Motions with Global Motion Aggregation. Proceedings of the 2021 IEEE/CVF International Conference on Computer Vision (ICCV), ELECTR NETWORK, Montreal, QC, Canada.10.1109/ICCV48922.2021.00963 ↗
49. RoFormer: Enhanced Transformer with Rotary Position Embedding10.1016/j.neucom.2023.127063 ↗
50. Realistic CG Stereo Image Dataset with Ground Truth Disparity Maps
51. Tremblay, J., To, T., and Birchfield, S. (2018, January 18–22). Falling Things: A Synthetic Dataset for 3D Object Detection and Pose Estimation. Proceedings of the 2018 IEEE/CVF Conference on Computer Vision and Pattern Recognition Workshops (CVPRW), Salt Lake City, UT, USA.10.1109/CVPRW.2018.00275 ↗
52. Wang, W., Zhu, D., Wang, X., Hu, Y., Qiu, Y., Wang, C., Hu, Y., Kapoor, A., and Scherer, S. (2020–24, January 24). TartanAir: A Dataset to Push the Limits of Visual SLAM. Proceedings of the 2020 IEEE/RSJ International Conference on Intelligent Robots and Systems (IROS), ELECTR NETWORK, Las Vegas, NV, USA.10.1109/IROS45743.2020.9341801 ↗
53. Paszke, A., Gross, S., Massa, F., Lerer, A., Bradbury, J., Chanan, G., Killeen, T., Lin, Z., Gimelshein, N., and Antiga, L. (2019, January 8–14). PyTorch: An Imperative Style, High-Performance Deep Learning Library. Proceedings of the Advances in Neural Information Processing Systems (NIPS), Vancouver, BC, Canada.
54. Loshchilov, I., and Hutter, F. (2017, January 24–26). Decoupled Weight Decay Regularization. Proceedings of the International Conference on Learning Representations (ICLR), Toulon, France.
55. Smith, L.N., and Topin, N. (2019, January 15–17). Super-Convergence: Very Fast Training of Neural Networks Using Large Learning Rates. Proceedings of the Artificial Intelligence and Machine Learning for Multi-Domain Operations Applications, Baltimore, MD, USA.10.1117/12.2520589 ↗