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
# CuRobo
from curobo.geom.cv import (
get_projection_rays,
project_depth_using_rays,
)
from curobo.cuda_robot_model.cuda_robot_model import CudaRobotModel
from curobo.geom.cv import get_projection_rays, project_depth_using_rays
from curobo.geom.types import PointCloud
from curobo.types.base import TensorDeviceType
from curobo.types.camera import CameraObservation
@@ -189,6 +187,18 @@ class RobotSegmenter:
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
def mask_image(