Skip to content

Commit d347c73

Browse files
committed
fix minor issue
1 parent af4002c commit d347c73

1 file changed

Lines changed: 2 additions & 5 deletions

File tree

gluefactory/utils/export_predictions.py

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -49,11 +49,8 @@ def export_predictions(
4949
)
5050
pred[k] = pred[k] * scales[None]
5151
if k == "lines":
52-
idx = k.replace("lines", "")
53-
scales = 1.0 / (
54-
data["scales"] if len(idx) == 0 else data[f"view{idx}"]["scales"]
55-
)
56-
pred[k] = pred[k] * scales[None]
52+
scales = 1.0 / data["scales"]
53+
pred["lines"] *= scales[None]
5754
if k.startswith("orig_lines"):
5855
idx = k.replace("orig_lines", "")
5956
scales = 1.0 / (

0 commit comments

Comments
 (0)