minor fix to robot segmentation api

This commit is contained in:
Balakumar Sundaralingam
2024-02-23 10:07:36 -08:00
parent 6928cd415d
commit f25281e930

View File

@@ -17,10 +17,8 @@ import torch
from torch.profiler import record_function from torch.profiler import record_function
# CuRobo # CuRobo
from curobo.geom.cv import ( from curobo.cuda_robot_model.cuda_robot_model import CudaRobotModel
get_projection_rays, from curobo.geom.cv import get_projection_rays, project_depth_using_rays
project_depth_using_rays,
)
from curobo.geom.types import PointCloud from curobo.geom.types import PointCloud
from curobo.types.base import TensorDeviceType from curobo.types.base import TensorDeviceType
from curobo.types.camera import CameraObservation from curobo.types.camera import CameraObservation
@@ -189,6 +187,18 @@ class RobotSegmenter:
return mask, filtered_image return mask, filtered_image
@property
def kinematics(self) -> CudaRobotModel:
return self._robot_world.kinematics
@property
def robot_world(self) -> RobotWorld:
return self._robot_world
@property
def base_link(self) -> str:
return self.kinematics.base_link
@torch.jit.script @torch.jit.script
def mask_image( def mask_image(