Spaces:
Running
Running
Realcat
commited on
Commit
•
c3f14e3
1
Parent(s):
e9f6961
fix: KeyError: 'geom_info'
Browse files- 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
|
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,
|