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_distancecompute_sdf_gradientSDFGridSDFGrid.forwardSDFGrid.backwardSDFGrid._backward_clsSDFGrid._compiled_autograd_backward_stateSDFGrid._compiled_autograd_keySDFGrid._get_compiled_autograd_symintsSDFGrid._input_metadataSDFGrid._is_compiled_autograd_tracingSDFGrid._materialize_non_diff_gradsSDFGrid._raw_saved_tensorsSDFGrid._register_hookSDFGrid._register_hook_dictSDFGrid._sequence_nrSDFGrid._set_sequence_nrSDFGrid.applySDFGrid.dirty_tensorsSDFGrid.generate_vmap_ruleSDFGrid.jvpSDFGrid.mark_dirtySDFGrid.mark_non_differentiableSDFGrid.mark_shared_storageSDFGrid.materialize_gradsSDFGrid.maybe_clear_saved_tensorsSDFGrid.metadataSDFGrid.nameSDFGrid.needs_input_gradSDFGrid.next_functionsSDFGrid.non_differentiableSDFGrid.register_hookSDFGrid.register_prehookSDFGrid.requires_gradSDFGrid.save_for_backwardSDFGrid.save_for_forwardSDFGrid.saved_for_forwardSDFGrid.saved_tensorsSDFGrid.saved_variablesSDFGrid.set_materialize_gradsSDFGrid.setup_contextSDFGrid.to_saveSDFGrid.vjpSDFGrid.vmap
- curobo.geom.sdf.utils module
- curobo.geom.sdf.warp_primitives module
SdfMeshWarpPySdfMeshWarpPy.forwardSdfMeshWarpPy.backwardSdfMeshWarpPy._backward_clsSdfMeshWarpPy._compiled_autograd_backward_stateSdfMeshWarpPy._compiled_autograd_keySdfMeshWarpPy._get_compiled_autograd_symintsSdfMeshWarpPy._input_metadataSdfMeshWarpPy._is_compiled_autograd_tracingSdfMeshWarpPy._materialize_non_diff_gradsSdfMeshWarpPy._raw_saved_tensorsSdfMeshWarpPy._register_hookSdfMeshWarpPy._register_hook_dictSdfMeshWarpPy._sequence_nrSdfMeshWarpPy._set_sequence_nrSdfMeshWarpPy.applySdfMeshWarpPy.dirty_tensorsSdfMeshWarpPy.generate_vmap_ruleSdfMeshWarpPy.jvpSdfMeshWarpPy.mark_dirtySdfMeshWarpPy.mark_non_differentiableSdfMeshWarpPy.mark_shared_storageSdfMeshWarpPy.materialize_gradsSdfMeshWarpPy.maybe_clear_saved_tensorsSdfMeshWarpPy.metadataSdfMeshWarpPy.nameSdfMeshWarpPy.needs_input_gradSdfMeshWarpPy.next_functionsSdfMeshWarpPy.non_differentiableSdfMeshWarpPy.register_hookSdfMeshWarpPy.register_prehookSdfMeshWarpPy.requires_gradSdfMeshWarpPy.save_for_backwardSdfMeshWarpPy.save_for_forwardSdfMeshWarpPy.saved_for_forwardSdfMeshWarpPy.saved_tensorsSdfMeshWarpPy.saved_variablesSdfMeshWarpPy.set_materialize_gradsSdfMeshWarpPy.setup_contextSdfMeshWarpPy.to_saveSdfMeshWarpPy.vjpSdfMeshWarpPy.vmap
SweptSdfMeshWarpPySweptSdfMeshWarpPy._backward_clsSweptSdfMeshWarpPy._compiled_autograd_backward_stateSweptSdfMeshWarpPy._compiled_autograd_keySweptSdfMeshWarpPy._get_compiled_autograd_symintsSweptSdfMeshWarpPy._input_metadataSweptSdfMeshWarpPy._is_compiled_autograd_tracingSweptSdfMeshWarpPy._materialize_non_diff_gradsSweptSdfMeshWarpPy._raw_saved_tensorsSweptSdfMeshWarpPy._register_hookSweptSdfMeshWarpPy._register_hook_dictSweptSdfMeshWarpPy._sequence_nrSweptSdfMeshWarpPy._set_sequence_nrSweptSdfMeshWarpPy.applySweptSdfMeshWarpPy.dirty_tensorsSweptSdfMeshWarpPy.forwardSweptSdfMeshWarpPy.generate_vmap_ruleSweptSdfMeshWarpPy.jvpSweptSdfMeshWarpPy.mark_dirtySweptSdfMeshWarpPy.mark_non_differentiableSweptSdfMeshWarpPy.mark_shared_storageSweptSdfMeshWarpPy.materialize_gradsSweptSdfMeshWarpPy.maybe_clear_saved_tensorsSweptSdfMeshWarpPy.metadataSweptSdfMeshWarpPy.nameSweptSdfMeshWarpPy.needs_input_gradSweptSdfMeshWarpPy.next_functionsSweptSdfMeshWarpPy.non_differentiableSweptSdfMeshWarpPy.register_hookSweptSdfMeshWarpPy.register_prehookSweptSdfMeshWarpPy.requires_gradSweptSdfMeshWarpPy.save_for_backwardSweptSdfMeshWarpPy.save_for_forwardSweptSdfMeshWarpPy.saved_for_forwardSweptSdfMeshWarpPy.saved_tensorsSweptSdfMeshWarpPy.saved_variablesSweptSdfMeshWarpPy.set_materialize_gradsSweptSdfMeshWarpPy.setup_contextSweptSdfMeshWarpPy.to_saveSweptSdfMeshWarpPy.vjpSweptSdfMeshWarpPy.vmapSweptSdfMeshWarpPy.backward
- curobo.geom.sdf.warp_sdf_fns module
- curobo.geom.sdf.warp_sdf_fns_deprecated module
- curobo.geom.sdf.world module
CollisionBufferCollisionQueryBufferCollisionQueryBuffer.primitive_collision_bufferCollisionQueryBuffer.mesh_collision_bufferCollisionQueryBuffer.blox_collision_bufferCollisionQueryBuffer.voxel_collision_bufferCollisionQueryBuffer.shapeCollisionQueryBuffer.cloneCollisionQueryBuffer.initialize_from_shapeCollisionQueryBuffer.create_from_shapeCollisionQueryBuffer.update_buffer_shapeCollisionQueryBuffer.get_gradient_buffer
CollisionCheckerTypeWorldCollisionConfigWorldCollisionWorldCollision.load_collision_modelWorldCollision.update_obstacle_pose_in_world_modelWorldCollision.get_sphere_distanceWorldCollision.get_sphere_collisionWorldCollision.get_swept_sphere_distanceWorldCollision.get_swept_sphere_collisionWorldCollision.get_voxels_in_bounding_boxWorldCollision.clear_voxelization_cacheWorldCollision.update_cache_voxelizationWorldCollision.get_occupancy_in_bounding_boxWorldCollision.get_esdf_in_bounding_boxWorldCollision.get_mesh_in_bounding_boxWorldCollision.get_obstacle_namesWorldCollision.check_obstacle_existsWorldCollision.cacheWorldCollision.checker_typeWorldCollision.load_from_dictWorldCollision.max_distanceWorldCollision.max_esdf_distanceWorldCollision.n_envsWorldCollision.world_modelWorldCollision.tensor_args
WorldPrimitiveCollisionWorldPrimitiveCollision._init_cacheWorldPrimitiveCollision.load_collision_modelWorldPrimitiveCollision.get_obstacle_namesWorldPrimitiveCollision.load_batch_collision_modelWorldPrimitiveCollision._load_collision_model_in_cacheWorldPrimitiveCollision._create_obb_cacheWorldPrimitiveCollision.add_obb_from_rawWorldPrimitiveCollision.add_obbWorldPrimitiveCollision.update_obb_dimsWorldPrimitiveCollision.enable_obstacleWorldPrimitiveCollision.enable_obbWorldPrimitiveCollision.update_obstacle_poseWorldPrimitiveCollision.update_obb_poseWorldPrimitiveCollision._get_obstacle_posesWorldPrimitiveCollision.get_obb_idxWorldPrimitiveCollision.get_sphere_distanceWorldPrimitiveCollision.get_sphere_collisionWorldPrimitiveCollision.get_swept_sphere_distanceWorldPrimitiveCollision.get_swept_sphere_collisionWorldPrimitiveCollision.clear_cacheWorldPrimitiveCollision.cacheWorldPrimitiveCollision.check_obstacle_existsWorldPrimitiveCollision.checker_typeWorldPrimitiveCollision.clear_voxelization_cacheWorldPrimitiveCollision.get_esdf_in_bounding_boxWorldPrimitiveCollision.get_mesh_in_bounding_boxWorldPrimitiveCollision.get_occupancy_in_bounding_boxWorldPrimitiveCollision.get_voxels_in_bounding_boxWorldPrimitiveCollision.load_from_dictWorldPrimitiveCollision.max_distanceWorldPrimitiveCollision.max_esdf_distanceWorldPrimitiveCollision.n_envsWorldPrimitiveCollision.update_cache_voxelizationWorldPrimitiveCollision.update_obstacle_pose_in_world_modelWorldPrimitiveCollision.world_modelWorldPrimitiveCollision.tensor_args
- curobo.geom.sdf.world_blox module
WorldBloxCollisionWorldBloxCollision.load_collision_modelWorldBloxCollision.clear_cacheWorldBloxCollision.clear_blox_layerWorldBloxCollision.get_sphere_distanceWorldBloxCollision.get_sphere_collisionWorldBloxCollision.get_swept_sphere_distanceWorldBloxCollision.get_swept_sphere_collisionWorldBloxCollision.enable_obstacleWorldBloxCollision.enable_bloxWorldBloxCollision.update_blox_poseWorldBloxCollision.clear_bounding_boxWorldBloxCollision.get_bounding_spheresWorldBloxCollision.add_camera_frameWorldBloxCollision.process_camera_framesWorldBloxCollision.update_blox_hashesWorldBloxCollision.update_blox_esdfWorldBloxCollision.update_blox_meshWorldBloxCollision.get_mesh_from_blox_layerWorldBloxCollision.save_layerWorldBloxCollision.decay_layerWorldBloxCollision.get_obstacle_namesWorldBloxCollision._get_blox_sdfWorldBloxCollision._get_blox_swept_sdfWorldBloxCollision._batch_tensor_voxelWorldBloxCollision._create_mesh_cacheWorldBloxCollision._create_obb_cacheWorldBloxCollision._create_voxel_cacheWorldBloxCollision._get_obstacle_posesWorldBloxCollision._get_sdfWorldBloxCollision._get_swept_sdfWorldBloxCollision._init_cacheWorldBloxCollision._load_batch_mesh_to_warpWorldBloxCollision._load_collision_model_in_cacheWorldBloxCollision._load_mesh_into_cacheWorldBloxCollision._load_mesh_to_warpWorldBloxCollision._load_voxel_collision_model_in_cacheWorldBloxCollision.add_meshWorldBloxCollision.add_obbWorldBloxCollision.add_obb_from_rawWorldBloxCollision.cacheWorldBloxCollision.check_obstacle_existsWorldBloxCollision.checker_typeWorldBloxCollision.clear_voxelization_cacheWorldBloxCollision.create_collision_cacheWorldBloxCollision.enable_meshWorldBloxCollision.enable_obbWorldBloxCollision.enable_voxelWorldBloxCollision.get_esdf_in_bounding_boxWorldBloxCollision.get_mesh_idxWorldBloxCollision.get_mesh_in_bounding_boxWorldBloxCollision.get_obb_idxWorldBloxCollision.get_occupancy_in_bounding_boxWorldBloxCollision.get_voxel_gridWorldBloxCollision.get_voxel_grid_shapeWorldBloxCollision.get_voxel_idxWorldBloxCollision.get_voxels_in_bounding_boxWorldBloxCollision.load_batch_collision_modelWorldBloxCollision.load_from_dictWorldBloxCollision.max_distanceWorldBloxCollision.max_esdf_distanceWorldBloxCollision.n_envsWorldBloxCollision.update_cache_voxelizationWorldBloxCollision.update_mesh_from_warpWorldBloxCollision.update_mesh_poseWorldBloxCollision.update_obb_dimsWorldBloxCollision.update_obb_poseWorldBloxCollision.update_obstacle_poseWorldBloxCollision.update_obstacle_pose_in_world_modelWorldBloxCollision.update_voxel_dataWorldBloxCollision.update_voxel_featuresWorldBloxCollision.update_voxel_poseWorldBloxCollision.world_modelWorldBloxCollision.tensor_args
- curobo.geom.sdf.world_mesh module
WarpMeshDataWorldMeshCollisionWorldMeshCollision.tensor_argsWorldMeshCollision._init_cacheWorldMeshCollision.load_collision_modelWorldMeshCollision.load_batch_collision_modelWorldMeshCollision._load_mesh_to_warpWorldMeshCollision._load_mesh_into_cacheWorldMeshCollision._load_batch_mesh_to_warpWorldMeshCollision.add_meshWorldMeshCollision.get_mesh_idxWorldMeshCollision.create_collision_cacheWorldMeshCollision._create_mesh_cacheWorldMeshCollision.update_mesh_poseWorldMeshCollision.update_mesh_from_warpWorldMeshCollision.update_obstacle_poseWorldMeshCollision.enable_obstacleWorldMeshCollision.get_obstacle_namesWorldMeshCollision.enable_meshWorldMeshCollision.get_sphere_distanceWorldMeshCollision.get_sphere_collisionWorldMeshCollision.get_swept_sphere_distanceWorldMeshCollision.get_swept_sphere_collisionWorldMeshCollision.clear_cacheWorldMeshCollision._get_sdfWorldMeshCollision._get_swept_sdfWorldMeshCollision._create_obb_cacheWorldMeshCollision._get_obstacle_posesWorldMeshCollision._load_collision_model_in_cacheWorldMeshCollision.add_obbWorldMeshCollision.add_obb_from_rawWorldMeshCollision.cacheWorldMeshCollision.check_obstacle_existsWorldMeshCollision.checker_typeWorldMeshCollision.clear_voxelization_cacheWorldMeshCollision.enable_obbWorldMeshCollision.get_esdf_in_bounding_boxWorldMeshCollision.get_mesh_in_bounding_boxWorldMeshCollision.get_obb_idxWorldMeshCollision.get_occupancy_in_bounding_boxWorldMeshCollision.get_voxels_in_bounding_boxWorldMeshCollision.load_from_dictWorldMeshCollision.max_distanceWorldMeshCollision.max_esdf_distanceWorldMeshCollision.n_envsWorldMeshCollision.update_cache_voxelizationWorldMeshCollision.update_obb_dimsWorldMeshCollision.update_obb_poseWorldMeshCollision.update_obstacle_pose_in_world_modelWorldMeshCollision.world_model
- curobo.geom.sdf.world_voxel module
WorldVoxelCollisionWorldVoxelCollision._init_cacheWorldVoxelCollision._create_voxel_cacheWorldVoxelCollision.load_collision_modelWorldVoxelCollision._load_voxel_collision_model_in_cacheWorldVoxelCollision._batch_tensor_voxelWorldVoxelCollision.load_batch_collision_modelWorldVoxelCollision.enable_obstacleWorldVoxelCollision.get_obstacle_namesWorldVoxelCollision.enable_voxelWorldVoxelCollision.update_obstacle_poseWorldVoxelCollision.update_voxel_dataWorldVoxelCollision.update_voxel_featuresWorldVoxelCollision.update_voxel_poseWorldVoxelCollision.get_voxel_idxWorldVoxelCollision.get_voxel_gridWorldVoxelCollision.get_sphere_distanceWorldVoxelCollision.get_sphere_collisionWorldVoxelCollision.get_swept_sphere_distanceWorldVoxelCollision.get_swept_sphere_collisionWorldVoxelCollision.clear_cacheWorldVoxelCollision.get_voxel_grid_shapeWorldVoxelCollision._create_mesh_cacheWorldVoxelCollision._create_obb_cacheWorldVoxelCollision._get_obstacle_posesWorldVoxelCollision._get_sdfWorldVoxelCollision._get_swept_sdfWorldVoxelCollision._load_batch_mesh_to_warpWorldVoxelCollision._load_collision_model_in_cacheWorldVoxelCollision._load_mesh_into_cacheWorldVoxelCollision._load_mesh_to_warpWorldVoxelCollision.add_meshWorldVoxelCollision.add_obbWorldVoxelCollision.add_obb_from_rawWorldVoxelCollision.cacheWorldVoxelCollision.check_obstacle_existsWorldVoxelCollision.checker_typeWorldVoxelCollision.clear_voxelization_cacheWorldVoxelCollision.create_collision_cacheWorldVoxelCollision.enable_meshWorldVoxelCollision.enable_obbWorldVoxelCollision.get_esdf_in_bounding_boxWorldVoxelCollision.get_mesh_idxWorldVoxelCollision.get_mesh_in_bounding_boxWorldVoxelCollision.get_obb_idxWorldVoxelCollision.get_occupancy_in_bounding_boxWorldVoxelCollision.get_voxels_in_bounding_boxWorldVoxelCollision.load_from_dictWorldVoxelCollision.max_distanceWorldVoxelCollision.max_esdf_distanceWorldVoxelCollision.n_envsWorldVoxelCollision.update_cache_voxelizationWorldVoxelCollision.update_mesh_from_warpWorldVoxelCollision.update_mesh_poseWorldVoxelCollision.update_obb_dimsWorldVoxelCollision.update_obb_poseWorldVoxelCollision.update_obstacle_pose_in_world_modelWorldVoxelCollision.world_modelWorldVoxelCollision.tensor_args