MCPcopy Create free account

hub / github.com/anthonysimeonov/rpdiff / functions

Functions592 in github.com/anthonysimeonov/rpdiff

↓ 2 callersFunctionsimulate_deform_contact_point
(pcd, deform_about_point, rot_grid, uniform=False)
src/rpdiff/utils/pcd_aug_util.py:306
↓ 2 callersFunctionsingle_shot_regression_scene_combine
Directly predict the relative transformation in one shot
src/rpdiff/utils/relational_policy/inference_methods.py:1590
↓ 2 callersFunctionsingle_shot_regression_scene_combine_sc
Directly predict the relative transformation in one shot
src/rpdiff/utils/relational_policy/inference_methods.py:438
↓ 2 callersFunctionsparse_collate
r"""Create input arguments for a sparse tensor `the documentation <https://nvidia.github.io/MinkowskiEngine/sparse_tensor.html>`_. Convert a
src/rpdiff/utils/batch_pcd_util.py:79
↓ 2 callersFunctiontrain_iter_coarse_aff
( coarse_aff_mi: dict, coarse_aff_gt: dict, coarse_aff_model: nn.Module,
src/rpdiff/training/train_loops.py:23
↓ 2 callersMethodtransform
(self, position, orientation)
src/rpdiff/utils/pb2mc/pybullet_meshcat.py:46
↓ 2 callersFunctiontrimesh_combine
( mesh_files: List[str], mesh_poses: List[np.ndarray], mesh_scales: List[np.ndarray]=
src/rpdiff/utils/mesh_util/three_util.py:35
↓ 2 callersMethodvoxel_affordance
(self, x)
src/rpdiff/model/coarse_affordance.py:71
↓ 2 callersMethodwith_pos_embed
(self, tensor, pos: Optional[Tensor])
src/rpdiff/model/transformer/transformer.py:146
↓ 1 callersMethod__init__
(self, in_dim, hidden_dim, out_dim, n_layers)
src/rpdiff/utils/torch_util.py:25
↓ 1 callersMethod__init__
(self, n_pts: int, pn_pts: int=None, cn_pts: int=None, fixed_scaling: float=None)
src/rpdiff/model/policy_feat_encoder.py:17
↓ 1 callersMethod__init__
(self, s, pos_encoding='linear')
src/rpdiff/model/encoder/common.py:339
↓ 1 callersMethod_build_jnt_id
Build the mapping from the joint name to joint index.
src/rpdiff/robot/floating_sphere_gripper.py:145
↓ 1 callersFunction_compute_rotation_matrix
(angle_axis, theta2, eps=1e-6)
src/rpdiff/utils/torch_util.py:241
↓ 1 callersFunction_compute_rotation_matrix_taylor
(angle_axis)
src/rpdiff/utils/torch_util.py:265
↓ 1 callersFunction_copysign
Return a tensor where each element has the absolute value taken from the, corresponding element of a, with sign taken from the corresponding
src/rpdiff/utils/torch3d_util.py:73
↓ 1 callersFunction_model_class
(class_name)
src/rpdiff/model/encoder/unet3d.py:513
↓ 1 callersMethod_reset_parameters
(self)
src/rpdiff/model/transformer/transformer.py:42
↓ 1 callersMethod_setup
(self)
src/rpdiff/robot/floating_sphere_gripper.py:48
↓ 1 callersMethod_setup_cameras
Function to set up multiple pybullet cameras in the simulated environment
src/rpdiff/robot/multicam.py:77
↓ 1 callersMethod_setup_general
(self)
src/rpdiff/training/dataio_full_chunked.py:187
↓ 1 callersMethod_setup_joints
(self)
src/rpdiff/robot/floating_sphere_gripper.py:125
↓ 1 callersMethod_setup_local_pose_prediction
(self)
src/rpdiff/training/dataio_full_chunked.py:319
↓ 1 callersMethod_setup_pose_prediction
(self)
src/rpdiff/training/dataio_full_chunked.py:279
↓ 1 callersMethod_setup_rotation_grid
(self)
src/rpdiff/training/dataio_full_chunked.py:274
↓ 1 callersMethod_setup_success
(self)
src/rpdiff/training/dataio_full_chunked.py:332
↓ 1 callersMethod_setup_voxel_grid
(self)
src/rpdiff/training/dataio_full_chunked.py:255
↓ 1 callersFunction_sqrt_positive_part
Returns torch.sqrt(torch.max(0, x)) but with a zero subgradient where x is 0.
src/rpdiff/utils/torch3d_util.py:91
↓ 1 callersFunctionattention
(query, key, value, mask=None)
src/rpdiff/model/transformer/nsm_transformer.py:10
↓ 1 callersFunctionaxis_angle_to_quaternion
Convert rotations given as axis/angle to quaternions. Args: axis_angle: Rotations given as a vector in axis angle form,
src/rpdiff/utils/torch3d_util.py:483
↓ 1 callersMethodbuild_bce_logits
(self)
src/rpdiff/training/losses.py:15
↓ 1 callersFunctionbuild_default_int_mat
(height=480, width=640, fov_deg=60)
src/rpdiff/utils/pcd_aug_util.py:58
↓ 1 callersFunctionchamfer_distance_naive
Naive implementation of the Chamfer distance. Args: points1 (numpy array): first point set points2 (numpy array): second point s
src/rpdiff/model/encoder/common.py:54
↓ 1 callersMethodcheck_2d
Check if obj_pcd is inside the 2D projection of our collision geometry Args: obj_pcd (np.ndarray): N X 3 pointcloud array, to be
src/rpdiff/utils/point_cloud_fusion.py:770
↓ 1 callersMethodcheck_triangles
(self, points, triangles)
src/rpdiff/utils/mesh_util/inside_mesh.py:134
↓ 1 callersFunctioncoarse_aff_to_coarse_aff
( coarse_aff_mi, coarse_aff_mo, coarse_aff_gt, rot_grid, voxel_grid_pred, voxel_grid, voxel_r
src/rpdiff/training/pred_util.py:279
↓ 1 callersFunctioncoarse_aff_to_refine_pose
( coarse_aff_mi, coarse_aff_mo, coarse_aff_gt, refine_pose_mi, refine_pose_gt, rot_g
src/rpdiff/training/pred_util.py:53
↓ 1 callersFunctioncoarse_aff_to_success
( coarse_aff_mi, coarse_aff_mo, coarse_aff_gt, success_mi, success_gt, rot_grid, vo
src/rpdiff/training/pred_util.py:130
↓ 1 callersFunctioncompute_coverage
(out_poses_wf, avail_poses_wf, out_scores, mc_vis, t_thresh=0.035, t_thresh_prec=0.04, r_thresh_deg=5)
src/rpdiff/utils/relational_policy/filter_multimodal_placement.py:237
↓ 1 callersMethodcompute_intersection_depth
(self, points, triangles)
src/rpdiff/utils/mesh_util/inside_mesh.py:80
↓ 1 callersMethodconstraint_grasp_close
(self, obj_id)
src/rpdiff/robot/floating_sphere_gripper.py:629
↓ 1 callersMethodconstraint_grasp_open
(self, cid=None)
src/rpdiff/robot/floating_sphere_gripper.py:658
↓ 1 callersFunctionconv3d
(in_channels, out_channels, kernel_size, bias, padding=1)
src/rpdiff/model/encoder/unet3d.py:15
↓ 1 callersFunctioncreate_conv
Create a list of modules with together constitute a single conv layer with non-linearity and optional batchnorm/groupnorm. Args:
src/rpdiff/model/encoder/unet3d.py:19
↓ 1 callersFunctioncreate_target_desc_subdir
(demo_path, parent_model_path, child_model_path)
src/rpdiff/data_gen/rel_demo_aug/gen_rel_demos.py:94
↓ 1 callersMethoddecode
(self, model_input: dict, encoder_output: dict, des_full_parent_coords: torch.FloatTensor, des
src/rpdiff/model/transformer/policy.py:472
↓ 1 callersMethodencode
(self, model_input: dict, des_full_parent_coords: torch.FloatTensor, des_full_child_coords: to
src/rpdiff/model/transformer/policy.py:410
↓ 1 callersMethodencode
(self, src, src_mask)
src/rpdiff/model/transformer/nsm_transformer.py:27
↓ 1 callersFunctioneuler_angles_to_matrix
Convert rotations given as Euler angles in radians to rotation matrices. Args: euler_angles: Euler angles in radians as tensor of sh
src/rpdiff/utils/torch3d_util.py:185
↓ 1 callersMethodforward_post
(self, src, src_mask: Optional[Tensor] = None,
src/rpdiff/model/transformer/transformer.py:149
↓ 1 callersMethodforward_post
(self, tgt, memory, tgt_mask: Optional[Tensor] = None, memory_mask:
src/rpdiff/model/transformer/transformer.py:212
↓ 1 callersMethodforward_pre
(self, src, src_mask: Optional[Tensor] = None, src_key_padding_mask: O
src/rpdiff/model/transformer/transformer.py:164
↓ 1 callersMethodforward_pre
(self, tgt, memory, tgt_mask: Optional[Tensor] = None, memory_mask: Op
src/rpdiff/model/transformer/transformer.py:235
↓ 1 callersFunctionfull_registration_np
Run full global + local registration routine, using numpy array pointclouds as input Args: source_np (np.ndarray): Source pointcloud,
src/rpdiff/utils/point_cloud_fusion.py:78
↓ 1 callersMethodgenerate_grid_features
(self, index, c)
src/rpdiff/model/scene_encoder.py:124
↓ 1 callersMethodget_aff_input_gt
(self, data: dict, parent_final_pcd: np.ndarray, child_final_pcd: np.ndarray)
src/rpdiff/training/dataio_full_chunked.py:732
↓ 1 callersFunctionget_binary
(it, thresh, mq_int)
src/rpdiff/training/train_util.py:56
↓ 1 callersMethodget_clip_scores
(self, image_paths, viz=False)
src/rpdiff/utils/relational_policy/human_query.py:46
↓ 1 callersMethodget_diff_pose_input_gt
(self, data: dict, parent_final_pcd: np.ndarray, child_final_pcd: np.ndarray)
src/rpdiff/training/dataio_full_chunked.py:1249
↓ 1 callersMethodget_euler_onehot
(self, euler_angles: np.ndarray, bins_per_axis: int=None, return_closest_euler: bool
src/rpdiff/training/dataio_full_chunked.py:708
↓ 1 callersMethodget_keyframe
(self)
src/rpdiff/utils/pb2mc/pybullet_meshcat.py:52
↓ 1 callersMethodget_pose_input_gt
(self, data: dict, parent_final_pcd: np.ndarray, child_final_pcd: np.ndarray)
src/rpdiff/training/dataio_full_chunked.py:901
↓ 1 callersMethodget_pp_feats
(self, model_input: dict, *args, **kwargs)
src/rpdiff/model/transformer/policy.py:363
↓ 1 callersFunctionget_raster_points
(voxel_resolution: int, padding: float=0.0)
src/rpdiff/utils/mesh_util/three_util.py:61
↓ 1 callersFunctionget_raster_points
(voxel_resolution)
src/rpdiff/data_gen/model_gen/cube_occ_gen.py:34
↓ 1 callersFunctionget_raster_points
(voxel_resolution)
src/rpdiff/data_gen/model_gen/syn_bookshelf_gen.py:67
↓ 1 callersFunctionget_raster_points
(voxel_resolution)
src/rpdiff/data_gen/model_gen/container_occ_gen.py:34
↓ 1 callersFunctionget_raster_points
(voxel_resolution)
src/rpdiff/data_gen/model_gen/syn_rack_gen.py:68
↓ 1 callersFunctionget_raster_points
(voxel_resolution)
src/rpdiff/data_gen/model_gen/syn_container_gen.py:60
↓ 1 callersMethodget_rotation_grid_index
(self, rot_mat: np.ndarray, return_closest_mat: bool=False)
src/rpdiff/training/dataio_full_chunked.py:679
↓ 1 callersMethodget_sample_success
(self, data)
src/rpdiff/training/dataio_full_chunked.py:352
↓ 1 callersMethodget_success_input_gt
(self, data: dict, parent_final_pcd: np.ndarray, child_final_pcd: np.ndarray)
src/rpdiff/training/dataio_full_chunked.py:1117
↓ 1 callersFunctionget_syn_bookshelf_default_cfg
()
src/rpdiff/data_gen/model_gen/syn_bookshelf_cfg.py:22
↓ 1 callersFunctionget_syn_can_cabinet_default_cfg
()
src/rpdiff/data_gen/model_gen/syn_can_cabinet_cfg.py:24
↓ 1 callersFunctionget_syn_container_default_cfg
()
src/rpdiff/data_gen/model_gen/syn_container_cfg.py:12
↓ 1 callersFunctionget_syn_rack_default_cfg
()
src/rpdiff/data_gen/model_gen/syn_rack_cfg.py:54
↓ 1 callersFunctionget_transform
Find transform that transforms pose source to pose target :param pose_frame_target: :param pose_frame_source: :return:
src/rpdiff/utils/util.py:232
↓ 1 callersMethodget_worker_ready
(self, worker_id)
src/rpdiff/data_gen/rel_demo_aug/parallel_proc_gen_rel_demos.py:871
↓ 1 callersMethodgo_home
(self, ignore_physics=False)
src/rpdiff/robot/floating_sphere_gripper.py:164
↓ 1 callersMethodhide_hand
(self)
src/rpdiff/robot/floating_sphere_gripper.py:51
↓ 1 callersFunctionhide_link
(obj_id, link_id)
src/rpdiff/data_gen/rel_demo_aug/bookshelf_demos.py:60
↓ 1 callersFunctionhide_link
(obj_id, link_id)
src/rpdiff/data_gen/rel_demo_aug/can_cabinet_demos.py:59
↓ 1 callersFunctionhide_link
(obj_id, link_id)
src/rpdiff/data_gen/rel_demo_aug/gen_rel_demos.py:73
↓ 1 callersFunctionhide_link
(obj_id, link_id)
src/rpdiff/data_gen/rel_demo_aug/proc_gen_rel_demos.py:63
↓ 1 callersFunctionhide_link
(obj_id, link_id)
src/rpdiff/data_gen/rel_demo_aug/parallel_proc_gen_rel_demos.py:40
↓ 1 callersMethodin_poly_hull_multi
Function to check if any points in a 3D pointcloud fall inside of the convex hull of a set of 3D points. A convex hull for the collision
src/rpdiff/utils/point_cloud_fusion.py:796
↓ 1 callersMethodinit_gripper
(self, transform=np.eye(4))
src/rpdiff/utils/point_cloud_fusion.py:182
↓ 1 callersMethodinit_obj
(self, transform=np.eye(3))
src/rpdiff/utils/point_cloud_fusion.py:155
↓ 1 callersFunctionmain
(args: config_util.AttrDict)
src/rpdiff/data_gen/object_scene_procgen_demos.py:50
↓ 1 callersFunctionmain
(args)
src/rpdiff/data_gen/rel_demo_aug/bookshelf_demos.py:85
↓ 1 callersFunctionmain
(args)
src/rpdiff/data_gen/rel_demo_aug/can_cabinet_demos.py:84
↓ 1 callersFunctionmain
(args)
src/rpdiff/data_gen/rel_demo_aug/scene_proc_gen.py:81
↓ 1 callersFunctionmain
(args)
src/rpdiff/data_gen/rel_demo_aug/gen_rel_demos.py:107
↓ 1 callersFunctionmain
(args)
src/rpdiff/data_gen/rel_demo_aug/proc_gen_rel_demos.py:84
↓ 1 callersFunctionmain
(args)
src/rpdiff/data_gen/rel_demo_aug/parallel_proc_gen_rel_demos.py:892
↓ 1 callersFunctionmain
(args: config_util.AttrDict)
src/rpdiff/eval/evaluate_rpdiff.py:73
↓ 1 callersFunctionmain
(args: config_util.AttrDict)
src/rpdiff/training/train_full.py:299
↓ 1 callersFunctionmain_mp
(args)
src/rpdiff/data_gen/model_gen/cube_occ_gen.py:81
↓ 1 callersFunctionmain_mp
(args, cfg)
src/rpdiff/data_gen/model_gen/syn_bookshelf_gen.py:131
↓ 1 callersFunctionmain_mp
(args)
src/rpdiff/data_gen/model_gen/container_occ_gen.py:81
← previousnext →101–200 of 592, ranked by callers