Realcat commited on
Commit
c3f14e3
1 Parent(s): e9f6961

fix: KeyError: 'geom_info'

Browse files
Files changed (1) hide show
  1. common/utils.py +2 -1
common/utils.py CHANGED
@@ -311,6 +311,7 @@ def set_null_pred(feature_type: str, pred: dict):
311
  pred["mline_keypoints0_orig"] = np.array([])
312
  pred["mline_keypoints1_orig"] = np.array([])
313
  pred["H"] = None
 
314
  return pred
315
 
316
 
@@ -973,7 +974,7 @@ def run_matching(
973
  "extractor_conf": extract_conf,
974
  },
975
  {
976
- "geom_info": pred["geom_info"],
977
  },
978
  output_wrapped,
979
  state_cache,
 
311
  pred["mline_keypoints0_orig"] = np.array([])
312
  pred["mline_keypoints1_orig"] = np.array([])
313
  pred["H"] = None
314
+ pred["geom_info"] = {}
315
  return pred
316
 
317
 
 
974
  "extractor_conf": extract_conf,
975
  },
976
  {
977
+ "geom_info": pred.get("geom_info", {}),
978
  },
979
  output_wrapped,
980
  state_cache,