From 04e37f3a395e0410f0c95371ef397cb05927f655 Mon Sep 17 00:00:00 2001 From: MedericFourmy Date: Thu, 18 Apr 2024 11:34:17 -0400 Subject: [PATCH] cosy prediction runner with detection th == 0 -> reproduces cosypose pbr official results --- .../cosypose/cosypose/evaluation/prediction_runner.py | 1 + 1 file changed, 1 insertion(+) diff --git a/happypose/pose_estimators/cosypose/cosypose/evaluation/prediction_runner.py b/happypose/pose_estimators/cosypose/cosypose/evaluation/prediction_runner.py index 5381826e..6b15f37c 100644 --- a/happypose/pose_estimators/cosypose/cosypose/evaluation/prediction_runner.py +++ b/happypose/pose_estimators/cosypose/cosypose/evaluation/prediction_runner.py @@ -122,6 +122,7 @@ def run_inference_pipeline( data_TCO_init=None, n_coarse_iterations=1, n_refiner_iterations=4, + detection_th=0.0 ) time.time() - t