Hugging Face's logo

Spaces:
Hoangthanh
/
image-matching-webui
No application file

App Files Community
image-matching-webui / third_party /DKM /docs /training.md
Vincentqyw
add: roma
c608946
preview code
|
raw
history blame
912 Bytes

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