loftr_ds_dense.py 461 B

12345678910111213141516
  1. from src.config.default import _CN as cfg
  2. cfg.LOFTR.MATCH_COARSE.MATCH_TYPE = 'dual_softmax'
  3. cfg.LOFTR.MATCH_COARSE.SPARSE_SPVS = False
  4. cfg.TRAINER.CANONICAL_LR = 8e-3
  5. cfg.TRAINER.WARMUP_STEP = 1875 # 3 epochs
  6. cfg.TRAINER.WARMUP_RATIO = 0.1
  7. cfg.TRAINER.MSLR_MILESTONES = [8, 12, 16, 20, 24]
  8. # pose estimation
  9. cfg.TRAINER.RANSAC_PIXEL_THR = 0.5
  10. cfg.TRAINER.OPTIMIZER = "adamw"
  11. cfg.TRAINER.ADAMW_DECAY = 0.1
  12. cfg.LOFTR.MATCH_COARSE.TRAIN_COARSE_PERCENT = 0.3