minor fix to robot segmentation api
This commit is contained in:
@@ -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(
|
||||||
|
|||||||
Reference in New Issue
Block a user