E3tereo: Improving Stereo Matching at Depth Boundaries with Learned Geometric Edges
Organizations: Xidian University · Hefei University
Abstract
Depth boundaries remain a major source of error in stereo matching. For IGEV on Scene Flow, geometric edge pixels account for only 6% of evaluated pixels but about half of the total absolute disparity error. Photometric edges can respond to texture changes without corresponding depth discontinuities, making them unreliable cues for boundary refinement. We propose E3Stereo, which learns geometric edges from disparity supervision and combines these boundary cues with the backbone disparity and image details to predict full-resolution disparity corrections. The same edge map also emphasizes boundary errors during training, while joint optimization allows the edge predictor to adapt to disparity refinement. On the Scene Flow dataset, E3Stereo reduces end-point error (EPE) at depth boundaries by 9.5%-14.3% across RAFT-Stereo, IGEV, and MonSter, while also reducing overall EPE. After fine-tuning, our IGEV-based model improves upon IGEV on all eleven reported metrics across ETH3D, KITTI 2012, and KITTI 2015. The source code will be made publicly available upon acceptance.
Figures & tables
| Configuration | EPE | EPE Edge | Params (M) |
|---|---|---|---|
| Pretrained, frozen GeoEdgeNet | 0.46 | 3.53 | 14.90 |
| Jointly trained full model | 0.43 | 3.23 | 12.78 |
| Method | ETH3D | Middlebury | KITTI 2015 | KITTI 2012 | Runtime (ms) | ||||||||||
| Bad1.0 Noc | Bad1.0 All | RMSE Noc | Bad2.0 Noc | Bad2.0 All | Bad4.0 All | D1-bg Noc | D1-all Noc | D1-bg All | D1-all All | Out-2 Noc | Out-2 All | Out-3 Noc | Out-3 All | ||
| Cost-volume methods | |||||||||||||||
| GwcNet [ 16 ] | 6.42 | 6.95 | 0.69 | – | – | – | 1.61 | 1.92 | 1.74 | 2.11 | 2.16 | 2.71 | 1.32 | 1.70 | 320 |
| GANet [ 47 ] | 6.22 | 6.86 | 0.75 | – | – | – | 1.34 | 1.63 | 1.48 | 1.81 | 1.89 | 2.50 | 1.19 | 1.60 | 1800 |
| AdaStereo [ 35 ] | 3.09 | 3.34 | 0.44 | 13.7 | 19.8 | 10.9 | 2.39 | 2.83 | 2.59 | 3.08 | – | – | – | – | 410 |
| CasStereo [ 11 ] | – | – | – | 18.8 | 26.0 | 18.5 | – | – | – | – | – | – | – | – | – |
| Backbone | EPE | EPE Edge | Params (M) | Time (ms) |
|---|---|---|---|---|
| RAFT-Stereo [ 22 ] | 0.72 0.67 | 5.14 4.45 | 11.12 11.30 | 380 390 |
| IGEV [ 41 ] | 0.47 0.43 | 3.77 3.23 | 12.60 12.78 | 180 190 |
| MonSter [ 7 ] | 0.37 0.35 | 3.27 2.96 | 388.69 388.87 | 450 460 |
| Dataset | Threshold (px) | Bad- (%) | |
|---|---|---|---|
| IGEV | E 3 Stereo | ||
| KITTI 2012 | 3 | 5.14 | 5.59 |
| KITTI 2015 | 3 | 6.04 | 6.25 |
| Middlebury Quarter | 2 | 6.21 | 6.80 |
| Middlebury Half | 2 | 7.08 | 7.43 |
| ETH3D | 1 | 3.61 | 3.97 |
| Configuration | EPE | EPE Edge | Params (M) | Time (ms) |
| Progressive component addition | ||||
| IGEV baseline | 0.47 | 3.77 | 12.60 | 180 |
| + edge-weighted loss (GT edges) | 0.46 | 3.68 | 12.60 | 180 |
| + GeoEdgeNet | 0.45 | 3.42 | 12.74 | 186 |
| + GeoRestore | 0.43 | 3.23 | 12.78 | 190 |
| Additional ablations | ||||
Stereo Matching
Stereo matching estimates depth by finding corresponding points in paired images and measuring their disparity.
See all 57 papers