loftr_ds_eval.py 723 B

12345678910111213141516
  1. """ A config only for reproducing the ScanNet evaluation results.
  2. We remove border matches by default, but the originally implemented
  3. `remove_border()` has a bug, leading to only two sides of
  4. all borders are actually removed. However, the [bug fix](https://github.com/zju3dv/LoFTR/commit/e9146c8144dea5f3cbdd98b225f3e147a171c216)
  5. makes the scannet evaluation results worse (auc@10=40.8 => 39.5), which should be
  6. caused by tiny result fluctuation of few image pairs. This config set `BORDER_RM` to 0
  7. to be consistent with the results in our paper.
  8. """
  9. from src.config.default import _CN as cfg
  10. cfg.LOFTR.COARSE.TEMP_BUG_FIX = False
  11. cfg.LOFTR.MATCH_COARSE.MATCH_TYPE = 'dual_softmax'
  12. cfg.LOFTR.MATCH_COARSE.BORDER_RM = 0