We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent af4002c commit d347c73Copy full SHA for d347c73
1 file changed
gluefactory/utils/export_predictions.py
@@ -49,11 +49,8 @@ def export_predictions(
49
)
50
pred[k] = pred[k] * scales[None]
51
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]
+ scales = 1.0 / data["scales"]
+ pred["lines"] *= scales[None]
57
if k.startswith("orig_lines"):
58
idx = k.replace("orig_lines", "")
59
scales = 1.0 / (
0 commit comments