Spaces:
Running
Running
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,
|