MCPcopy Create free account

hub / github.com/NVlabs/GraspDataGen / functions

Functions312 in github.com/NVlabs/GraspDataGen

↓ 75 callersFunctionprint_blue
(*args, **kwargs)
scripts/graspgen/graspgen_utils.py:212
↓ 69 callersFunctionadd_arg_to_group
Add an argument to a specific group, creating the group if it doesn't exist. Args: group_name: Name of the group to add the argument
scripts/graspgen/graspgen_utils.py:126
↓ 61 callersFunctionprint_yellow
(*args, **kwargs)
scripts/graspgen/graspgen_utils.py:216
↓ 21 callersFunctionprint_red
(*args, **kwargs)
scripts/graspgen/graspgen_utils.py:220
↓ 20 callersMethodload
Load gripper state from disk. Args: config (GripperConfig): Configuration object containing necessary parameters
scripts/graspgen/gripper.py:216
↓ 19 callersFunctionprint_green
(*args, **kwargs)
scripts/graspgen/graspgen_utils.py:224
↓ 13 callersFunctionprint_purple
(*args, **kwargs)
scripts/graspgen/graspgen_utils.py:228
↓ 11 callersFunctionformat_value
Format a value for display, truncating large arrays.
scripts/graspgen/tools/compare_grippers.py:34
↓ 10 callersFunctionregister_argument_group
Register a new argument group with the global registry. Args: parser: The argparse parser to add the group to group_name: Nam
scripts/graspgen/graspgen_utils.py:91
↓ 8 callersFunctionget_simulation_app
Get the current simulation app instance, initializing it if needed. Args: file_name (str): Name of the file calling this function (e.
scripts/graspgen/graspgen_utils.py:347
↓ 7 callersFunctionadd_isaac_lab_args_if_needed
(parser)
scripts/graspgen/graspgen_utils.py:148
↓ 6 callersFunctionmat44_to_transform
(src: wp.mat44)
scripts/graspgen/warp_functions.py:32
↓ 6 callersFunctionstart_isaac_lab_if_needed
Start Isaac Lab if it is not already running. Args: file_name (str): Name of the file that started the lab headless (bool): W
scripts/graspgen/graspgen_utils.py:327
↓ 5 callersMethodfrom_args
(cls, args)
scripts/graspgen/gripper.py:200
↓ 5 callersMethodgen_seed
Generate a new seed that is reproducible but varied. Uses a hash of the original seed combined with a counter to ensure each
scripts/graspgen/grasp_guess.py:516
↓ 5 callersFunctionprint_color
Print text in colored format and reset to default color. This function accepts all the same arguments as the built-in print function.
scripts/graspgen/graspgen_utils.py:232
↓ 4 callersFunctionadd_gripper_args
(parser, param_dict, default_gripper_config=default_gripper_config,
scripts/graspgen/gripper.py:53
↓ 4 callersFunctionapply_gripper_configuration
Apply gripper configuration if specified in args. Args: args: The parsed arguments object Returns: bool: Tr
scripts/graspgen/gripper.py:89
↓ 4 callersFunctioncollect_gripper_args
(input_dict)
scripts/graspgen/gripper.py:35
↓ 4 callersMethodgenerate_grasps
(self, object, num_successes, num_fails)
scripts/graspgen/grasp_guess.py:805
↓ 4 callersMethodget_simulation_app
Get the simulation app, initializing it if needed. Returns: The simulation app instance or None if initialization failed
scripts/graspgen/graspgen_utils.py:298
↓ 4 callersFunctionopen_configuration_string_to_dict
(input_string)
scripts/graspgen/graspgen_utils.py:172
↓ 4 callersFunctiontriangle_mesh_intersect
(face: int, xform: wp.mat44, verts0: wp.array(dtype=wp.vec3), tris0: wp.array(dtyp
scripts/graspgen/warp_functions.py:63
↓ 3 callersFunctionadd_grasp_guess_args
(parser, param_dict, default_seed=default_seed, default_num_grasps=default_num_grasps, default_num_orientation
scripts/graspgen/grasp_guess.py:83
↓ 3 callersFunctionadd_grasp_sim_args
(parser, param_dict, default_grasp_file=default_grasp_file, default_max_num_envs=defau
scripts/graspgen/grasp_sim.py:75
↓ 3 callersFunctionapply_gripper_config_to_args
Apply gripper configuration overrides to the args object. Only overrides values that weren't explicitly provided on the command line.
scripts/graspgen/gripper_configurations.py:75
↓ 3 callersFunctioncollect_grasp_guess_args
(input_dict)
scripts/graspgen/grasp_guess.py:60
↓ 3 callersFunctioncollect_grasp_sim_args
(input_dict)
scripts/graspgen/grasp_sim.py:47
↓ 3 callersFunctioncompute_offset_along_negative_normal
(offset: float, xform: wp.mat44, open_axis: int)
scripts/graspgen/warp_functions.py:20
↓ 3 callersFunctionconvert_usd_to_obj
Convert a single USD file to OBJ format. Args: usd_path: Path to input USD file obj_path: Path to output OBJ file sca
scripts/graspgen/usd_to_obj_converter.py:166
↓ 3 callersMethodcreate_isaac_grasp_data
(self, grasp_sim_buffer, save_successes = True, save_fails = False, only_driven_joints = True, save_to_folder
scripts/graspgen/grasp_sim.py:558
↓ 3 callersFunctionget_gripper_config
Get gripper configuration based on gripper type string. Args: gripper_type (str): One of the predefined gripper types
scripts/graspgen/gripper_configurations.py:54
↓ 3 callersFunctionpredict_grasp_data_filepath
Predict the file path that would be generated by create_isaac_grasp_data functions without actually writing any data. Args:
scripts/graspgen/graspgen_utils.py:398
↓ 3 callersMethodvalidate_grasps
(self)
scripts/graspgen/grasp_sim.py:914
↓ 2 callersMethod_convert_yaml_to_json_format
Convert YAML grasp format to JSON format expected by load_grasps. YAML format has grasps as a dictionary with grasp names as
scripts/graspgen/grasp_guess.py:725
↓ 2 callersMethod_get_cache_filepath
Get the filepath for the debug cache.
scripts/graspgen/grasp_guess.py:883
↓ 2 callersFunction_visualize_object_mesh
Helper function to visualize the object mesh. Args: vis: Meshcat visualizer instance object: GuessObject instance col
scripts/graspgen/tools/visualize_tools.py:94
↓ 2 callersFunctionadd_create_gripper_args
(parser, param_dict, default_measure_convergence=DEFAULT_MEASURE_CONVERGENCE,
scripts/graspgen/graspgen_utils.py:461
↓ 2 callersFunctionadd_object_args
(parser, param_dict, default_object_file=default_object_file, default_
scripts/graspgen/object.py:43
↓ 2 callersFunctionanalyze_single_file
Analyze a single grasp file. Args: file_path: Path to the grasp file detailed: Whether to include detailed output
scripts/graspgen/tools/analyze_grasp_data.py:197
↓ 2 callersMethodcheck_gripper_body_collisions
Check every body except the fingers for collision against the object in transforms positions. is_invalid[i] set to Grasp.IN_COLLISION
scripts/graspgen/grasp_guess.py:593
↓ 2 callersFunctioncollect_create_gripper_args
Collect convergence-related arguments for gripper creation.
scripts/graspgen/graspgen_utils.py:449
↓ 2 callersFunctioncollect_object_args
(input_dict)
scripts/graspgen/object.py:26
↓ 2 callersFunctionconvert_mesh_to_usd
( usd_path: str, mesh_path: str, overwrite: bool = True, vertex_scale: Optional[float] = None
scripts/graspgen/mesh_utils.py:196
↓ 2 callersMethodcreate_gripper
(self, save_gripper=False, measure_convergence=DEFAULT_MEASURE_CONVERGENCE, convergence_iterations=DEFAULT_CON
scripts/graspgen/create_gripper_lab.py:225
↓ 2 callersFunctioncreate_gripper_with_lab
(gripper_config, save_gripper=True, measure_convergence=DEFAULT_MEASURE_CONVERGENCE, convergence_iterations=DE
scripts/graspgen/create_gripper_lab.py:484
↓ 2 callersMethodcreate_visualizer
Create and return a meshcat visualizer instance.
scripts/graspgen/tools/visualize_debug.py:82
↓ 2 callersFunctioncw_max
(a: wp.vec3, b: wp.vec3)
scripts/graspgen/warp_functions.py:28
↓ 2 callersFunctioncw_min
(a: wp.vec3, b: wp.vec3)
scripts/graspgen/warp_functions.py:24
↓ 2 callersFunctionextract_confidence_values
Extract grasp names and their confidence values.
scripts/graspgen/tools/compare_grasp_sims.py:51
↓ 2 callersFunctionextract_mesh_from_usd
Extract mesh data from a USD file and return as a trimesh object. Args: usd_path: Path to the USD file scale: Scale fact
scripts/graspgen/usd_to_obj_converter.py:49
↓ 2 callersMethodfrom_file
(cls, file_path, scale, args=None)
scripts/graspgen/object.py:86
↓ 2 callersMethodfrom_isaac_grasp_dict
(cls, grasp_dict, grasp_file_args=None)
scripts/graspgen/object.py:106
↓ 2 callersMethodget_open_limit_axes_and_finger_indices
(self, gripper)
scripts/graspgen/create_gripper_lab.py:195
↓ 2 callersMethodget_random_mesh_samples
(self, object, work_points, work_normals)
scripts/graspgen/grasp_guess.py:876
↓ 2 callersMethodgraspgen_json_to_transforms
Extract transforms and confidences from graspgen JSON format.
scripts/graspgen/tools/visualize_grasp_data.py:215
↓ 2 callersFunctionload_grasp_data
Load grasp data from YAML file and return grasps dict, object file, and gripper file.
scripts/graspgen/tools/compare_grasp_sims.py:35
↓ 2 callersFunctionload_npz_file
Load NPZ file and return as dictionary.
scripts/graspgen/tools/compare_grippers.py:25
↓ 2 callersMethodload_object_mesh
Load and scale an object mesh from file. Args: object_mesh_path: Path to the object mesh file object_scale: S
scripts/graspgen/tools/visualize_grasp_data.py:249
↓ 2 callersFunctionsave_yaml
Save grasps to a YAML file. Args: grasps: Dictionary of grasps to save output_file (str): Path to save the grasps to
scripts/graspgen/graspgen_utils.py:261
↓ 2 callersMethodset_transform_body_frame
(self, ref_idx)
scripts/graspgen/gripper.py:382
↓ 2 callersMethodshrink_to_actual_size
Shrink the GraspGuessBuffer to only contain the actual number of grasps found. Creates new arrays with the correct size and copies th
scripts/graspgen/grasp_guess.py:214
↓ 2 callersFunctiontransform_to_mat44
(src: wp.transform)
scripts/graspgen/warp_functions.py:41
↓ 2 callersMethodvisualize_debug_folder
Visualize all meshes from a debug folder. Args: folder_path: Path to debug folder folder_name: Optional name for grou
scripts/graspgen/tools/visualize_debug.py:248
↓ 2 callersFunctionwp_inverse_rigid_transform
(T: wp.mat44)
scripts/graspgen/warp_functions.py:105
↓ 2 callersFunctionwp_plane_transform_axis
Parameters ---------- origin : wp.vec3 The origin point normal : wp.vec3 The normal vector axis : int The
scripts/graspgen/warp_functions.py:150
↓ 2 callersFunctionwp_svd_normal
Given the a normalized vec3, return the U from normal.T = U*Sigma*V.T. Sigma and V.T are always 1 with a normal vector, so don't return those
scripts/graspgen/warp_functions.py:220
↓ 1 callersFunction_create_graspgen_config
Create the YAML configuration file for GraspGen.
scripts/graspgen/graspgen.py:169
↓ 1 callersMethod_quaternion_to_matrix
Convert quaternion and position to 4x4 transformation matrix.
scripts/graspgen/grasp_guess.py:765
↓ 1 callersMethod_setup_pvd_recording
Setup PVD recording (requires Isaac Lab to be started).
scripts/graspgen/grasp_sim.py:334
↓ 1 callersMethod_visualize_grasp
(self, vis: meshcat.Visualizer, name: str, transform: np.ndarray, color: List[float], grasp_vertices: List[np.
scripts/graspgen/tools/visualize_grasp_data.py:527
↓ 1 callersFunction_visualize_sample_normals
Helper function to visualize normals at sample points. Args: vis: Meshcat visualizer instance points: Numpy array of shape (N
scripts/graspgen/tools/visualize_tools.py:127
↓ 1 callersFunction_visualize_transform_frames
Helper function to visualize coordinate frames at transform locations. Args: vis: Meshcat visualizer instance transforms: Num
scripts/graspgen/tools/visualize_tools.py:306
↓ 1 callersFunctionadd_arg_if_not_exists
Add argument only if it doesn't already exist
scripts/graspgen/graspgen_utils.py:116
↓ 1 callersFunctionadd_grasp_display_args
(parser, _, default_grasp_file=default_grasp_file, default_max_num_grasps=default_max_num_grasps,
scripts/graspgen/grasp_display.py:45
↓ 1 callersFunctionadd_plot_args
(parser, param_dict)
scripts/graspgen/tools/plot_gripper_3d.py:50
↓ 1 callersFunctionanalyze_confidence_comparison
Analyze confidence values between two grasp sets.
scripts/graspgen/tools/compare_grasp_sims.py:60
↓ 1 callersFunctionanalyze_criteria
Analyze files against specified criteria. Args: files: List of file analysis results min_success: Minimum successful gra
scripts/graspgen/tools/analyze_grasp_data.py:287
↓ 1 callersFunctionanalyze_directory
Analyze all grasp files in a directory. Args: directory_path: Path to the directory detailed: Whether to include detaile
scripts/graspgen/tools/analyze_grasp_data.py:458
↓ 1 callersFunctionarrays_equal
Check if two arrays are equal, handling different dtypes and shapes.
scripts/graspgen/tools/compare_grippers.py:64
↓ 1 callersFunctionbalance_grasp_data
Load grasp data files from input folder and create balanced files in output folder. For files that contain grasp data (with 'grasps.transfor
scripts/graspgen/tools/balance_grasp_data.py:30
↓ 1 callersFunctionbatch_convert_usd_to_obj
Batch convert USD files to OBJ format based on a scales JSON file. Args: object_scales_json: Path to JSON file containing object
scripts/graspgen/usd_to_obj_converter.py:197
↓ 1 callersMethodbuild_grasp_display_scene_cfg
(self, num_envs)
scripts/graspgen/grasp_display.py:180
↓ 1 callersMethodbuild_grasp_sim_scene_cfg
(self, num_envs)
scripts/graspgen/grasp_sim.py:760
↓ 1 callersMethodbuild_scene_cfg
(self)
scripts/graspgen/create_gripper_lab.py:61
↓ 1 callersFunctioncalculate_statistics
Calculate comprehensive statistics from grasp data. Args: confidence_values: List of confidence values success_flags: Li
scripts/graspgen/tools/analyze_grasp_data.py:142
↓ 1 callersFunctioncalculate_validation_stats
Calculate validation statistics.
scripts/graspgen/graspgen.py:536
↓ 1 callersMethodcenter_finger_opening
(self, object, finger1_successes_work_transform, finger1_successes_offset, distances0, distances1, num_transfo
scripts/graspgen/grasp_guess.py:557
↓ 1 callersFunctioncheck_existing_grasp_file
Check if existing grasp file exists and determine if more grasps are needed.
scripts/graspgen/graspgen.py:333
↓ 1 callersFunctioncollect_grasp_display_args
(input_dict)
scripts/graspgen/grasp_display.py:30
↓ 1 callersFunctioncombine_existing_and_new_grasps
Combine existing grasps with new ones, filtering to only what's needed.
scripts/graspgen/graspgen.py:460
↓ 1 callersFunctioncompare_npz_files
Compare two NPZ files and display differences.
scripts/graspgen/tools/compare_grippers.py:136
↓ 1 callersFunctioncompare_values
Compare two values and return (is_equal, difference_description).
scripts/graspgen/tools/compare_grippers.py:91
↓ 1 callersFunctionconvert_to_relative_path
Convert absolute path to relative path from current working dir.
scripts/graspgen/tools/package_debug_bundle.py:143
↓ 1 callersFunctionconvert_usd_to_obj_direct
Convert a USD file to OBJ format using direct Python function calls. Args: usd_path: Path to the USD file scale: Scale f
scripts/graspgen/grasp_guess.py:341
↓ 1 callersFunctionconvert_yaml_to_json
Convert YAML grasp file to JSON graspgen format.
scripts/graspgen/tools/convert_yaml_to_json.py:55
↓ 1 callersFunctioncreate_and_plot_gripper
Create gripper and plot its body positions
scripts/graspgen/tools/plot_gripper_3d.py:180
↓ 1 callersFunctioncreate_debug_package
Create a debug package containing all necessary files. Args: yaml_file: Path to the input YAML file output_zip: Path to
scripts/graspgen/tools/package_debug_bundle.py:235
↓ 1 callersFunctioncreate_grasp_data_structure
Create the basic grasp data structure for JSON output.
scripts/graspgen/graspgen.py:515
↓ 1 callersFunctioncreate_graspgen_config_files
Create GraspGen configuration files for a gripper. This function generates two configuration files in the GraspGen format: 1. A YAML
scripts/graspgen/graspgen.py:88
next →1–100 of 312, ranked by callers