curobo.geom.sdf package¶
This module contains code for geometric processing such as pointcloud processing, analytic signed distance computation for the environment, and also signed distance computation between robot and the environment. These functions can be used for robot self collision checking and also for robot environment collision checking. The module also contains neural networks for learned signed distance fields.
Submodules¶
- curobo.geom.sdf.sdf_grid module
lookup_distance
compute_sdf_gradient
SDFGrid
SDFGrid.forward
SDFGrid.backward
SDFGrid._backward_cls
SDFGrid._compiled_autograd_backward_state
SDFGrid._compiled_autograd_key
SDFGrid._get_compiled_autograd_symints
SDFGrid._input_metadata
SDFGrid._is_compiled_autograd_tracing
SDFGrid._materialize_non_diff_grads
SDFGrid._raw_saved_tensors
SDFGrid._register_hook
SDFGrid._register_hook_dict
SDFGrid._sequence_nr
SDFGrid._set_sequence_nr
SDFGrid.apply
SDFGrid.dirty_tensors
SDFGrid.generate_vmap_rule
SDFGrid.jvp
SDFGrid.mark_dirty
SDFGrid.mark_non_differentiable
SDFGrid.mark_shared_storage
SDFGrid.materialize_grads
SDFGrid.maybe_clear_saved_tensors
SDFGrid.metadata
SDFGrid.name
SDFGrid.needs_input_grad
SDFGrid.next_functions
SDFGrid.non_differentiable
SDFGrid.register_hook
SDFGrid.register_prehook
SDFGrid.requires_grad
SDFGrid.save_for_backward
SDFGrid.save_for_forward
SDFGrid.saved_for_forward
SDFGrid.saved_tensors
SDFGrid.saved_variables
SDFGrid.set_materialize_grads
SDFGrid.setup_context
SDFGrid.to_save
SDFGrid.vjp
SDFGrid.vmap
- curobo.geom.sdf.utils module
- curobo.geom.sdf.warp_primitives module
SdfMeshWarpPy
SdfMeshWarpPy.forward
SdfMeshWarpPy.backward
SdfMeshWarpPy._backward_cls
SdfMeshWarpPy._compiled_autograd_backward_state
SdfMeshWarpPy._compiled_autograd_key
SdfMeshWarpPy._get_compiled_autograd_symints
SdfMeshWarpPy._input_metadata
SdfMeshWarpPy._is_compiled_autograd_tracing
SdfMeshWarpPy._materialize_non_diff_grads
SdfMeshWarpPy._raw_saved_tensors
SdfMeshWarpPy._register_hook
SdfMeshWarpPy._register_hook_dict
SdfMeshWarpPy._sequence_nr
SdfMeshWarpPy._set_sequence_nr
SdfMeshWarpPy.apply
SdfMeshWarpPy.dirty_tensors
SdfMeshWarpPy.generate_vmap_rule
SdfMeshWarpPy.jvp
SdfMeshWarpPy.mark_dirty
SdfMeshWarpPy.mark_non_differentiable
SdfMeshWarpPy.mark_shared_storage
SdfMeshWarpPy.materialize_grads
SdfMeshWarpPy.maybe_clear_saved_tensors
SdfMeshWarpPy.metadata
SdfMeshWarpPy.name
SdfMeshWarpPy.needs_input_grad
SdfMeshWarpPy.next_functions
SdfMeshWarpPy.non_differentiable
SdfMeshWarpPy.register_hook
SdfMeshWarpPy.register_prehook
SdfMeshWarpPy.requires_grad
SdfMeshWarpPy.save_for_backward
SdfMeshWarpPy.save_for_forward
SdfMeshWarpPy.saved_for_forward
SdfMeshWarpPy.saved_tensors
SdfMeshWarpPy.saved_variables
SdfMeshWarpPy.set_materialize_grads
SdfMeshWarpPy.setup_context
SdfMeshWarpPy.to_save
SdfMeshWarpPy.vjp
SdfMeshWarpPy.vmap
SweptSdfMeshWarpPy
SweptSdfMeshWarpPy._backward_cls
SweptSdfMeshWarpPy._compiled_autograd_backward_state
SweptSdfMeshWarpPy._compiled_autograd_key
SweptSdfMeshWarpPy._get_compiled_autograd_symints
SweptSdfMeshWarpPy._input_metadata
SweptSdfMeshWarpPy._is_compiled_autograd_tracing
SweptSdfMeshWarpPy._materialize_non_diff_grads
SweptSdfMeshWarpPy._raw_saved_tensors
SweptSdfMeshWarpPy._register_hook
SweptSdfMeshWarpPy._register_hook_dict
SweptSdfMeshWarpPy._sequence_nr
SweptSdfMeshWarpPy._set_sequence_nr
SweptSdfMeshWarpPy.apply
SweptSdfMeshWarpPy.dirty_tensors
SweptSdfMeshWarpPy.forward
SweptSdfMeshWarpPy.generate_vmap_rule
SweptSdfMeshWarpPy.jvp
SweptSdfMeshWarpPy.mark_dirty
SweptSdfMeshWarpPy.mark_non_differentiable
SweptSdfMeshWarpPy.mark_shared_storage
SweptSdfMeshWarpPy.materialize_grads
SweptSdfMeshWarpPy.maybe_clear_saved_tensors
SweptSdfMeshWarpPy.metadata
SweptSdfMeshWarpPy.name
SweptSdfMeshWarpPy.needs_input_grad
SweptSdfMeshWarpPy.next_functions
SweptSdfMeshWarpPy.non_differentiable
SweptSdfMeshWarpPy.register_hook
SweptSdfMeshWarpPy.register_prehook
SweptSdfMeshWarpPy.requires_grad
SweptSdfMeshWarpPy.save_for_backward
SweptSdfMeshWarpPy.save_for_forward
SweptSdfMeshWarpPy.saved_for_forward
SweptSdfMeshWarpPy.saved_tensors
SweptSdfMeshWarpPy.saved_variables
SweptSdfMeshWarpPy.set_materialize_grads
SweptSdfMeshWarpPy.setup_context
SweptSdfMeshWarpPy.to_save
SweptSdfMeshWarpPy.vjp
SweptSdfMeshWarpPy.vmap
SweptSdfMeshWarpPy.backward
- curobo.geom.sdf.warp_sdf_fns module
- curobo.geom.sdf.warp_sdf_fns_deprecated module
- curobo.geom.sdf.world module
CollisionBuffer
CollisionQueryBuffer
CollisionQueryBuffer.primitive_collision_buffer
CollisionQueryBuffer.mesh_collision_buffer
CollisionQueryBuffer.blox_collision_buffer
CollisionQueryBuffer.voxel_collision_buffer
CollisionQueryBuffer.shape
CollisionQueryBuffer.clone
CollisionQueryBuffer.initialize_from_shape
CollisionQueryBuffer.create_from_shape
CollisionQueryBuffer.update_buffer_shape
CollisionQueryBuffer.get_gradient_buffer
CollisionCheckerType
WorldCollisionConfig
WorldCollision
WorldCollision.load_collision_model
WorldCollision.update_obstacle_pose_in_world_model
WorldCollision.get_sphere_distance
WorldCollision.get_sphere_collision
WorldCollision.get_swept_sphere_distance
WorldCollision.get_swept_sphere_collision
WorldCollision.get_voxels_in_bounding_box
WorldCollision.clear_voxelization_cache
WorldCollision.update_cache_voxelization
WorldCollision.get_occupancy_in_bounding_box
WorldCollision.get_esdf_in_bounding_box
WorldCollision.get_mesh_in_bounding_box
WorldCollision.get_obstacle_names
WorldCollision.check_obstacle_exists
WorldCollision.cache
WorldCollision.checker_type
WorldCollision.load_from_dict
WorldCollision.max_distance
WorldCollision.max_esdf_distance
WorldCollision.n_envs
WorldCollision.world_model
WorldCollision.tensor_args
WorldPrimitiveCollision
WorldPrimitiveCollision._init_cache
WorldPrimitiveCollision.load_collision_model
WorldPrimitiveCollision.get_obstacle_names
WorldPrimitiveCollision.load_batch_collision_model
WorldPrimitiveCollision._load_collision_model_in_cache
WorldPrimitiveCollision._create_obb_cache
WorldPrimitiveCollision.add_obb_from_raw
WorldPrimitiveCollision.add_obb
WorldPrimitiveCollision.update_obb_dims
WorldPrimitiveCollision.enable_obstacle
WorldPrimitiveCollision.enable_obb
WorldPrimitiveCollision.update_obstacle_pose
WorldPrimitiveCollision.update_obb_pose
WorldPrimitiveCollision._get_obstacle_poses
WorldPrimitiveCollision.get_obb_idx
WorldPrimitiveCollision.get_sphere_distance
WorldPrimitiveCollision.get_sphere_collision
WorldPrimitiveCollision.get_swept_sphere_distance
WorldPrimitiveCollision.get_swept_sphere_collision
WorldPrimitiveCollision.clear_cache
WorldPrimitiveCollision.cache
WorldPrimitiveCollision.check_obstacle_exists
WorldPrimitiveCollision.checker_type
WorldPrimitiveCollision.clear_voxelization_cache
WorldPrimitiveCollision.get_esdf_in_bounding_box
WorldPrimitiveCollision.get_mesh_in_bounding_box
WorldPrimitiveCollision.get_occupancy_in_bounding_box
WorldPrimitiveCollision.get_voxels_in_bounding_box
WorldPrimitiveCollision.load_from_dict
WorldPrimitiveCollision.max_distance
WorldPrimitiveCollision.max_esdf_distance
WorldPrimitiveCollision.n_envs
WorldPrimitiveCollision.update_cache_voxelization
WorldPrimitiveCollision.update_obstacle_pose_in_world_model
WorldPrimitiveCollision.world_model
WorldPrimitiveCollision.tensor_args
- curobo.geom.sdf.world_blox module
WorldBloxCollision
WorldBloxCollision.load_collision_model
WorldBloxCollision.clear_cache
WorldBloxCollision.clear_blox_layer
WorldBloxCollision.get_sphere_distance
WorldBloxCollision.get_sphere_collision
WorldBloxCollision.get_swept_sphere_distance
WorldBloxCollision.get_swept_sphere_collision
WorldBloxCollision.enable_obstacle
WorldBloxCollision.enable_blox
WorldBloxCollision.update_blox_pose
WorldBloxCollision.clear_bounding_box
WorldBloxCollision.get_bounding_spheres
WorldBloxCollision.add_camera_frame
WorldBloxCollision.process_camera_frames
WorldBloxCollision.update_blox_hashes
WorldBloxCollision.update_blox_esdf
WorldBloxCollision.update_blox_mesh
WorldBloxCollision.get_mesh_from_blox_layer
WorldBloxCollision.save_layer
WorldBloxCollision.decay_layer
WorldBloxCollision.get_obstacle_names
WorldBloxCollision._get_blox_sdf
WorldBloxCollision._get_blox_swept_sdf
WorldBloxCollision._batch_tensor_voxel
WorldBloxCollision._create_mesh_cache
WorldBloxCollision._create_obb_cache
WorldBloxCollision._create_voxel_cache
WorldBloxCollision._get_obstacle_poses
WorldBloxCollision._get_sdf
WorldBloxCollision._get_swept_sdf
WorldBloxCollision._init_cache
WorldBloxCollision._load_batch_mesh_to_warp
WorldBloxCollision._load_collision_model_in_cache
WorldBloxCollision._load_mesh_into_cache
WorldBloxCollision._load_mesh_to_warp
WorldBloxCollision._load_voxel_collision_model_in_cache
WorldBloxCollision.add_mesh
WorldBloxCollision.add_obb
WorldBloxCollision.add_obb_from_raw
WorldBloxCollision.cache
WorldBloxCollision.check_obstacle_exists
WorldBloxCollision.checker_type
WorldBloxCollision.clear_voxelization_cache
WorldBloxCollision.create_collision_cache
WorldBloxCollision.enable_mesh
WorldBloxCollision.enable_obb
WorldBloxCollision.enable_voxel
WorldBloxCollision.get_esdf_in_bounding_box
WorldBloxCollision.get_mesh_idx
WorldBloxCollision.get_mesh_in_bounding_box
WorldBloxCollision.get_obb_idx
WorldBloxCollision.get_occupancy_in_bounding_box
WorldBloxCollision.get_voxel_grid
WorldBloxCollision.get_voxel_grid_shape
WorldBloxCollision.get_voxel_idx
WorldBloxCollision.get_voxels_in_bounding_box
WorldBloxCollision.load_batch_collision_model
WorldBloxCollision.load_from_dict
WorldBloxCollision.max_distance
WorldBloxCollision.max_esdf_distance
WorldBloxCollision.n_envs
WorldBloxCollision.update_cache_voxelization
WorldBloxCollision.update_mesh_from_warp
WorldBloxCollision.update_mesh_pose
WorldBloxCollision.update_obb_dims
WorldBloxCollision.update_obb_pose
WorldBloxCollision.update_obstacle_pose
WorldBloxCollision.update_obstacle_pose_in_world_model
WorldBloxCollision.update_voxel_data
WorldBloxCollision.update_voxel_features
WorldBloxCollision.update_voxel_pose
WorldBloxCollision.world_model
WorldBloxCollision.tensor_args
- curobo.geom.sdf.world_mesh module
WarpMeshData
WorldMeshCollision
WorldMeshCollision.tensor_args
WorldMeshCollision._init_cache
WorldMeshCollision.load_collision_model
WorldMeshCollision.load_batch_collision_model
WorldMeshCollision._load_mesh_to_warp
WorldMeshCollision._load_mesh_into_cache
WorldMeshCollision._load_batch_mesh_to_warp
WorldMeshCollision.add_mesh
WorldMeshCollision.get_mesh_idx
WorldMeshCollision.create_collision_cache
WorldMeshCollision._create_mesh_cache
WorldMeshCollision.update_mesh_pose
WorldMeshCollision.update_mesh_from_warp
WorldMeshCollision.update_obstacle_pose
WorldMeshCollision.enable_obstacle
WorldMeshCollision.get_obstacle_names
WorldMeshCollision.enable_mesh
WorldMeshCollision.get_sphere_distance
WorldMeshCollision.get_sphere_collision
WorldMeshCollision.get_swept_sphere_distance
WorldMeshCollision.get_swept_sphere_collision
WorldMeshCollision.clear_cache
WorldMeshCollision._get_sdf
WorldMeshCollision._get_swept_sdf
WorldMeshCollision._create_obb_cache
WorldMeshCollision._get_obstacle_poses
WorldMeshCollision._load_collision_model_in_cache
WorldMeshCollision.add_obb
WorldMeshCollision.add_obb_from_raw
WorldMeshCollision.cache
WorldMeshCollision.check_obstacle_exists
WorldMeshCollision.checker_type
WorldMeshCollision.clear_voxelization_cache
WorldMeshCollision.enable_obb
WorldMeshCollision.get_esdf_in_bounding_box
WorldMeshCollision.get_mesh_in_bounding_box
WorldMeshCollision.get_obb_idx
WorldMeshCollision.get_occupancy_in_bounding_box
WorldMeshCollision.get_voxels_in_bounding_box
WorldMeshCollision.load_from_dict
WorldMeshCollision.max_distance
WorldMeshCollision.max_esdf_distance
WorldMeshCollision.n_envs
WorldMeshCollision.update_cache_voxelization
WorldMeshCollision.update_obb_dims
WorldMeshCollision.update_obb_pose
WorldMeshCollision.update_obstacle_pose_in_world_model
WorldMeshCollision.world_model
- curobo.geom.sdf.world_voxel module
WorldVoxelCollision
WorldVoxelCollision._init_cache
WorldVoxelCollision._create_voxel_cache
WorldVoxelCollision.load_collision_model
WorldVoxelCollision._load_voxel_collision_model_in_cache
WorldVoxelCollision._batch_tensor_voxel
WorldVoxelCollision.load_batch_collision_model
WorldVoxelCollision.enable_obstacle
WorldVoxelCollision.get_obstacle_names
WorldVoxelCollision.enable_voxel
WorldVoxelCollision.update_obstacle_pose
WorldVoxelCollision.update_voxel_data
WorldVoxelCollision.update_voxel_features
WorldVoxelCollision.update_voxel_pose
WorldVoxelCollision.get_voxel_idx
WorldVoxelCollision.get_voxel_grid
WorldVoxelCollision.get_sphere_distance
WorldVoxelCollision.get_sphere_collision
WorldVoxelCollision.get_swept_sphere_distance
WorldVoxelCollision.get_swept_sphere_collision
WorldVoxelCollision.clear_cache
WorldVoxelCollision.get_voxel_grid_shape
WorldVoxelCollision._create_mesh_cache
WorldVoxelCollision._create_obb_cache
WorldVoxelCollision._get_obstacle_poses
WorldVoxelCollision._get_sdf
WorldVoxelCollision._get_swept_sdf
WorldVoxelCollision._load_batch_mesh_to_warp
WorldVoxelCollision._load_collision_model_in_cache
WorldVoxelCollision._load_mesh_into_cache
WorldVoxelCollision._load_mesh_to_warp
WorldVoxelCollision.add_mesh
WorldVoxelCollision.add_obb
WorldVoxelCollision.add_obb_from_raw
WorldVoxelCollision.cache
WorldVoxelCollision.check_obstacle_exists
WorldVoxelCollision.checker_type
WorldVoxelCollision.clear_voxelization_cache
WorldVoxelCollision.create_collision_cache
WorldVoxelCollision.enable_mesh
WorldVoxelCollision.enable_obb
WorldVoxelCollision.get_esdf_in_bounding_box
WorldVoxelCollision.get_mesh_idx
WorldVoxelCollision.get_mesh_in_bounding_box
WorldVoxelCollision.get_obb_idx
WorldVoxelCollision.get_occupancy_in_bounding_box
WorldVoxelCollision.get_voxels_in_bounding_box
WorldVoxelCollision.load_from_dict
WorldVoxelCollision.max_distance
WorldVoxelCollision.max_esdf_distance
WorldVoxelCollision.n_envs
WorldVoxelCollision.update_cache_voxelization
WorldVoxelCollision.update_mesh_from_warp
WorldVoxelCollision.update_mesh_pose
WorldVoxelCollision.update_obb_dims
WorldVoxelCollision.update_obb_pose
WorldVoxelCollision.update_obstacle_pose_in_world_model
WorldVoxelCollision.world_model
WorldVoxelCollision.tensor_args