MCPcopy Create free account

hub / github.com/NVlabs/sage / functions

Functions7,169 in github.com/NVlabs/sage

⨍ Functions 7,169◇ Types & classes 1,266↳ Endpoints 1
↓ 4 callersMethodflush
Flush the episode data to disk.
IsaacLab/source/extensions/omni.isaac.lab/omni/isaac/lab/utils/datasets/hdf5_dataset_file_handler.py:177
↓ 4 callersMethodforward
Process each set of inputs in its own observation group. Args: inputs (dict): a dictionary of dictionaries with one dict
robomimic/robomimic/models/obs_nets.py:639
↓ 4 callersMethodgenerate_3d_from_text
Generate a 3D model from text input and return the task ID
server/isaacsim/isaac.sim.mcp_extension/isaac_sim_mcp_extension/gen3d.py:158
↓ 4 callersMethodgenerate_summary_report
Generate a text summary of the floor plan.
server/visualizer.py:405
↓ 4 callersFunctionget_door_window_placements
Get door and window placements as initial state for the solver. Uses actual wall thickness from the Wall model and properly maps doors/window
server/objects/object_placement_planner.py:1641
↓ 4 callersFunctionget_env_type
Helper function to get env_type from a variety of inputs. Args: env_meta (dict): environment metadata, which should be loaded from d
robomimic/robomimic/utils/env_utils.py:47
↓ 4 callersMethodget_episode
Returns the episode data for the given environment id. Args: env_id: The environment id. Returns: The episod
IsaacLab/source/extensions/omni.isaac.lab/omni/isaac/lab/managers/recorder_manager.py:237
↓ 4 callersFunctionget_full_view_camera_sampling
Sample camera position and lookat position to ensure the entire room is visible. Uses optimization to find the minimum radius that ensur
IsaacLab/source/extensions/omni.isaac.lab_tasks/omni/isaac/lab_tasks/manager_based/manipulation/mobile_manipulation_obj_scene_v3/config/franka/ik_abs_env_cfg.py:44
↓ 4 callersFunctionget_full_view_camera_sampling
Sample camera position and lookat position to ensure the entire room is visible. Uses optimization to find the minimum radius that ensur
IsaacLab/source/extensions/omni.isaac.lab_tasks/omni/isaac/lab_tasks/manager_based/manipulation/lift_single_obj_scene/config/franka/ik_abs_env_cfg.py:199
↓ 4 callersMethodget_last_layer
(self)
matfuse-sd/src/ldm/models/autoencoder.py:284
↓ 4 callersMethodget_last_layer
(self)
matfuse-sd/src/ldm/models/autoencoder.py:670
↓ 4 callersFunctionget_mask
(mask, mask_all, latent_shape)
matfuse-sd/src/utils/inference_helpers.py:86
↓ 4 callersFunctionget_movement_location_from_claude
Main dispatcher function that determines movement location type and calls appropriate handler. Currently only supports floor movement.
server/objects/object_movement_planner.py:3026
↓ 4 callersFunctionget_object_mesh
(source, source_id, layout_id)
server/objects/get_objects.py:246
↓ 4 callersMethodget_observation
Get current environment observation dictionary. Args: di (dict): current raw observation dictionary from robosuite to wr
robomimic/robomimic/envs/env_robosuite.py:241
↓ 4 callersFunctionget_or_create_collection
Get or create a collection
server/eval/eval_appearance.py:220
↓ 4 callersFunctionget_or_create_collection
Get or create a collection
server/eval/render_top_down_annotation.py:220
↓ 4 callersFunctionget_random_placements_on_target_object
( layout: FloorPlan, room: Room, target_object_id: str, object_to_place: Object, sample_count: int =
server/objects/object_on_top_placement.py:40
↓ 4 callersFunctionget_set_colors
()
M2T2/m2t2/plot_utils.py:22
↓ 4 callersFunctionget_textured_object_mesh_from_layout_id
(layout_id, room, room_id, object_id)
server/tex_utils.py:1618
↓ 4 callersFunctionhide_walls_based_on_camera
Hide/show walls based on camera view direction
server/eval/visualize_room_bpy.py:225
↓ 4 callersMethodinfer
(self, data, cfg)
M2T2/m2t2/m2t2.py:122
↓ 4 callersFunctioninfer_attributes_from_claude
(mesh_dict, caption=None)
server/objects/object_attribute_inference.py:256
↓ 4 callersFunctionis_ee_reach_goal
(ee_goal, ee_frame_data, env_i)
server/isaaclab/data_generation_grasp_aug_pose_v1_parallel.py:212
↓ 4 callersFunctionis_ee_reach_goal
(ee_goal, ee_frame_data, env_i)
server/isaaclab/data_generation_grasp_aug_pose_v2_parallel.py:197
↓ 4 callersFunctionis_ee_reach_goal
(ee_goal, ee_frame_data)
server/isaaclab/data_generation_grasp_aug_pose_v1.py:211
↓ 4 callersFunctionis_ee_reach_goal
(ee_goal, ee_frame_data)
server/isaaclab/data_generation_grasp_aug_pose.py:208
↓ 4 callersFunctionis_ee_reach_goal
(ee_goal, ee_frame_data, env_i)
server/isaaclab/exps_fig_teaser.py:215
↓ 4 callersFunctionis_ee_reach_goal
(ee_goal, ee_frame_data, env_i)
server/isaaclab/exps_fig_articulation.py:215
↓ 4 callersFunctionis_ee_reach_goal
(ee_goal, ee_frame_data, env_i)
server/isaaclab/data_generation_grasp_aug_pose_v2_on_aug_type_parallel.py:212
↓ 4 callersFunctionis_ee_reach_goal
(ee_goal, ee_frame_data, env_i)
server/isaaclab/data_generation_franka_pnp_from_layout_parsing.py:215
↓ 4 callersFunctionis_ee_reach_goal
(ee_goal, ee_frame_data, env_i)
server/isaaclab/data_generation_grasp_aug_pose_v2_parallel_vis.py:207
↓ 4 callersFunctionis_ee_reach_goal
(ee_goal, ee_frame_data, env_i)
server/isaaclab/data_generation_franka_pnp_type_aug_with_pose_aug.py:209
↓ 4 callersFunctionis_ee_reach_goal
(ee_goal, ee_frame_data, env_i)
server/isaaclab/data_generation_grasp_aug_pose_v2_on_aug_type_parallel_vis.py:212
↓ 4 callersFunctionis_ee_reach_goal
(ee_goal, ee_frame_data)
server/isaaclab/data_generation_grasp_v5.py:209
↓ 4 callersFunctionis_ee_reach_goal
(ee_goal, ee_frame_data)
server/isaaclab/data_generation_grasp_v6.py:237
↓ 4 callersMethodis_visible
Checks the visibility of the markers. Returns: True if the markers are visible, False otherwise.
IsaacLab/source/extensions/omni.isaac.lab/omni/isaac/lab/markers/visualization_markers.py:209
↓ 4 callersFunctionload_scene_meshes_into_blender
Load room layout meshes from files into Blender
server/eval/visualize_room_bpy.py:126
↓ 4 callersMethodlog_info
Process info dictionary from @train_on_batch to summarize information to pass to tensorboard for logging. Args:
robomimic/robomimic/algo/bc.py:216
↓ 4 callersFunctionlog_txt_as_img
(wh, xc, size=10)
matfuse-sd/src/ldm/util.py:12
↓ 4 callersFunctionmove_object
Main dispatcher function for moving objects. Currently only supports floor movement. Args: room: Target room current
server/objects/object_movement_planner.py:3057
↓ 4 callersFunctionnoise_like
(shape, device, repeat=False)
matfuse-sd/src/ldm/modules/diffusionmodules/util.py:291
↓ 4 callersFunctionnormalize
Normalizes a given input tensor to unit length. Args: x: Input tensor of shape (N, dims). eps: A small value to avoid division by
IsaacLab/source/extensions/omni.isaac.lab/omni/isaac/lab/utils/math.py:82
↓ 4 callersFunctionnormalize_angle
(angle)
IsaacLab/source/extensions/omni.isaac.lab_tasks/omni/isaac/lab_tasks/direct/cart_double_pendulum/cart_double_pendulum_env.py:191
↓ 4 callersMethodon_epoch_end
Called at the end of each epoch.
robomimic/robomimic/algo/cql.py:626
↓ 4 callersFunctionplace_objects
Place selected objects in a room using Claude API for intelligent placement. Args: selected_objects: List of objects to place
server/objects/object_placement_planner.py:126
↓ 4 callersMethodprocess_batch_for_training
Processes input batch from a data loader to filter out relevant information and prepare the batch for training. Args:
robomimic/robomimic/algo/gl.py:98
↓ 4 callersFunctionquat_from_euler_xyz
Convert rotations given as Euler angles in radians to Quaternions. Note: The euler angles are assumed in XYZ convention. Args:
IsaacLab/source/extensions/omni.isaac.lab/omni/isaac/lab/utils/math.py:252
↓ 4 callersFunctionquat_inv
Compute the inverse of a quaternion. Args: q: The quaternion orientation in (w, x, y, z). Shape is (N, 4). Returns: The inve
IsaacLab/source/extensions/omni.isaac.lab/omni/isaac/lab/utils/math.py:239
↓ 4 callersFunctionrandn_range
(shape, mean, std, dtype=torch.float32)
matfuse-sd/src/ldm/modules/losses/rendering.py:34
↓ 4 callersFunctionrasterize_mesh_dict_list_with_uv_efficient
(mesh_dict_list, projection, glctx, resolution, c2w=None)
server/nvdiffrast_rendering/render.py:202
↓ 4 callersFunctionrasterize_mesh_with_uv
(mesh_dict, projection, glctx, resolution, c2w=None)
server/nvdiffrast_rendering/render.py:94
↓ 4 callersMethodrecord_pre_reset
Record data at the beginning of env.reset() before reset is effective. Args: env_ids: The environment ids. All environments shoul
IsaacLab/source/extensions/omni.isaac.lab/omni/isaac/lab/managers/recorder_manager.py:78
↓ 4 callersMethodrender
( self, svbrdf, wi, wo, currentConeTargetPos, tensorboard="",
matfuse-sd/src/ldm/modules/losses/rendering.py:254
↓ 4 callersFunctionrender_room_four_edges_view
(layout: FloorPlan, room_id: str, resolution = 1024)
server/room_render.py:199
↓ 4 callersFunctionrepeat_texture
(texture_map_pil, times)
server/layout_wo_robot_multiroom.py:638
↓ 4 callersFunctionrepeat_texture
(texture_map_pil, times)
server/layout.py:635
↓ 4 callersFunctionrepeat_texture
(texture_map_pil, times)
server/layout_wo_robot_scene_aug.py:635
↓ 4 callersFunctionrepeat_texture
(texture_map_pil, times)
server/layout_wo_robot.py:635
↓ 4 callersMethodreset
Resets all the environments and returns observations. This function calls the :meth:`_reset_idx` function to reset all the environments.
IsaacLab/source/extensions/omni.isaac.lab/omni/isaac/lab/envs/direct_rl_env.py:251
↓ 4 callersMethodreset
Resets all the environments and returns observations. Args: seed: The seed to use for randomization. Defaults to None, in which c
IsaacLab/source/extensions/omni.isaac.lab/omni/isaac/lab/envs/direct_marl_env.py:286
↓ 4 callersMethodreset
(self, env_ids: Sequence[int])
IsaacLab/source/extensions/omni.isaac.lab/omni/isaac/lab/actuators/actuator_pd.py:128
↓ 4 callersFunctionreshape_dimensions_single
Reshape selected dimensions in a tensor to a target dimension. Args: x (torch.Tensor): tensor to reshape begin_axis (int): b
robomimic/robomimic/utils/tensor_utils.py:478
↓ 4 callersMethodsample
(self, batch_size=16, return_intermediates=False)
matfuse-sd/src/ldm/models/diffusion/ddpm.py:426
↓ 4 callersFunctionsave_chat_log_html
Convert client chat logs to HTML and save to file. Args: log_file: Path to the JSON log file output_file: Optional outpu
client/log.py:910
↓ 4 callersFunctionsave_images_grid
Save images in a grid with optional subtitles and title. Args: images: A list of images to be plotted. Shape of each image should be (H,
IsaacLab/source/standalone/demos/cameras.py:118
↓ 4 callersMethodsave_usd_with_ids
(self, usd_file_path, mesh_info_dict, room_base_ids)
server/isaacsim/isaac.sim.mcp_extension/isaac_sim_mcp_extension/extension.py:1412
↓ 4 callersMethodset_command
Set target end-effector pose command.
IsaacLab/source/extensions/omni.isaac.lab/omni/isaac/lab/controllers/rmp_flow.py:122
↓ 4 callersMethodset_gumbel_temperature
Used by external algorithms to schedule Gumbel-Softmax temperature, which is used during reparametrization at train-time. Should only
robomimic/robomimic/models/vae_nets.py:1371
↓ 4 callersMethodset_world_poses_from_view
Set the poses of the camera from the eye position and look-at target position. Args: eyes: The positions of the camera's eye. Sha
IsaacLab/source/extensions/omni.isaac.lab/omni/isaac/lab/sensors/ray_caster/ray_caster_camera.py:208
↓ 4 callersFunctionsetup_camera_look_at
Position camera and make it look at target position
server/eval/visualize_room_bpy.py:212
↓ 4 callersFunctionsplit_train_val_from_hdf5
Splits data into training set and validation set from HDF5 file. Args: hdf5_path (str): path to the hdf5 file to load th
robomimic/robomimic/scripts/split_train_val.py:26
↓ 4 callersMethodstart
Build and start the Docker container using the Docker compose command.
IsaacLab/docker/utils/container_interface.py:96
↓ 4 callersMethodstate
(self)
IsaacLab/source/extensions/omni.isaac.lab/omni/isaac/lab/envs/utils/marl.py:265
↓ 4 callersFunctiontest_example_script
Helper function to run an example script with filename @script_name and with test name @test_name (which will be printed to terminal with
robomimic/tests/test_examples.py:23
↓ 4 callersFunctiontest_repeated_objects_terrain
( difficulty: float, object_type: str, output_dir: str, headless: bool, provide_as_string: bool = False )
IsaacLab/source/extensions/omni.isaac.lab/test/terrains/check_mesh_subterrains.py:336
↓ 4 callersFunctiontrain
Train a model using the algorithm.
robomimic/robomimic/scripts/train.py:67
↓ 4 callersMethodtransform
Transform a USD model by applying position and scale. Args: prim (UsdGeom.Xformable, optional): The USD prim to
server/isaacsim/isaac.sim.mcp_extension/isaac_sim_mcp_extension/usd.py:184
↓ 4 callersFunctionunscale
(x, lower, upper)
IsaacLab/source/extensions/omni.isaac.lab_tasks/omni/isaac/lab_tasks/direct/shadow_hand_over/shadow_hand_over_env.py:415
↓ 4 callersMethodupdate
(self, dt: float)
IsaacLab/source/extensions/omni.isaac.lab/omni/isaac/lab/assets/articulation/articulation.py:201
↓ 4 callersFunctionvalidate_room_only_layout
Validate room-only layout for overlaps and connectivity issues. This is for the first stage of layout generation (rooms without doors/windows
server/validation.py:87
↓ 4 callersMethodvisualize_2d_render
Visualize the room in 2D render with overlay elements.
server/visualizer.py:849
↓ 4 callersMethodwrite_object_com_velocity_to_sim
Set the object center of mass velocity over selected environment and object indices into the simulation. Args: object_velocity: O
IsaacLab/source/extensions/omni.isaac.lab/omni/isaac/lab/assets/rigid_object_collection/rigid_object_collection.py:402
↓ 4 callersMethodwrite_root_com_state_to_sim
Set the root center of mass state over selected environment indices into the simulation. The root state comprises of the cartesian position,
IsaacLab/source/extensions/omni.isaac.lab/omni/isaac/lab/assets/rigid_object/rigid_object.py:164
↓ 4 callersFunctionzero_module
Zero out the parameters of a module and return it.
matfuse-sd/src/ldm/modules/diffusionmodules/util.py:200
↓ 3 callersFunctionSkrlVecEnvWrapper
Wraps around Isaac Lab environment for skrl. This function wraps around the Isaac Lab environment. Since the wrapping functionality is define
IsaacLab/source/extensions/omni.isaac.lab_tasks/omni/isaac/lab_tasks/utils/wrappers/skrl.py:39
↓ 3 callersMethod__init__
Args: embed_dim (int): dimension of embeddings to use for keys, queries, and values used in self-attention
robomimic/robomimic/models/transformers.py:315
↓ 3 callersMethod__init__
Args: input_shapes (OrderedDict): a dictionary that maps modality to expected shapes for all encoder-specific inp
robomimic/robomimic/models/vae_nets.py:774
↓ 3 callersMethod__init__
(self, embed_dim, hidden_dim, activation="ReLU")
M2T2/m2t2/contact_decoder.py:38
↓ 3 callersMethod__init__
( self, mlp, npoint=None, radius=None, nsample=None, bn=True, use_xyz=True )
M2T2/pointnet2_ops/pointnet2_ops/pointnet2_modules.py:135
↓ 3 callersMethod__post_init__
(self)
IsaacLab/source/extensions/omni.isaac.lab_tasks/omni/isaac/lab_tasks/manager_based/manipulation/inhand/config/allegro_hand/allegro_env_cfg.py:18
↓ 3 callersFunction_aggregate_traj_stats
Helper function to aggregate trajectory statistics. See https://en.wikipedia.org/wiki/Algorithms_for_calculating_variance#Parallel_algorithm
robomimic/robomimic/utils/dataset.py:825
↓ 3 callersMethod_auto_save_log
Auto-save the chat log periodically
client/client_generation_room_desc.py:713
↓ 3 callersMethod_auto_save_log
Auto-save the chat log periodically
client/client_generation.py:713
↓ 3 callersMethod_auto_save_log
Auto-save the chat log periodically
client/client_generation_scene_aug.py:732
↓ 3 callersMethod_auto_save_log
Auto-save the chat log periodically
client/client_generation_robot_task.py:764
↓ 3 callersFunction_build_object_id_map
(room: Room)
server/objects/object_augmentation.py:543
↓ 3 callersMethod_calculate_contact_length
Calculate the length of contact between two rectangles. Returns 0 if rectangles are not in contact.
server/room_solver.py:1448
↓ 3 callersMethod_check_epoch
Helper function to check whether backprop should happen this epoch. Args: net_name (str): name of network in @self.nets
robomimic/robomimic/algo/bcq.py:140
↓ 3 callersMethod_check_valid_tensor
Checks if given data does not have corrupted values. Args: data: Data buffer. Returns: True if the data is v
IsaacLab/source/extensions/omni.isaac.lab_tasks/test/wrappers/test_rsl_rl_wrapper.py:143
← previousnext →601–700 of 7,169, ranked by callers