curobo package¶
cuRobo provides accelerated modules for robotics which can be used to build high-performance robotics applications. The library has several modules for numerical optimization, robot kinematics, geometry processing, collision checking, graph search planning. cuRobo provides high-level APIs for performing tasks like collision-free inverse kinematics, model predictive control, and motion planning.
Public API Quick Start:
from curobo.inverse_kinematics import InverseKinematics, InverseKinematicsCfg
from curobo.trajectory_optimizer import TrajectoryOptimizer, TrajectoryOptimizerCfg
from curobo.motion_planner import MotionPlanner, MotionPlannerCfg
from curobo.model_predictive_control import ModelPredictiveControl, ModelPredictiveControlCfg
from curobo.kinematics import Kinematics, KinematicsCfg
from curobo.scene import Scene, Cuboid, Sphere, Mesh
from curobo.types import JointState
# Forward kinematics
kin = Kinematics(KinematicsCfg.from_robot_yaml_file("franka.yml"))
js = JointState.from_position(q, joint_names=kin.joint_names)
state = kin.compute_kinematics(js)
# Scene representation
scene = Scene(
cuboid=[Cuboid(name="table", dims=[1, 1, 0.1], pose=[0, 0, 0.5, 1, 0, 0, 0])],
)
# Inverse kinematics
ik_config = InverseKinematicsCfg.create(robot="franka.yml")
ik = InverseKinematics(ik_config)
result = ik.solve_pose(goal_tool_poses=target_poses)
# Trajectory optimization
trajopt = TrajectoryOptimizer(trajopt_config)
trajectory = trajopt.solve_pose(goal_tool_poses=target_poses)
# Model predictive control
mpc = ModelPredictiveControl(mpc_config)
result = mpc.optimize_action_sequence(current_state)
Public API Modules:
curobo.kinematics- Forward kinematicscurobo.inverse_kinematics- Inverse kinematics solvercurobo.trajectory_optimizer- Trajectory optimizationcurobo.motion_planner- Motion planningcurobo.model_predictive_control- Model predictive controlcurobo.motion_retargeter- Motion retargeting using inverse kinematics and model predictive controlcurobo.scene- Scene representation with obstaclescurobo.collision_checking- Robot collision checking (for custom pipelines)curobo.perception- Perception utilities (robot segmentation)curobo.robot_builder- Build robot configs from URDFcurobo.viewer- Visualization (Rerun, Viser)curobo.types- Common data types (JointState, Pose, etc.)
Subpackages¶
Submodules¶
- curobo.batch_motion_planner module
BatchMotionPlannerBatchMotionPlanner.__init__BatchMotionPlanner.destroyBatchMotionPlanner.attachment_managerBatchMotionPlanner.batch_sizeBatchMotionPlanner.warmupBatchMotionPlanner.plan_poseBatchMotionPlanner.plan_cspaceBatchMotionPlanner.plan_graspBatchMotionPlanner.joint_namesBatchMotionPlanner.action_dimBatchMotionPlanner.tool_framesBatchMotionPlanner.default_joint_stateBatchMotionPlanner.kinematicsBatchMotionPlanner.compute_kinematicsBatchMotionPlanner.update_worldBatchMotionPlanner.clear_scene_cacheBatchMotionPlanner.reset_seedBatchMotionPlanner.enable_link_collisionBatchMotionPlanner.disable_link_collisionBatchMotionPlanner.update_tool_pose_criteriaBatchMotionPlanner.update_link_inertialBatchMotionPlanner.update_links_inertial
MotionPlannerCfg
- curobo.collision_checking module
- curobo.config_io module
- curobo.inverse_kinematics module
- curobo.kinematics module
KinematicsKinematics.__init__Kinematics.tool_framesKinematics.update_batch_sizeKinematics.compute_kinematicsKinematics.get_robot_link_meshesKinematics.get_robot_as_meshKinematics.get_robot_as_spheresKinematics.get_link_posesKinematics.all_articulated_joint_namesKinematics.get_self_collision_configKinematics.get_link_meshKinematics.get_link_transformKinematics.get_all_link_transformsKinematics.get_dofKinematics.dofKinematics.joint_namesKinematics.total_spheresKinematics.lock_jointstateKinematics.get_full_jsKinematics.get_mimic_jsKinematics.update_kinematics_configKinematics.get_active_jsKinematics.base_linkKinematics.robot_spheresKinematics.default_joint_positionKinematics.default_joint_stateKinematics.get_joint_limitsKinematics.kinematics_config
KinematicsCfgKinematicsCfg.device_cfgKinematicsCfg.tool_framesKinematicsCfg.kinematics_configKinematicsCfg.self_collision_configKinematicsCfg.kinematics_parserKinematicsCfg.generator_configKinematicsCfg.get_joint_limitsKinematicsCfg.from_basic_urdfKinematicsCfg.from_content_pathKinematicsCfg.from_robot_yaml_fileKinematicsCfg.from_config_fileKinematicsCfg.from_data_dictKinematicsCfg.from_configKinematicsCfg.cspaceKinematicsCfg.dofKinematicsCfg.__init__
KinematicsStateKinematicsState.tool_posesKinematicsState.tool_jacobiansKinematicsState.robot_spheresKinematicsState.robot_comKinematicsState.robot_collision_geometryKinematicsState.tool_framesKinematicsState.get_link_spheresKinematicsState.cloneKinematicsState.detachKinematicsState.copy_KinematicsState.__init__
- curobo.logging module
- curobo.model_predictive_control module
- curobo.motion_planner module
MotionPlannerMotionPlanner.__init__MotionPlanner.destroyMotionPlanner.attachment_managerMotionPlanner.joint_namesMotionPlanner.action_dimMotionPlanner.tool_framesMotionPlanner.default_joint_stateMotionPlanner.kinematicsMotionPlanner.compute_kinematicsMotionPlanner.warmupMotionPlanner.plan_poseMotionPlanner.plan_cspaceMotionPlanner.plan_graspMotionPlanner.enable_link_collisionMotionPlanner.disable_link_collisionMotionPlanner.update_worldMotionPlanner.clear_scene_cacheMotionPlanner.reset_seedMotionPlanner.update_link_inertialMotionPlanner.update_links_inertialMotionPlanner.update_tool_pose_criteria
MotionPlannerCfgGraspPlanResultGraspPlanResult.successGraspPlanResult.approach_successGraspPlanResult.grasp_successGraspPlanResult.lift_successGraspPlanResult.approach_trajectoryGraspPlanResult.approach_trajectory_dtGraspPlanResult.approach_interpolated_trajectoryGraspPlanResult.grasp_trajectoryGraspPlanResult.grasp_trajectory_dtGraspPlanResult.grasp_interpolated_trajectoryGraspPlanResult.lift_trajectoryGraspPlanResult.lift_trajectory_dtGraspPlanResult.lift_interpolated_trajectoryGraspPlanResult.approach_interpolated_last_tstepGraspPlanResult.grasp_interpolated_last_tstepGraspPlanResult.lift_interpolated_last_tstepGraspPlanResult.statusGraspPlanResult.planning_timeGraspPlanResult.goalset_indexGraspPlanResult.__init__
- curobo.motion_retargeter module
GoalToolPoseGoalToolPose.tool_framesGoalToolPose.positionGoalToolPose.quaternionGoalToolPose.batch_sizeGoalToolPose.horizonGoalToolPose.num_linksGoalToolPose.num_goalsetGoalToolPose.shapeGoalToolPose.ndimGoalToolPose.deviceGoalToolPose.from_posesGoalToolPose.get_link_poseGoalToolPose.to_dictGoalToolPose.copy_GoalToolPose.requires_grad_GoalToolPose.cloneGoalToolPose.detachGoalToolPose.reorder_linksGoalToolPose.__init__
MotionRetargeterMotionRetargeter.__init__MotionRetargeter.joint_namesMotionRetargeter.action_dimMotionRetargeter.tool_framesMotionRetargeter.kinematicsMotionRetargeter.default_joint_stateMotionRetargeter.num_dofMotionRetargeter.configMotionRetargeter.resetMotionRetargeter.solve_frameMotionRetargeter.solve_sequence
MotionRetargeterCfgMotionRetargeterCfg.robotMotionRetargeterCfg.tool_pose_criteriaMotionRetargeterCfg.num_envsMotionRetargeterCfg.use_mpcMotionRetargeterCfg.self_collision_checkMotionRetargeterCfg.scene_modelMotionRetargeterCfg.optimization_dtMotionRetargeterCfg.num_seeds_globalMotionRetargeterCfg.position_toleranceMotionRetargeterCfg.orientation_toleranceMotionRetargeterCfg.device_cfgMotionRetargeterCfg.load_collision_spheresMotionRetargeterCfg.ik_optimizer_configsMotionRetargeterCfg.mpc_optimizer_configsMotionRetargeterCfg.num_seeds_localMotionRetargeterCfg.num_control_pointsMotionRetargeterCfg.steps_per_targetMotionRetargeterCfg.velocity_regularization_weightMotionRetargeterCfg.acceleration_regularization_weightMotionRetargeterCfg.collision_activation_distanceMotionRetargeterCfg.global_ik_num_itersMotionRetargeterCfg.local_ik_num_itersMotionRetargeterCfg.mpc_warm_start_num_itersMotionRetargeterCfg.mpc_cold_start_num_itersMotionRetargeterCfg.tool_framesMotionRetargeterCfg.createMotionRetargeterCfg.__init__
RetargetResultSequenceGoalToolPoseSequenceGoalToolPose.tool_framesSequenceGoalToolPose.positionSequenceGoalToolPose.quaternionSequenceGoalToolPose.num_framesSequenceGoalToolPose.num_envsSequenceGoalToolPose.num_linksSequenceGoalToolPose.num_goalsetSequenceGoalToolPose.get_frameSequenceGoalToolPose.cloneSequenceGoalToolPose.deviceSequenceGoalToolPose.__init__
- curobo.optim module
EvolutionStrategiesEvolutionStrategies.__init__EvolutionStrategies.configEvolutionStrategies.device_cfgEvolutionStrategies.opt_dtEvolutionStrategies.use_cuda_graphEvolutionStrategies.enabledEvolutionStrategies.enableEvolutionStrategies.disableEvolutionStrategies.action_horizonEvolutionStrategies.action_dimEvolutionStrategies.opt_dimEvolutionStrategies.outer_itersEvolutionStrategies.horizonEvolutionStrategies.action_bound_lowsEvolutionStrategies.action_bound_highsEvolutionStrategies.action_step_maxEvolutionStrategies.action_horizon_bounds_lowsEvolutionStrategies.action_horizon_bounds_highsEvolutionStrategies.solve_timeEvolutionStrategies.solver_namesEvolutionStrategies.rollout_fnEvolutionStrategies.mean_actionEvolutionStrategies.best_trajEvolutionStrategies.cov_actionEvolutionStrategies.scale_trilEvolutionStrategies.inv_cov_actionEvolutionStrategies.full_scale_trilEvolutionStrategies.full_inv_covEvolutionStrategies.sample_libEvolutionStrategies.particles_per_problemEvolutionStrategies.sampled_particles_per_problemEvolutionStrategies.null_per_problemEvolutionStrategies.neg_per_problemEvolutionStrategies.total_num_particlesEvolutionStrategies.null_act_seqsEvolutionStrategies.problem_colEvolutionStrategies.top_trajsEvolutionStrategies.gamma_seqEvolutionStrategies.optimizeEvolutionStrategies.reinitializeEvolutionStrategies.shiftEvolutionStrategies.update_num_problemsEvolutionStrategies.update_rollout_paramsEvolutionStrategies.update_goal_dtEvolutionStrategies.get_all_rollout_instancesEvolutionStrategies.compute_metricsEvolutionStrategies.reset_shapeEvolutionStrategies.reset_seedEvolutionStrategies.reset_cuda_graphEvolutionStrategies.get_recorded_traceEvolutionStrategies.update_solver_paramsEvolutionStrategies.update_nitersEvolutionStrategies.debug_dumpEvolutionStrategies.update_seedEvolutionStrategies.update_init_meanEvolutionStrategies.get_rolloutsEvolutionStrategies.reset_distributionEvolutionStrategies.reset_meanEvolutionStrategies.reset_covarianceEvolutionStrategies.initialize_samplesEvolutionStrategies.update_samplesEvolutionStrategies.sample_actionsEvolutionStrategies.generate_noise
EvolutionStrategiesCfgLBFGSOptLBFGSOpt.__init__LBFGSOpt.configLBFGSOpt.device_cfgLBFGSOpt.opt_dtLBFGSOpt.use_cuda_graphLBFGSOpt.enabledLBFGSOpt.enableLBFGSOpt.disableLBFGSOpt.action_horizonLBFGSOpt.action_dimLBFGSOpt.opt_dimLBFGSOpt.outer_itersLBFGSOpt.horizonLBFGSOpt.action_bound_lowsLBFGSOpt.action_bound_highsLBFGSOpt.action_step_maxLBFGSOpt.action_horizon_step_maxLBFGSOpt.action_horizon_bounds_lowsLBFGSOpt.action_horizon_bounds_highsLBFGSOpt.solve_timeLBFGSOpt.solver_namesLBFGSOpt.rollout_fnLBFGSOpt.optimizeLBFGSOpt.reinitializeLBFGSOpt.shiftLBFGSOpt.update_num_problemsLBFGSOpt.update_rollout_paramsLBFGSOpt.update_goal_dtLBFGSOpt.get_all_rollout_instancesLBFGSOpt.compute_metricsLBFGSOpt.reset_shapeLBFGSOpt.reset_seedLBFGSOpt.reset_cuda_graphLBFGSOpt.get_recorded_traceLBFGSOpt.update_solver_paramsLBFGSOpt.update_nitersLBFGSOpt.debug_dump
LBFGSOptCfgLBFGSOptCfg.num_itersLBFGSOptCfg.solver_typeLBFGSOptCfg.solver_nameLBFGSOptCfg.device_cfgLBFGSOptCfg.store_debugLBFGSOptCfg.debug_infoLBFGSOptCfg.num_problemsLBFGSOptCfg.num_particlesLBFGSOptCfg.sync_cuda_timeLBFGSOptCfg.use_coo_sparseLBFGSOptCfg.step_scaleLBFGSOptCfg.inner_itersLBFGSOptCfg.cost_convergenceLBFGSOptCfg.cost_delta_thresholdLBFGSOptCfg.cost_relative_thresholdLBFGSOptCfg.converged_ratioLBFGSOptCfg.fixed_itersLBFGSOptCfg.convergence_iterationLBFGSOptCfg.minimum_itersLBFGSOptCfg.return_best_actionLBFGSOptCfg.line_search_scaleLBFGSOptCfg.line_search_typeLBFGSOptCfg.use_cuda_kernel_line_searchLBFGSOptCfg.fix_terminal_actionLBFGSOptCfg.line_search_wolfe_c_1LBFGSOptCfg.line_search_wolfe_c_2LBFGSOptCfg.historyLBFGSOptCfg.epsilonLBFGSOptCfg.use_cuda_kernel_step_directionLBFGSOptCfg.stable_modeLBFGSOptCfg.use_cuda_kernel_shared_buffersLBFGSOptCfg.initial_step_scaleLBFGSOptCfg.num_rollout_instancesLBFGSOptCfg.outer_itersLBFGSOptCfg.create_data_dictLBFGSOptCfg.update_nitersLBFGSOptCfg.__init__
MPPIMPPI.__init__MPPI.configMPPI.device_cfgMPPI.opt_dtMPPI.use_cuda_graphMPPI.enabledMPPI.enableMPPI.disableMPPI.action_horizonMPPI.action_dimMPPI.opt_dimMPPI.outer_itersMPPI.horizonMPPI.action_bound_lowsMPPI.action_bound_highsMPPI.action_step_maxMPPI.action_horizon_bounds_lowsMPPI.action_horizon_bounds_highsMPPI.solve_timeMPPI.solver_namesMPPI.rollout_fnMPPI.mean_actionMPPI.best_trajMPPI.cov_actionMPPI.scale_trilMPPI.inv_cov_actionMPPI.full_scale_trilMPPI.full_inv_covMPPI.sample_libMPPI.entropyMPPI.squashed_meanMPPI.particles_per_problemMPPI.sampled_particles_per_problemMPPI.null_per_problemMPPI.neg_per_problemMPPI.total_num_particlesMPPI.null_act_seqsMPPI.problem_colMPPI.top_trajsMPPI.gamma_seqMPPI.optimizeMPPI.reinitializeMPPI.shiftMPPI.update_num_problemsMPPI.update_rollout_paramsMPPI.update_goal_dtMPPI.get_all_rollout_instancesMPPI.compute_metricsMPPI.reset_shapeMPPI.reset_seedMPPI.reset_cuda_graphMPPI.get_recorded_traceMPPI.update_solver_paramsMPPI.update_nitersMPPI.debug_dumpMPPI.update_seedMPPI.update_init_meanMPPI.get_rolloutsMPPI.reset_distributionMPPI.reset_meanMPPI.reset_covarianceMPPI.initialize_samplesMPPI.update_samplesMPPI.sample_actionsMPPI.generate_noise
MPPICfgMPPICfg.num_itersMPPICfg.solver_typeMPPICfg.solver_nameMPPICfg.device_cfgMPPICfg.store_debugMPPICfg.debug_infoMPPICfg.num_problemsMPPICfg.num_particlesMPPICfg.sync_cuda_timeMPPICfg.use_coo_sparseMPPICfg.step_scaleMPPICfg.inner_itersMPPICfg.gammaMPPICfg.sample_modeMPPICfg.seedMPPICfg.store_rolloutsMPPICfg.null_act_fracMPPICfg.init_meanMPPICfg.init_covMPPICfg.base_actionMPPICfg.step_size_meanMPPICfg.step_size_covMPPICfg.squash_fnMPPICfg.cov_typeMPPICfg.sample_paramsMPPICfg.update_covMPPICfg.random_meanMPPICfg.betaMPPICfg.alphaMPPICfg.kappaMPPICfg.sample_per_problemMPPICfg.num_rollout_instancesMPPICfg.outer_itersMPPICfg.create_data_dictMPPICfg.update_nitersMPPICfg.__init__
MultiStageOptimizerMultiStageOptimizer.__init__MultiStageOptimizer.enabledMultiStageOptimizer.enableMultiStageOptimizer.disableMultiStageOptimizer.action_horizonMultiStageOptimizer.action_dimMultiStageOptimizer.opt_dimMultiStageOptimizer.outer_itersMultiStageOptimizer.solver_namesMultiStageOptimizer.solve_timeMultiStageOptimizer.optimizeMultiStageOptimizer.reinitializeMultiStageOptimizer.shiftMultiStageOptimizer.update_num_problemsMultiStageOptimizer.update_rollout_paramsMultiStageOptimizer.update_goal_dtMultiStageOptimizer.get_all_rollout_instancesMultiStageOptimizer.compute_metricsMultiStageOptimizer.reset_shapeMultiStageOptimizer.reset_seedMultiStageOptimizer.reset_cuda_graphMultiStageOptimizer.get_recorded_traceMultiStageOptimizer.update_solver_paramsMultiStageOptimizer.update_nitersMultiStageOptimizer.debug_dump
ScipyOptScipyOpt.__init__ScipyOpt.enabledScipyOpt.enableScipyOpt.disableScipyOpt.action_horizonScipyOpt.action_dimScipyOpt.opt_dimScipyOpt.outer_itersScipyOpt.horizonScipyOpt.action_bound_lowsScipyOpt.action_bound_highsScipyOpt.action_horizon_bounds_lowsScipyOpt.action_horizon_bounds_highsScipyOpt.solve_timeScipyOpt.solver_namesScipyOpt.optimizeScipyOpt.reinitializeScipyOpt.shiftScipyOpt.update_num_problemsScipyOpt.update_rollout_paramsScipyOpt.update_goal_dtScipyOpt.get_all_rollout_instancesScipyOpt.compute_metricsScipyOpt.reset_shapeScipyOpt.reset_seedScipyOpt.reset_cuda_graphScipyOpt.get_recorded_traceScipyOpt.update_solver_paramsScipyOpt.update_nitersScipyOpt.debug_dump
ScipyOptCfgScipyOptCfg.num_itersScipyOptCfg.solver_typeScipyOptCfg.solver_nameScipyOptCfg.device_cfgScipyOptCfg.store_debugScipyOptCfg.debug_infoScipyOptCfg.num_problemsScipyOptCfg.num_particlesScipyOptCfg.sync_cuda_timeScipyOptCfg.use_coo_sparseScipyOptCfg.step_scaleScipyOptCfg.inner_itersScipyOptCfg.scipy_minimize_methodScipyOptCfg.scipy_minimize_kwargsScipyOptCfg.use_float64_on_cpuScipyOptCfg.num_rollout_instancesScipyOptCfg.outer_itersScipyOptCfg.create_data_dictScipyOptCfg.update_nitersScipyOptCfg.__init__
TorchOptTorchOpt.__init__TorchOpt.enabledTorchOpt.enableTorchOpt.disableTorchOpt.action_horizonTorchOpt.action_dimTorchOpt.opt_dimTorchOpt.outer_itersTorchOpt.horizonTorchOpt.action_bound_lowsTorchOpt.action_bound_highsTorchOpt.solve_timeTorchOpt.solver_namesTorchOpt.optimizeTorchOpt.reinitializeTorchOpt.shiftTorchOpt.update_num_problemsTorchOpt.update_rollout_paramsTorchOpt.update_goal_dtTorchOpt.get_all_rollout_instancesTorchOpt.compute_metricsTorchOpt.reset_shapeTorchOpt.reset_seedTorchOpt.reset_cuda_graphTorchOpt.get_recorded_traceTorchOpt.update_solver_paramsTorchOpt.update_nitersTorchOpt.debug_dump
TorchOptCfgTorchOptCfg.num_itersTorchOptCfg.solver_typeTorchOptCfg.solver_nameTorchOptCfg.device_cfgTorchOptCfg.store_debugTorchOptCfg.debug_infoTorchOptCfg.num_problemsTorchOptCfg.num_particlesTorchOptCfg.sync_cuda_timeTorchOptCfg.use_coo_sparseTorchOptCfg.step_scaleTorchOptCfg.inner_itersTorchOptCfg.torch_optim_nameTorchOptCfg.torch_optim_kwargsTorchOptCfg.torch_optim_classTorchOptCfg.num_rollout_instancesTorchOptCfg.outer_itersTorchOptCfg.create_data_dictTorchOptCfg.update_nitersTorchOptCfg.__init__
- curobo.perception module
DetectorCfgDetectorCfg.n_mesh_points_coarseDetectorCfg.n_observed_points_coarseDetectorCfg.n_rotation_samplesDetectorCfg.n_iterations_coarseDetectorCfg.distance_threshold_coarseDetectorCfg.n_mesh_points_fineDetectorCfg.n_observed_points_fineDetectorCfg.n_iterations_fineDetectorCfg.distance_threshold_fineDetectorCfg.use_svdDetectorCfg.use_huber_lossDetectorCfg.huber_deltaDetectorCfg.save_iterationsDetectorCfg.device_cfgDetectorCfg.__init__
FilterDepthMapperMapper.__init__Mapper.integratorMapper.tsdfMapper.integrateMapper.clear_regionMapper.clear_blocksMapper.compute_esdfMapper.extract_meshMapper.extract_occupied_voxelsMapper.extract_matching_feature_voxelsMapper.get_matching_feature_voxelsMapper.resetMapper.get_statsMapper.memory_usage_mbMapper.update_static_obstaclesMapper.renderMapper.render_colorMapper.render_depthMapper.render_color_onlyMapper.render_shadedMapper.render_depth_colormapMapper.render_normal_colormap
MapperCfgMapperCfg.extent_meters_xyzMapperCfg.voxel_sizeMapperCfg.grid_centerMapperCfg.truncation_distanceMapperCfg.minimum_tsdf_weightMapperCfg.depth_minimum_distanceMapperCfg.depth_maximum_distanceMapperCfg.decay_factorMapperCfg.frustum_decay_factorMapperCfg.block_sizeMapperCfg.hash_load_factorMapperCfg.deviceMapperCfg.extent_meters_xyzMapperCfg.voxel_sizeMapperCfg.esdf_voxel_sizeMapperCfg.extent_esdf_meters_xyzMapperCfg.grid_centerMapperCfg.truncation_distanceMapperCfg.minimum_tsdf_weightMapperCfg.depth_minimum_distanceMapperCfg.depth_maximum_distanceMapperCfg.decay_factorMapperCfg.frustum_decay_factorMapperCfg.block_sizeMapperCfg.hash_load_factorMapperCfg.roughnessMapperCfg.seeding_methodMapperCfg.edt_solverMapperCfg.enable_staticMapperCfg.static_obstacle_colorMapperCfg.num_camerasMapperCfg.image_heightMapperCfg.image_widthMapperCfg.feature_dimMapperCfg.feature_grid_heightMapperCfg.feature_grid_widthMapperCfg.max_visible_blocks_per_integrationMapperCfg.max_support_pixels_per_block_cameraMapperCfg.feature_channels_per_threadMapperCfg.max_feature_tile_channelsMapperCfg.feature_integration_kernelMapperCfg.profile_integration_kernel_timingsMapperCfg.accumulator_w_maxMapperCfg.deviceMapperCfg.grid_shapeMapperCfg.max_blocksMapperCfg.hash_capacityMapperCfg.get_actual_extentMapperCfg.voxel_to_worldMapperCfg.world_to_voxelMapperCfg.__init__MapperCfg.get_grid_bounds
MatchedVoxelsOccupiedVoxelsPoseDetectorRobotMeshRobotMesh.__init__RobotMesh.from_trimeshRobotMesh.from_kinematicsRobotMesh.meshRobotMesh.mesh_idRobotMesh.verticesRobotMesh.facesRobotMesh.n_verticesRobotMesh.n_facesRobotMesh.is_articulatedRobotMesh.current_joint_anglesRobotMesh.updateRobotMesh.sample_surface_pointsRobotMesh.get_dofRobotMesh.get_trimesh
RobotSegmenterSDFDetectorCfgSDFDetectorCfg.max_iterationsSDFDetectorCfg.inner_iterationsSDFDetectorCfg.convergence_thresholdSDFDetectorCfg.rotation_convergence_thresholdSDFDetectorCfg.use_cuda_graphSDFDetectorCfg.distance_thresholdSDFDetectorCfg.min_valid_ratioSDFDetectorCfg.use_huberSDFDetectorCfg.huber_deltaSDFDetectorCfg.lambda_initialSDFDetectorCfg.lambda_factorSDFDetectorCfg.lambda_minSDFDetectorCfg.lambda_maxSDFDetectorCfg.rho_minSDFDetectorCfg.n_pointsSDFDetectorCfg.device_cfgSDFDetectorCfg.max_distanceSDFDetectorCfg.__init__
SDFPoseDetector
- curobo.profiling module
- curobo.robot_builder module
RobotBuilderRobotBuilder.__init__RobotBuilder.from_configRobotBuilder.fit_collision_spheresRobotBuilder.refit_link_spheresRobotBuilder.compute_collision_matrixRobotBuilder.add_collision_ignoreRobotBuilder.remove_collision_ignoreRobotBuilder.buildRobotBuilder.saveRobotBuilder.save_xrdfRobotBuilder.visualizeRobotBuilder.tool_framesRobotBuilder.collision_link_namesRobotBuilder.collision_spheresRobotBuilder.collision_matrixRobotBuilder.num_spheresRobotBuilder.link_metrics
RobotDebuggerRobotDebugger.__init__RobotDebugger.from_xrdfRobotDebugger.check_default_joint_configuration_collisionRobotDebugger.check_collision_at_configRobotDebugger.sample_collision_checksRobotDebugger.find_never_colliding_pairsRobotDebugger.visualize_collision_at_configRobotDebugger.print_collision_matrix_statsRobotDebugger.robot_configRobotDebugger.robot_model
- curobo.robot_parser module
UrdfRobotParserUrdfRobotParser.__init__UrdfRobotParser.build_link_parentUrdfRobotParser.get_link_parametersUrdfRobotParser.add_absolute_path_to_link_meshesUrdfRobotParser.get_urdf_stringUrdfRobotParser.get_link_geometryUrdfRobotParser.get_link_meshUrdfRobotParser.root_linkUrdfRobotParser.get_link_names_from_urdfUrdfRobotParser.get_joint_names_from_urdf
- curobo.rollout module
RosenbrockCfgRosenbrockRolloutRosenbrockRollout.__init__RosenbrockRollout.action_dimRosenbrockRollout.action_bound_lowsRosenbrockRollout.action_bound_highsRosenbrockRollout.action_boundsRosenbrockRollout.horizonRosenbrockRollout.action_horizonRosenbrockRollout.state_boundsRosenbrockRollout.batch_sizeRosenbrockRollout.dtRosenbrockRollout.evaluate_actionRosenbrockRollout.compute_metrics_from_stateRosenbrockRollout.compute_metrics_from_actionRosenbrockRollout.update_paramsRosenbrockRollout.update_batch_sizeRosenbrockRollout.update_dtRosenbrockRollout.resetRosenbrockRollout.reset_shapeRosenbrockRollout.reset_seedRosenbrockRollout.reset_cuda_graphRosenbrockRollout.sample_random_actionsRosenbrockRollout.get_initial_actionRosenbrockRollout.get_all_cost_components
- curobo.runtime module
- curobo.scene module
SceneSceneDataSceneData.cuboidsSceneData.meshesSceneData.voxelsSceneData.num_envsSceneData.device_cfgSceneData.scene_modelSceneData.get_valid_dataSceneData.create_cacheSceneData.from_scene_cfgSceneData.from_batch_scene_cfgSceneData.add_obstacleSceneData.update_obstacle_poseSceneData.enable_obstacleSceneData.get_obstacle_namesSceneData.check_obstacle_existsSceneData.clearSceneData.load_from_scene_cfgSceneData.has_cuboidsSceneData.has_meshesSceneData.has_voxelsSceneData.get_active_typesSceneData.to_warpSceneData.__init__
ObstacleObstacle.nameObstacle.poseObstacle.scaleObstacle.colorObstacle.texture_idObstacle.textureObstacle.materialObstacle.device_cfgObstacle.get_trimesh_meshObstacle.save_as_meshObstacle.get_cuboidObstacle.get_meshObstacle.get_transform_matrixObstacle.get_sphereObstacle.get_bounding_spheresObstacle.__init__
CuboidSphereCapsuleCylinderMeshVoxelGrid
- curobo.sphere_fit module
SphereFitMetricsSphereFitMetrics.num_spheresSphereFitMetrics.coverageSphereFitMetrics.protrusionSphereFitMetrics.protrusion_dist_meanSphereFitMetrics.protrusion_dist_p95SphereFitMetrics.surface_gap_meanSphereFitMetrics.surface_gap_p95SphereFitMetrics.max_uncovered_gapSphereFitMetrics.volume_ratioSphereFitMetrics.__init__
SphereFitResultSphereFitTypeestimate_sphere_countfit_spheres_to_mesh
- curobo.trajectory_optimizer module
- curobo.types module
JointStateJointState.positionJointState.velocityJointState.accelerationJointState.joint_namesJointState.jerkJointState.device_cfgJointState.dtJointState.aux_dataJointState.knotJointState.knot_dtJointState.control_spaceJointState.data_ptrJointState.deviceJointState.dtypeJointState.shapeJointState.ndimJointState.from_numpyJointState.from_positionJointState.from_state_tensorJointState.from_listJointState.zerosJointState.toJointState.cloneJointState.detachJointState.copy_referenceJointState.copy_JointState.copy_dataJointState.unsqueezeJointState.squeezeJointState.viewJointState.blendJointState.get_state_tensorJointState.stackJointState.catJointState.repeatJointState.repeat_seedsJointState.apply_kernelJointState.scaleJointState.scale_by_dtJointState.scale_timeJointState.calculate_fd_from_positionJointState.reorderJointState.reindexJointState.get_augmented_joint_stateJointState.append_jointsJointState.gather_by_seed_indexJointState.copy_only_indexJointState.copy_at_indexJointState.copy_at_batch_seed_indicesJointState.get_trajectory_at_horizon_indexJointState.trim_trajectoryJointState.index_dofJointState.__init__
RobotStateRobotState.joint_stateRobotState.joint_torqueRobotState.cuda_robot_model_stateRobotState.data_ptrRobotState.detachRobotState.robot_spheresRobotState.link_posesRobotState.tool_posesRobotState.tool_framesRobotState.copy_at_batch_seed_indicesRobotState.copy_only_indexRobotState.get_link_poseRobotState.cloneRobotState.copy_RobotState.__init__
PosePose.positionPose.quaternionPose.rotationPose.batch_sizePose.namePose.normalize_rotationPose.detachPose.requires_grad_Pose.batchPose.devicePose.ndimPose.from_matrixPose.from_euler_xyzPose.from_euler_xyz_intrinsicPose.get_rotation_matrixPose.get_rotationPose.stackPose.repeatPose.unsqueezePose.squeezePose.repeat_seedsPose.get_indexPose.apply_kernelPose.from_numpyPose.from_listPose.from_batch_listPose.to_listPose.tolistPose.clonePose.toPose.get_matrixPose.get_affine_matrixPose.get_numpy_affine_matrixPose.get_numpy_matrixPose.inversePose.get_pose_vectorPose.copy_Pose.catPose.distancePose.angular_distancePose.linear_distancePose.multiplyPose.transform_pointPose.transform_pointsPose.batch_transform_pointsPose.batch_transform_points_inversePose.shapePose.compute_offset_posePose.compute_local_posePose.contiguousPose.__init__
ToolPoseToolPose.tool_framesToolPose.positionToolPose.quaternionToolPose.batch_sizeToolPose.horizonToolPose.num_linksToolPose.shapeToolPose.ndimToolPose.deviceToolPose.get_link_poseToolPose.to_dictToolPose.copy_ToolPose.requires_grad_ToolPose.cloneToolPose.detachToolPose.contiguousToolPose.reorder_linksToolPose.as_goalToolPose.__init__
GoalToolPoseGoalToolPose.tool_framesGoalToolPose.positionGoalToolPose.quaternionGoalToolPose.batch_sizeGoalToolPose.horizonGoalToolPose.num_linksGoalToolPose.num_goalsetGoalToolPose.shapeGoalToolPose.ndimGoalToolPose.deviceGoalToolPose.from_posesGoalToolPose.get_link_poseGoalToolPose.to_dictGoalToolPose.copy_GoalToolPose.requires_grad_GoalToolPose.cloneGoalToolPose.detachGoalToolPose.reorder_linksGoalToolPose.__init__
ToolPoseCriteriaToolPoseCriteria.terminal_pose_axes_weight_factorToolPoseCriteria.non_terminal_pose_axes_weight_factorToolPoseCriteria.terminal_pose_convergence_toleranceToolPoseCriteria.non_terminal_pose_convergence_toleranceToolPoseCriteria.project_distance_to_goalToolPoseCriteria.device_cfgToolPoseCriteria.cloneToolPoseCriteria.copy_ToolPoseCriteria.track_positionToolPoseCriteria.track_orientationToolPoseCriteria.track_position_and_orientationToolPoseCriteria.linear_motionToolPoseCriteria.disabledToolPoseCriteria.__init__
CameraObservationCameraObservation.nameCameraObservation.rgb_imageCameraObservation.depth_imageCameraObservation.image_segmentationCameraObservation.projection_matrixCameraObservation.projection_raysCameraObservation.resolutionCameraObservation.poseCameraObservation.intrinsicsCameraObservation.timestampCameraObservation.depth_to_meterCameraObservation.feature_gridCameraObservation.filter_depthCameraObservation.shapeCameraObservation.copy_CameraObservation.cloneCameraObservation.toCameraObservation.get_pointcloudCameraObservation.extract_depth_from_structured_pointcloudCameraObservation.update_projection_raysCameraObservation.stackCameraObservation.save_to_fileCameraObservation.__init__
ContentPathContentPath.robot_config_root_pathContentPath.robot_xrdf_root_pathContentPath.robot_urdf_root_pathContentPath.robot_asset_root_pathContentPath.scene_config_root_pathContentPath.world_asset_root_pathContentPath.robot_config_absolute_pathContentPath.robot_xrdf_absolute_pathContentPath.robot_urdf_absolute_pathContentPath.robot_asset_absolute_pathContentPath.scene_config_absolute_pathContentPath.robot_config_fileContentPath.robot_xrdf_fileContentPath.robot_urdf_fileContentPath.robot_asset_subroot_pathContentPath.scene_config_fileContentPath.get_robot_configuration_pathContentPath.__init__
DeviceCfg
- curobo.viewer module