File size: 2,245 Bytes
fab39ea | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 | ---
license: apache-2.0
tags:
- computer-vision
- stereo-matching
- depth-estimation
- surface-normal-estimation
- diffusion
- geometry
library_name: pytorch
pipeline_tag: depth-estimation
paper:
- arxiv: 2607.24024
---
<div align="center">
<h1>
GeoStereo: A Unified Stereo Geometry Estimation Framework for Disparity and Surface Normal
</h1>
<p>
Qizhe Wei, Xianda Guo, Shaocong Xu, Hong Li, Runyi Yang, Hao Zhao
</p>
<p>
<a href="https://arxiv.org/abs/2607.24024">
<img src="https://img.shields.io/badge/arXiv-2607.24024-b31b1b.svg" alt="arXiv">
</a>
<a href="https://qz-wei.github.io/GeoStereo.github.io/">
<img src="https://img.shields.io/badge/Project_Page-green" alt="Project Page">
</a>
<a href="https://huggingface.co/qz-wei/GeoStereo">
<img src="https://img.shields.io/badge/%F0%9F%A4%97%20Hugging%20Face-Model_Weights-yellow" alt="Hugging Face">
</a>
</p>
</div>
# GeoStereo
Official implementation of:
**GeoStereo: A Unified Stereo Geometry Estimation Framework for Disparity and Surface Normal**
GeoStereo is a unified stereo geometry estimation framework that jointly predicts disparity and surface normal by combining stereo matching with diffusion-based geometric priors.
## Paper
**A Unified Stereo Geometry Estimation Framework for Disparity and Surface Normal**
arXiv:
https://arxiv.org/abs/2607.24024
## Project Links
- Project Page:
https://qz-wei.github.io/GeoStereo.github.io/
- Hugging Face:
https://huggingface.co/qz-wei/GeoStereo
- Paper:
https://arxiv.org/abs/2607.24024
# Method Overview
GeoStereo explores the joint estimation of stereo disparity and surface normal within a unified geometric framework.
The framework consists of:
- A stereo matching branch for disparity estimation.
- A diffusion-based normal estimation branch for surface geometry modeling.
- Geometry-aware feature interaction between disparity and normal representations.
- Joint optimization that enables mutual enhancement between stereo correspondence and surface geometry.
# Model Zoo
| Model | Description |
|------|-------------|
| GeoStereo | Full model checkpoint containing stereo and diffusion-based normal estimation modules |
# Download
## Using Hugging Face CLI
```bash
hf download qz-wei/GeoStereo |