GREAT-IGEV
IGEV-Stereo extended with global attention (spatial, epipolar-row and volume attention). Commercially usable.
Commercial use: yes.
| variant | input | backend | time (ms) | peak memory (MiB, allocated / reserved) |
|---|---|---|---|---|
great-igev-sceneflow | Middlebury 2014 half (994x1482) | original (fp32) original portable kernels | 2252 1458 1085 393 | 8964 / 9268 8861 / 9206 1402 / 1678 1192 / 1564 |
great-igev-middlebury (max_disp 768) | Middlebury 2014 half (994x1482) | original (fp32) original portable kernels | OOM OOM 1785 579 | – – 1875 / 2078 1553 / 1744 |
Details
- Original code: https://github.com/JarvisLee0423/GREAT-Stereo at
2a6b1efede8c - Code licence: Apache-2.0 (permissive); files:
LICENSE - Weights licence: Apache-2.0 (assumed from code)
- Licence evidence: https://github.com/JarvisLee0423/GREAT-Stereo/blob/2a6b1efede8c3a9eec9540790c2700ea2f3dee7d/README.md#L558
- Training data: Freiburg SceneFlow (FlyingThings3D, Driving, Monkaa)
great-igev-kitti: Freiburg SceneFlow (FlyingThings3D, Driving, Monkaa), KITTI stereo 2012 / 2015 (SceneFlow, then fine-tuned on KITTI 2012 + 2015 (launchers/.../great_igev_trainer_kitti.sh; KITTI submission model))great-igev-eth3d: Freiburg SceneFlow (FlyingThings3D, Driving, Monkaa), CREStereo synthetic stereo dataset (Megvii), InStereo2K, ETH3D two-view stereo, not stated by the authors (ETH3D submission model; original recipe (launchers, utils/stereo_matching/data_utils/data_fetchers.py): SceneFlow model fine-tuned on eth3d_finetune (ETH3D, InStereo2K, CREStereo); whether the eth3d_train mixture (+ TartanAir, Sintel) ran first is not stated)great-igev-middlebury: Freiburg SceneFlow (FlyingThings3D, Driving, Monkaa), CREStereo synthetic stereo dataset (Megvii), InStereo2K, HR-VS (high-resolution virtual stereo, Yang et al. CVPR 2019), Middlebury stereo (2014), NVIDIA Falling Things (FAT), not stated by the authors (Middlebury submission model; original recipe (launchers, utils/stereo_matching/data_utils/data_fetchers.py): SceneFlow model fine-tuned on middlebury_finetune (CREStereo, InStereo2K, HR-VS, Middlebury 2005 / 2006 / 2014 / 2021 / MiddEval3 H + F, Falling Things); whether the middlebury_train mixture (+ TartanAir) ran first is not stated)
- Options:
iters= 32 (>= 1),cuda_graphs= False (speed only) - Kernel libraries:
igev,igevpp,cutlass_conv,cudnn_hacc - Variants:
great-igev-sceneflow,great-igev-kitti,great-igev-eth3d,great-igev-middlebury - Outputs: disparity
- Input size: any H x W >= 32, padded to a multiple of 32
- Tested up to (portable, fp16, RTX 5060 Ti 16 GB):
great-igev-sceneflownot measured yet;great-igev-kittinot measured yet;great-igev-eth3dnot measured yet;great-igev-middleburynot measured yet great-igev-kitti,great-igev-eth3d: same network and settings asgreat-igev-sceneflow, measured once (rows not repeated).great-igev-scenefloworiginal (fp32) 994x1482, original 994x1482, portable 994x1482, kernels 994x1482: measured on great-igev-eth3d (same architecture, other checkpoint)great-igev-sceneflowload / first call: original 0.44 s / 4.596 s (375x1242), portable 0.32 s / 1.102† s, kernels 0.23† s / 0.397† sgreat-igev-kittiload / first call: original 0.44 s / 4.596 s (375x1242), portable 0.32 s / 1.102† s, kernels 0.23† s / 0.397† sgreat-igev-eth3dload / first call: original 0.44 s / 4.596 s (375x1242), portable 0.32 s / 1.102† s, kernels 0.23† s / 0.397† sgreat-igev-middleburyload / first call: original 0.39 s / 4.586 s (375x1242), portable 3.91 s / 1.808† s, kernels 0.61† s / 0.589† s- † measured in the speed sweep after the portable load (or an earlier size) in the same process: file cache and kernels are warm, so the time is lower than a fresh start.
- Protocol, median of 5 after 3 warm-up, uint8 CUDA input (synthetic for bench_inputs sizes), card defaults; gc before each load; peak = max over the first call and the timed calls
Measured
Original = the original code named under Details, same checkpoint and protocol.
| variant | dataset | original fp32 | original fp16 | portable | kernels | note |
|---|---|---|---|---|---|---|
great-igev-sceneflow | ETH3D two-view train, bad > 1 px % / EPE | 4.48 / 0.364 | 4.41 / 0.363 | 4.43 / 0.364 | 4.44 / 0.364 | |
great-igev-sceneflow | Middlebury 2014 trainingH, bad > 2 px % / EPE | 11.22 / 2.710 | 11.19 / 2.698 | 11.14 / 2.687 | 11.16 / 2.691 | |
great-igev-kitti | ETH3D two-view train, bad > 1 px % / EPE | 23.35 / 3.115 | 23.36 / 3.116 | 23.37 / 3.113 | 23.36 / 3.121 | |
great-igev-kitti | Middlebury 2014 trainingH, bad > 2 px % / EPE | 17.99 / 4.677 | 18.00 / 4.679 | 17.98 / 4.670 | 18.00 / 4.676 | |
great-igev-eth3d | ETH3D two-view train, bad > 1 px % / EPE | 0.49 / 0.116 | 0.49 / 0.117 | 0.49 / 0.118 | 0.49 / 0.118 | trained on this dataset's public images; not a zero-shot result |
great-igev-eth3d | Middlebury 2014 trainingH, bad > 2 px % / EPE | 11.20 / 2.894 | 11.21 / 2.903 | 11.19 / 2.946 | 11.21 / 2.883 | possibly in the training data |
great-igev-middlebury | ETH3D two-view train, bad > 1 px % / EPE | 27.42 / 1.133 | 27.43 / 1.134 | 27.44 / 1.131 | 27.53 / 1.135 | possibly in the training data |
great-igev-middlebury | Middlebury 2014 trainingH, bad > 2 px % / EPE | out of memory | out of memory | 4.14 / 0.707 | 4.14 / 0.708 | trained on this dataset's public images; not a zero-shot result |
Measured on NVIDIA GeForce RTX 5060 Ti, torch 2.14.1+cu130, fastforks 468fba10, 2026-11-05.
Parity
CPU, float32: max |diff| <= 0.002 vs the original. GPU, float16 (the default), portable and kernels: mean |diff| <= 0.02 px and p99 <= 0.25 px.
| variant | kernels error vs original, mean / p99 | GPU, torch | kernels used |
|---|---|---|---|
great-igev-sceneflow | 0.00166 / 0.02857 | RTX 5060 Ti, torch 2.14.1+cu130 | cudnn_hacc@cudnn92400, cutlass_conv@988138e042cb:sm_120, igev@f075827d68c6:sm_120, igevpp@df1bb4dbe09a:sm_120 |
great-igev-kitti | 0.00284 / 0.03707 | RTX 5060 Ti, torch 2.14.1+cu130 | cudnn_hacc@cudnn92400, cutlass_conv@988138e042cb:sm_120, igev@f075827d68c6:sm_120, igevpp@df1bb4dbe09a:sm_120 |
great-igev-eth3d | 0.00214 / 0.03202 | RTX 5060 Ti, torch 2.14.1+cu130 | cudnn_hacc@cudnn92400, cutlass_conv@988138e042cb:sm_120, igev@f075827d68c6:sm_120, igevpp@df1bb4dbe09a:sm_120 |
great-igev-middlebury | 0.00162 / 0.00911 | RTX 5060 Ti, torch 2.14.1+cu130 | cudnn_hacc@cudnn92400, cutlass_conv@988138e042cb:sm_120, igev@f075827d68c6:sm_120, igevpp@df1bb4dbe09a:sm_120 |