Fix Teaserpp imports

agimus-project · Aug 24, 2023 · a6e5b9c · a6e5b9c
1 parent 664edf6
commit a6e5b9c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/happypose/pose_estimators/megapose/src/megapose/evaluation/evaluation.py b/happypose/pose_estimators/megapose/src/megapose/evaluation/evaluation.py
@@ -25,7 +25,7 @@
 
 # MegaPose
 import happypose
-import happypose.pose_estimators.megapose.src.megapose as megapose
+# import happypose.pose_estimators.megapose.src.megapose as megapose
 import happypose.pose_estimators.megapose.src.megapose.evaluation.evaluation_runner
 import happypose.toolbox.datasets.datasets_cfg
 import happypose.toolbox.inference.utils
@@ -55,7 +55,6 @@
 )
 from happypose.toolbox.datasets.datasets_cfg import make_object_dataset
 
-# from happypose.pose_estimators.megapose.src.megapose.inference.teaserpp_refiner import TeaserppRefiner
 from happypose.toolbox.lib3d.rigid_mesh_database import MeshDataBase
 from happypose.toolbox.utils.distributed import get_rank, get_tmp_dir
 from happypose.toolbox.utils.logging import get_logger
@@ -165,6 +164,7 @@ def run_eval(
         if cfg.inference.depth_refiner == "icp":
             depth_refiner: Optional[DepthRefiner] = ICPRefiner(mesh_db, renderer)
         elif cfg.inference.depth_refiner == "teaserpp":
+            from happypose.pose_estimators.megapose.src.megapose.inference.teaserpp_refiner import TeaserppRefiner
             depth_refiner = TeaserppRefiner(mesh_db, renderer)
         else:
             depth_refiner = None

diff --git a/happypose/pose_estimators/megapose/src/megapose/inference/teaserpp_refiner.py b/happypose/pose_estimators/megapose/src/megapose/inference/teaserpp_refiner.py
@@ -30,8 +30,8 @@
 from happypose.pose_estimators.megapose.src.megapose.inference.refiner_utils import compute_masks, numpy_to_open3d
 from happypose.pose_estimators.megapose.src.megapose.inference.types import PoseEstimatesType
 from happypose.toolbox.lib3d.rigid_mesh_database import BatchedMeshes
-from happypose.pose_estimators.megapose.src.megapose.panda3d_renderer.panda3d_batch_renderer import Panda3dBatchRenderer
-from happypose.pose_estimators.megapose.src.megapose.panda3d_renderer.types import Panda3dLightData
+from happypose.toolbox.renderer.panda3d_batch_renderer import Panda3dBatchRenderer
+from happypose.toolbox.renderer.types import Panda3dLightData
 from happypose.toolbox.visualization.meshcat_utils import get_pointcloud