From 88255525a6351917aeebd54348f635fac5af1836 Mon Sep 17 00:00:00 2001 From: Misha Date: Wed, 20 Sep 2023 10:13:55 +0000 Subject: [PATCH] add data in matches for doppelganger --- hloc/match_features.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/hloc/match_features.py b/hloc/match_features.py index f9ef3e5f..dc20419a 100644 --- a/hloc/match_features.py +++ b/hloc/match_features.py @@ -146,6 +146,12 @@ def writer_fn(inp, match_path): if 'matching_scores0' in pred: scores = pred['matching_scores0'][0].cpu().half().numpy() grp.create_dataset('matching_scores0', data=scores) + if 'matches' in pred: + matches = pred['matches'][0].cpu().short().numpy() + grp.create_dataset('matches', data=matches) + if 'scores' in pred: + scores = pred['scores'][0].cpu().half().numpy() + grp.create_dataset('scores', data=scores) def main(conf: Dict,