Skip to content

Latest commit

 

History

History
21 lines (18 loc) · 912 Bytes

training.md

File metadata and controls

21 lines (18 loc) · 912 Bytes
Raw

Here we provide instructions for how to train our models, including download of datasets.

MegaDepth

First the MegaDepth dataset needs to be downloaded and preprocessed. This can be done by the following steps:

  1. Download MegaDepth from here: https://www.cs.cornell.edu/projects/megadepth/
  2. Extract and preprocess: See https://github.com/mihaidusmanu/d2-net
  3. Download our prepared scene info from here: https://github.com/Parskatt/storage/releases/download/prep_scene_info/prep_scene_info.tar
  4. File structure should be data/megadepth/phoenix, data/megadepth/Undistorted_SfM, data/megadepth/prep_scene_info. Then run
python experiments/dkmv3/train_DKMv3_outdoor.py --gpus 4

Megadepth + Scannet

First follow the steps outlined above. Then, see https://github.com/zju3dv/LoFTR/blob/master/docs/TRAINING.md

Then run

python experiments/dkmv3/train_DKMv3_indoor.py --gpus 4