MCPcopy Create free account

hub / github.com/XuAdventurer/PIN-WM / functions

Functions473 in github.com/XuAdventurer/PIN-WM

↓ 1 callersMethodload_gaussian
(self,gaussian_path)
builder.py:43
↓ 1 callersFunctionload_video_dataset
(data_args)
utils/data_utils.py:74
↓ 1 callersFunctionlpips
r"""Function that measures Learned Perceptual Image Patch Similarity (LPIPS). Arguments: x, y (torch.Tensor): the input tensors to co
diff_rendering/gaussian_splatting_2d/lpipsPyTorch/__init__.py:6
↓ 1 callersFunctionmake_output_format
return a logger for the requested format :param _format: the requested format to log to ('stdout', 'log', 'json' or 'csv' or 'tensorboard')
utils/sys_utils.py:362
↓ 1 callersFunctionmarching_cubes_with_contraction
( sdf, resolution=512, bounding_box_min=(-1.0, -1.0, -1.0), bounding_box_max=(1.0, 1.0, 1.0),
diff_rendering/gaussian_splatting_2d/utils/mcube_utils.py:17
↓ 1 callersFunctionmean_psnr
(images,gt_images)
utils/metric_utils.py:5
↓ 1 callersFunctionmkdir_p
(folder_path)
diff_rendering/gaussian_splatting_2d/utils/system_utils.py:16
↓ 1 callersMethodmu
(self,contact_infos)
diff_simulation/solver/lcp_solver.py:109
↓ 1 callersMethodnormal2quaternion
(self,normals)
diff_rendering/gaussian_splatting_2d/scene/mesh_gaussian_model.py:71
↓ 1 callersFunctionnormalize
r"""Normalizes a quaternion to unit norm. Args: quaternion (torch.Tensor): Quaternion to normalize (shape: :math:`(4)`) (Assu
utils/quaternion_utils.py:35
↓ 1 callersFunctionnormalize_activation
(x, eps=1e-10)
diff_rendering/gaussian_splatting_2d/lpipsPyTorch/modules/utils.py:6
↓ 1 callersMethodoneupSHdegree
(self)
diff_rendering/gaussian_splatting_2d/scene/gaussian_model.py:146
↓ 1 callersFunctionorthogonal
Get any orthogonal vector in 3D space
diff_simulation/solver/util.py:126
↓ 1 callersFunctionparse_args
()
scripts/collect_2dgs_data.py:47
↓ 1 callersFunctionparse_args
()
scripts/train_2dgs.py:51
↓ 1 callersFunctionparse_args
()
scripts/collect_video_data.py:47
↓ 1 callersFunctionpost_process_mesh
Post-process a mesh to filter out floaters and disconnected parts
diff_rendering/gaussian_splatting_2d/utils/mesh_utils.py:22
↓ 1 callersFunctionquaternion_conjugate
(q)
utils/quaternion_utils.py:5
↓ 1 callersFunctionrasterize_gaussians
( means3D, means2D, sh, colors_precomp, opacities, scales, rotations, cov3Ds_p
diff_rendering/gaussian_splatting_2d/submodules/diff-surfel-rasterization/diff_surfel_rasterization/__init__.py:21
↓ 1 callersFunctionread
()
diff_rendering/gaussian_splatting_2d/gaussian_renderer/network_gui.py:43
↓ 1 callersFunctionreadColmapCameras
(cam_extrinsics, cam_intrinsics, images_folder)
diff_rendering/gaussian_splatting_2d/scene/dataset_readers.py:70
↓ 1 callersFunctionreadImages
(renders_dir, gt_dir)
diff_rendering/gaussian_splatting_2d/metrics.py:24
↓ 1 callersFunctionread_extrinsics_binary
see: src/base/reconstruction.cc void Reconstruction::ReadImagesBinary(const std::string& path) void Reconstruction::WriteImagesBi
diff_rendering/gaussian_splatting_2d/scene/colmap_loader.py:180
↓ 1 callersFunctionread_extrinsics_text
Taken from https://github.com/colmap/colmap/blob/dev/scripts/python/read_write_model.py
diff_rendering/gaussian_splatting_2d/scene/colmap_loader.py:244
↓ 1 callersFunctionread_intrinsics_binary
see: src/base/reconstruction.cc void Reconstruction::WriteCamerasBinary(const std::string& path) void Reconstruction::ReadCameras
diff_rendering/gaussian_splatting_2d/scene/colmap_loader.py:215
↓ 1 callersFunctionread_intrinsics_text
Taken from https://github.com/colmap/colmap/blob/dev/scripts/python/read_write_model.py
diff_rendering/gaussian_splatting_2d/scene/colmap_loader.py:156
↓ 1 callersFunctionread_points3D_binary
see: src/base/reconstruction.cc void Reconstruction::ReadPoints3DBinary(const std::string& path) void Reconstruction::WritePoints
diff_rendering/gaussian_splatting_2d/scene/colmap_loader.py:125
↓ 1 callersFunctionread_points3D_text
see: src/base/reconstruction.cc void Reconstruction::ReadPoints3DText(const std::string& path) void Reconstruction::WritePoints3D
diff_rendering/gaussian_splatting_2d/scene/colmap_loader.py:83
↓ 1 callersFunctionrender
Render the scene. Background tensor (bg_color) must be on GPU!
diff_rendering/gaussian_splatting_2d/gaussian_renderer/__init__.py:19
↓ 1 callersFunctionrendering_loss_batch
(images,gt_images)
diff_rendering/gaussian_splatting_2d/utils/loss_utils.py:75
↓ 1 callersMethodreplace_tensor_to_optimizer
(self, tensor, name)
diff_rendering/gaussian_splatting_2d/scene/gaussian_model.py:288
↓ 1 callersMethodreset
(self)
diff_simulation/body/base.py:68
↓ 1 callersMethodreset_opacity
(self)
diff_rendering/gaussian_splatting_2d/scene/gaussian_model.py:240
↓ 1 callersMethodreset_position_rotation
(self,world_position,world_rotation)
diff_rendering/gaussian_splatting_2d/scene/gaussian_model.py:564
↓ 1 callersMethodrestitutions
(self,contact_infos)
diff_simulation/solver/lcp_solver.py:97
↓ 1 callersMethodrestore
(self, model_args, training_args)
diff_rendering/gaussian_splatting_2d/scene/gaussian_model.py:101
↓ 1 callersFunctionsave_data
(images,image_datas,ee_goal_position,pose_datas,data_path,data_folder_name,json_name)
utils/data_utils.py:152
↓ 1 callersMethodsave_disk
(self,disk_path,f)
diff_rendering/gaussian_splatting_2d/scene/gaussian_model.py:516
↓ 1 callersMethodsave_ply
(self, point_cloud_path)
diff_rendering/gaussian_splatting_2d/scene/gaussian_model.py:218
↓ 1 callersFunctionsave_video_data
(images,image_datas,ee_init_position,ee_goal_position,pose_datas,data_path)
utils/data_utils.py:173
↓ 1 callersMethodset_cam_pos
(self, cam_pos)
envs/data_collect/camera.py:38
↓ 1 callersMethodset_physical_parameters
(self,body_id,physical_parameters)
diff_simulation/simulator.py:236
↓ 1 callersMethodset_seed
(self,seed)
builder.py:33
↓ 1 callersMethodsetup_functions
(self)
diff_rendering/gaussian_splatting_2d/scene/gaussian_model.py:33
↓ 1 callersMethodshow
(self, body_id, color)
envs/base.py:55
↓ 1 callersMethodsolve_constraint
(self,contact_infos)
diff_simulation/solver/lcp_solver.py:122
↓ 1 callersMethodsolve_lcp_no_contact
(self,inv,u)
diff_simulation/solver/lcp_solver.py:148
↓ 1 callersMethodsolve_lcp_with_contact
(self,eps=1e-12, verbose=0, notImprovedLim=3, max_iter=20, solver=1,
diff_simulation/solver/lcp_solver.py:180
↓ 1 callersMethodstep_simulation
(self)
envs/base.py:41
↓ 1 callersMethodstep_video_motion
(self, ee_goal_position)
envs/data_collect/data_collect_env.py:83
↓ 1 callersFunctionstrip_lowerdiag
(L)
diff_rendering/gaussian_splatting_2d/utils/general_utils.py:64
↓ 1 callersFunctionto_cam_open3d
(viewpoint_stack)
diff_rendering/gaussian_splatting_2d/utils/mesh_utils.py:45
↓ 1 callersFunctiontrain_2dgs
(args=None)
scripts/train_2dgs.py:63
↓ 1 callersFunctiontrain_physical_parameters
(output_path,builder,obj_id,ee_id,dataset,simulator:Simulator, gaussians:Gaussian
scripts/identify_physical_parameters.py:61
↓ 1 callersFunctiontrain_static_2dgs
(gaussian_args,scene_info,mesh=None,logger=None)
diff_rendering/gaussian_splatting_2d/train_2dgs.py:260
↓ 1 callersFunctiontraining
(dataset, opt, pipe, testing_iterations, saving_iterations, checkpoint_iterations, checkpoint,scene_info,mesh=
diff_rendering/gaussian_splatting_2d/train_2dgs.py:37
↓ 1 callersFunctiontraining_report
(logger, iteration, Ll1, loss, l1_loss, elapsed, testing_iterations, scene : Scene, renderFunc, renderArgs)
diff_rendering/gaussian_splatting_2d/train_2dgs.py:197
↓ 1 callersFunctiontransformPoint4x3
diff_rendering/gaussian_splatting_2d/submodules/diff-surfel-rasterization/cuda_rasterizer/auxiliary.h:78
↓ 1 callersFunctiontransformPoint4x4
diff_rendering/gaussian_splatting_2d/submodules/diff-surfel-rasterization/cuda_rasterizer/auxiliary.h:88
↓ 1 callersFunctiontransform_poses_pca
Transforms poses so principal components lie on XYZ axes. Args: poses: a (N, 3, 4) array containing the cameras' camera to world transforms.
diff_rendering/gaussian_splatting_2d/utils/render_utils.py:76
↓ 1 callersMethodtranslate2localframe
(self,world_position,world_rotation)
diff_rendering/gaussian_splatting_2d/scene/gaussian_model.py:521
↓ 1 callersMethodupdate_learning_rate
Learning rate scheduling per step
diff_rendering/gaussian_splatting_2d/scene/gaussian_model.py:194
↓ 1 callersMethodvideo_push_shoot
(self, ee_goal_position)
envs/data_collect/data_collect_env.py:90
↓ 1 callersMethodwarn
Write the sequence of args, with no separators, to the console and output files (if you've configured an output file). Using
utils/sys_utils.py:484
↓ 1 callersMethodwrite_sequence
write_sequence an array to file :param sequence:
utils/sys_utils.py:145
↓ 1 callersFunctionxyzw_quaternion_to_rotmat
r"""Converts a quaternion to a :math:`3 \times 3` rotation matrix. Args: quaternion (torch.Tensor): Quaternion to convert (shape: :math:`
utils/quaternion_utils.py:125
↓ 1 callersMethodz_score
(self, x: torch.Tensor)
diff_rendering/gaussian_splatting_2d/lpipsPyTorch/modules/networks.py:50
MethodJ
(self)
diff_simulation/constraints/rot_constraint.py:9
MethodJ
(self)
diff_simulation/constraints/trans_constraint.py:9
MethodJ
(self)
diff_simulation/constraints/fix_constraint.py:9
FunctionL1_norm
(para)
diff_rendering/gaussian_splatting_2d/utils/loss_utils.py:108
FunctionPYBIND11_MODULE
diff_rendering/gaussian_splatting_2d/submodules/diff-surfel-rasterization/ext.cpp:15
FunctionPYBIND11_MODULE
diff_rendering/gaussian_splatting_2d/submodules/simple-knn/ext.cpp:15
Method__deepcopy__
(self,memo)
diff_rendering/gaussian_splatting_2d/scene/gaussian_model.py:74
Method__init__
(self,all_args)
builder.py:25
Method__init__
(self, figure: plt.figure, close: bool)
utils/sys_utils.py:83
Method__init__
(self, image: Union[torch.Tensor, np.ndarray, str], dataformats: str)
utils/sys_utils.py:97
Method__init__
(self, unsupported_formats: Sequence[str], value_description: str)
utils/sys_utils.py:103
Method__init__
log to a file, in a human readable format :param filename_or_file: the file to write the log to :param max_length: the maximu
utils/sys_utils.py:159
Method__init__
Dumps key/value pairs into TensorBoard's numeric format. :param folder: the folder to write the log to
utils/sys_utils.py:264
Method__init__
(self, folder: Optional[str], output_formats: List[KVWriter])
utils/sys_utils.py:395
Method__init__
(self, sim_args)
envs/base.py:28
Method__init__
(self, data_args, sim_args)
envs/data_collect/data_collect_env.py:13
Method__init__
(self, cam_pos, cam_tar, cam_up_vector, near, far, size, fov)
envs/data_collect/camera.py:9
Method__init__
(self,requires_grad,device)
diff_simulation/physical_parameters.py:5
Method__init__
(self,dtime,device,vis)
diff_simulation/simulator.py:17
Method__init__
(self, body_id)
diff_simulation/constraints/rot_constraint.py:6
Method__init__
(self,body1_id,body2_id,constraint_dim)
diff_simulation/constraints/base.py:10
Method__init__
(self, body_id)
diff_simulation/constraints/trans_constraint.py:6
Method__init__
(self, body_id)
diff_simulation/constraints/fix_constraint.py:6
Method__init__
(self,direction,magnitude=10.0,starttime=0.0,endtime=1e5,)
diff_simulation/force/base.py:6
Method__init__
(self,direction,magnitude,starttime=0.0,endtime=1e5,)
diff_simulation/force/constant_force.py:4
Method__init__
(self,simulator)
diff_simulation/solver/lcp_solver.py:20
Method__init__
(self,collision_geom,visual_geom,physical_parameters:Physical_Parameters, urdf,world_position
diff_simulation/body/base.py:13
Method__init__
(self,mesh:trimesh.base.Trimesh,physical_parameters:Physical_Parameters,urdf, requires_grad,d
diff_simulation/body/body_mesh.py:13
Method__init__
(self, sh_degree : int)
diff_rendering/gaussian_splatting_2d/scene/gaussian_model.py:51
Method__init__
b :param path: Path to colmap scene main folder.
diff_rendering/gaussian_splatting_2d/scene/__init__.py:29
Method__init__
(self, sh_degree : int, mesh:trimesh.base.Trimesh)
diff_rendering/gaussian_splatting_2d/scene/mesh_gaussian_model.py:34
Method__init__
(self, colmap_id, R, T, FoVx, FoVy, image, gt_alpha_mask, image_name, uid, t
diff_rendering/gaussian_splatting_2d/scene/cameras.py:18
Method__init__
(self, width, height, fovy, fovx, znear, zfar, world_view_transform, full_proj_transform)
diff_rendering/gaussian_splatting_2d/scene/cameras.py:63
← previousnext →201–300 of 473, ranked by callers