adapt scale to have comparable recall

vita-epfl · Apr 22, 2021 · 6e37001 · 6e37001
1 parent 7ac6855
commit 6e37001
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/monoloco/eval/generate_kitti.py b/monoloco/eval/generate_kitti.py
@@ -233,7 +233,8 @@ def save_txts(path_txt, all_inputs, all_outputs, all_params, net='monoloco', cat
             if net in ('monstereo', 'monoloco_pp'):
                 alpha, ry = float(yaws[0][idx]), float(yaws[1][idx])
                 hwl = [float(hs[idx]), float(ws[idx]), float(ls[idx])]
-                conf_scale = 0.035  # scale to obtain (approximately) same recall at evaluation
+                # scale to obtain (approximately) same recall at evaluation
+                conf_scale = 0.035 if net == 'monoloco_pp' else 0.033
             else:
                 alpha, ry, hwl = -10., -10., [0, 0, 0]
                 conf_scale = 0.05