File size: 912 Bytes
62c7319 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
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
``` bash
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
``` bash
python experiments/dkmv3/train_DKMv3_indoor.py --gpus 4
```
|