search

princeton-vl / RAFT-Stereo

MIT License
686 stars 131 forks source link

Training Datasets and schedule #72

Open orram opened 1 year ago

orram commented 1 year ago

https://github.com/princeton-vl/RAFT-Stereo/blob/0e2a12746143a7552e30ef2f4b1d4c3214388a1a/train_stereo.py#L222 Hi there, thank you for supplying such a clear code! I have a question regarding the training procedure; as I understand, the training you suggest on the git page includes only sceneflow (and refine on Middlebury) with no reference to other datasets - FallingThings and Tartanair, which you reference in the paper. Do you use them in any additional training? Can you clarify? Thank you!

lahavlipson commented 1 year ago

The model weights in this git repo are pre-trained only on Sceneflow, and refined on Middlebury, ETH3D or KITTI.