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.
High-level APIs:
Motion Generation / Planning:
curobo.wrap.reacher.motion_gen.Inverse Kinematics:
curobo.wrap.reacher.ik_solver.Model Predictive Control:
curobo.wrap.reacher.mpc.Trajectory Optimization:
curobo.wrap.reacher.trajopt.
cuRobo package is split into several modules:
curobo.optcontains optimization solvers.curobo.cuda_robot_modelcontains robot kinematics.curobo.curobolibcontains the cuda kernels and python bindings for them.curobo.geomcontains geometry processing, collision checking and frame transforms.curobo.graphcontains geometric planning with graph search methods.curobo.rolloutcontains methods that map actions to costs. This class wraps instances ofcurobo.cuda_robot_modelandcurobo.geomto compute costs given trajectory of actions.curobo.utilcontains utility methods.curobo.wrapadds the user-level api for task programming. Includes implementation of collision-free reacher and batched robot world collision checking.curobo.typescontains custom dataclasses for common data types in robotics, includingJointState,CameraObservation,Pose.
Subpackages¶
- curobo.cuda_robot_model package
- Submodules
- curobo.cuda_robot_model.cuda_robot_generator module
- curobo.cuda_robot_model.cuda_robot_model module
- curobo.cuda_robot_model.kinematics_parser module
- curobo.cuda_robot_model.types module
- curobo.cuda_robot_model.urdf_kinematics_parser module
- curobo.cuda_robot_model.usd_kinematics_parser module
- curobo.cuda_robot_model.util module
- Submodules
- curobo.curobolib package
- Submodules
- curobo.curobolib.geom module
- curobo.curobolib.geom_cu module
- curobo.curobolib.kinematics module
- curobo.curobolib.kinematics_fused_cu module
- curobo.curobolib.lbfgs_step_cu module
- curobo.curobolib.line_search_cu module
- curobo.curobolib.ls module
- curobo.curobolib.opt module
- curobo.curobolib.tensor_step module
- curobo.curobolib.tensor_step_cu module
- curobo.curobolib.util_file module
- Submodules
- curobo.geom package
- Subpackages
- Submodules
- curobo.geom.cv module
- curobo.geom.sphere_fit module
- curobo.geom.transform module
transform_pointsbatch_transform_pointsget_inv_transformtransform_point_inversematrix_to_quaternioncuda_matrix_to_quaternionquaternion_to_matrixtorch_quaternion_to_matrixpose_to_matrixpose_multiplypose_inverseTransformPointBatchTransformPointBatchTransformPoseTransformPosePoseInverseQuatToMatrixMatrixToQuaternion
- curobo.geom.types module
- curobo.graph package
- curobo.opt package
- curobo.rollout package
- curobo.types package
- curobo.util package
- Submodules
- curobo.util.error_metrics module
- curobo.util.helpers module
- curobo.util.logger module
- curobo.util.metrics module
- curobo.util.sample_lib module
SampleConfigBaseSampleLibHaltonSampleLibbsplineKnotSampleLibRandomSampleLibSineSampleLibStompSampleLibSampleLibget_ranged_halton_samplesHaltonGeneratorbound_samplesgaussian_transformgenerate_noisegenerate_noise_npgenerate_prime_numbersgenerate_van_der_corput_samplegenerate_van_der_corput_samples_batchgenerate_halton_samplesgenerate_gaussian_halton_samplesgenerate_gaussian_sobol_samples
- curobo.util.state_filter module
- curobo.util.tensor_util module
- curobo.util.torch_utils module
find_first_idxfind_last_idxis_cuda_graph_availableis_cuda_graph_reset_availableis_torch_compile_availableget_torch_compile_optionsdisable_torch_compile_globalset_torch_compile_global_optionsget_torch_jit_decoratoris_lru_cache_avaiableget_cache_fn_decoratorempty_decoratorround_away_from_zero
- curobo.util.trajectory module
InterpolateTypeget_linear_trajget_smooth_trajectoryget_spline_interpolated_trajectoryget_batch_interpolated_trajectoryget_cpu_linear_interpolationget_cpu_kunz_stilman_interpolationget_interpolated_trajectorylinear_smoothcalculate_dt_fixedcalculate_dtcalculate_dt_no_clampcalculate_traj_stepscalculate_tsteps
- curobo.util.usd_helper module
set_prim_translateset_prim_transformget_prim_world_poseget_transformget_position_quatset_geom_mesh_attrsset_geom_cube_attrsset_geom_cylinder_attrsset_geom_sphere_attrsset_cylinder_attrsget_cylinder_attrsget_capsule_attrsget_cube_attrsget_sphere_attrsget_mesh_attrscreate_stageUsdHelper
- curobo.util.warp module
- curobo.util.warp_interpolation module
- curobo.util.xrdf_utils module
- Submodules
- curobo.wrap package
Submodules¶
- curobo.util_file module
get_module_pathget_root_pathget_content_pathget_configs_pathget_assets_pathget_weights_pathjoin_pathload_yamlwrite_yamlget_robot_pathget_task_configs_pathget_robot_configs_pathget_world_configs_pathget_debug_pathget_cpp_pathadd_cpp_pathcopy_file_to_pathget_filenameget_path_of_dirget_files_from_dirfile_existsget_motion_gen_robot_listget_robot_listget_multi_arm_robot_listmerge_dict_a_into_bis_platform_windowsis_platform_linuxis_file_xrdf