Skip to content

Latest commit

 

History

History
executable file
·
74 lines (45 loc) · 3.29 KB

DATASETS.md

File metadata and controls

executable file
·
74 lines (45 loc) · 3.29 KB

Datasets

Optical Flow

The datasets used to train and evaluate our GMFlow model are as follows:

By default the dataloader dataloader/flow/datasets.py assumes the datasets are located in the datasets directory.

It is recommended to symlink your dataset root to datasets:

ln -s $YOUR_DATASET_ROOT datasets

Otherwise, you may need to change the corresponding paths in dataloader/flow/datasets.py.

Stereo Matching

The datasets used to train and evaluate our GMStereo model are as follows:

By default the dataloader dataloader/stereo/datasets.py assumes the datasets are located in the datasets directory.

It is recommended to symlink your dataset root to datasets:

ln -s $YOUR_DATASET_ROOT datasets

Otherwise, you may need to change the corresponding paths in dataloader/stereo/datasets.py.

Depth Estimation

The datasets used to train and evaluate our GMDepth model are as follows:

We support downloading and extracting the DeMoN dataset in our code: dataloader/depth/download_demon_train.sh, dataloader/depth/download_demon_test.sh, dataloader/depth/prepare_demon_train.sh and dataloader/depth/prepare_demon_test.sh.

By default the dataloader dataloader/depth/datasets.py assumes the datasets are located in the datasets directory.

It is recommended to symlink your dataset root to datasets:

ln -s $YOUR_DATASET_ROOT datasets

Otherwise, you may need to change the corresponding paths in dataloader/depth/datasets.py.