MCPcopy Create free account

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

Functions473 in github.com/XuAdventurer/PIN-WM

↓ 19 callersMethodrecord
Log a value of some diagnostic Call this once for each diagnostic quantity, each iteration If called many times, last value w
utils/sys_utils.py:403
↓ 14 callersFunctionexpandParam
(X, nBatch, nDim)
diff_simulation/solver/util.py:68
↓ 14 callersMethodlog
Write the sequence of args, with no separators, to the console and output files (if you've configured an output file). level:
utils/sys_utils.py:454
↓ 13 callersMethodget_material
(self,material_name)
diff_simulation/physical_parameters.py:26
↓ 13 callersFunctionread_next_bytes
Read and unpack the next bytes from a binary file. :param fid: :param num_bytes: Sum of combination of {2, 4, 8}, e.g. 2, 6, 16, 30, etc.
diff_rendering/gaussian_splatting_2d/scene/colmap_loader.py:72
↓ 10 callersMethodget_body
(self,body_id)
diff_simulation/simulator.py:49
↓ 10 callersMethodget_body_list_index
(self,body_id)
diff_simulation/simulator.py:61
↓ 9 callersMethodwrite
Write a dictionary to file :param key_values: :param key_excluded: :param step:
utils/sys_utils.py:119
↓ 8 callersMethoddump
Write all of the diagnostics from the current iteration
utils/sys_utils.py:440
↓ 8 callersFunctionget_sizes
(G, A=None)
diff_simulation/solver/util.py:46
↓ 8 callersFunctionnormalize
Normalization helper function.
diff_rendering/gaussian_splatting_2d/utils/render_utils.py:28
↓ 8 callersFunctionrotate_point_by_quaternion
(q, p)
utils/quaternion_utils.py:9
↓ 7 callersFunctionbger
(x, y)
diff_simulation/solver/util.py:42
↓ 7 callersMethoderror
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:493
↓ 7 callersMethodgetTrainCameras
(self, scale=1.0)
diff_rendering/gaussian_splatting_2d/scene/__init__.py:107
↓ 7 callersFunctionlu_hack
(x)
diff_simulation/solver/batch.py:32
↓ 7 callersFunctionquaternion_multiply
(q1, q2)
utils/quaternion_utils.py:22
↓ 7 callersFunctionsave_img_u8
Save an image (probably RGB) in [0, 1] to disk as a uint8 PNG.
diff_rendering/gaussian_splatting_2d/utils/render_utils.py:271
↓ 7 callersMethodwrite
(self, x)
diff_rendering/gaussian_splatting_2d/utils/general_utils.py:118
↓ 6 callersFunctionfocal2fov
(focal, pixels)
diff_rendering/gaussian_splatting_2d/utils/graphics_utils.py:76
↓ 6 callersMethodsave
(self, iteration)
diff_rendering/gaussian_splatting_2d/scene/__init__.py:100
↓ 5 callersFunctionSH2RGB
(sh)
diff_rendering/gaussian_splatting_2d/utils/sh_utils.py:117
↓ 5 callersMethodclose
closes the file
utils/sys_utils.py:519
↓ 5 callersMethodreconstruction
reconstruct radiance field given cameras
diff_rendering/gaussian_splatting_2d/utils/mesh_utils.py:90
↓ 5 callersFunctionwxyz2xyzw
(q)
utils/quaternion_utils.py:173
↓ 4 callersMethod__init__
(self)
diff_rendering/gaussian_splatting_2d/lpipsPyTorch/modules/networks.py:89
↓ 4 callersFunctionbdiag
(d)
diff_simulation/solver/util.py:60
↓ 4 callersFunctioncal_transform_matrix
(scale_factors,rotation_quaternion,translation_vectors)
utils/mesh_utils.py:158
↓ 4 callersFunctioncameraList_from_camInfos
(cam_infos, resolution_scale, args)
diff_rendering/gaussian_splatting_2d/utils/camera_utils.py:56
↓ 4 callersMethodclose
(self)
diff_simulation/simulator.py:255
↓ 4 callersMethodexport_image
(self, path)
diff_rendering/gaussian_splatting_2d/utils/mesh_utils.py:274
↓ 4 callersMethodextract
(self, args)
diff_rendering/gaussian_splatting_2d/arguments/__init__.py:40
↓ 4 callersMethodflush
(self)
diff_rendering/gaussian_splatting_2d/utils/general_utils.py:125
↓ 4 callersFunctionfov2focal
(fov, pixels)
diff_rendering/gaussian_splatting_2d/utils/graphics_utils.py:73
↓ 4 callersFunctionget_step
(v, dv)
diff_simulation/solver/batch.py:244
↓ 4 callersMethodhide
(self, body_id)
envs/base.py:50
↓ 4 callersMethodno_optimize
(self,material_name)
diff_simulation/physical_parameters.py:61
↓ 4 callersFunctionpad_poses
Pad [..., 3, 4] pose matrices with a homogeneous bottom row [0,0,0,1].
diff_rendering/gaussian_splatting_2d/utils/render_utils.py:32
↓ 4 callersMethodseed
(self, seed=None)
envs/base.py:21
↓ 4 callersFunctionstorePly
(path, xyz, rgb)
diff_rendering/gaussian_splatting_2d/scene/dataset_readers.py:117
↓ 3 callersFunctionPILtoTorch
(pil_image, resolution)
diff_rendering/gaussian_splatting_2d/utils/general_utils.py:21
↓ 3 callersMethod__init__
(self, parser: ArgumentParser, name : str, fill_none = False)
diff_rendering/gaussian_splatting_2d/arguments/__init__.py:20
↓ 3 callersMethod_truncate
(self, string: str)
utils/sys_utils.py:241
↓ 3 callersMethodapply_external_forces
(self, cur_time)
diff_simulation/body/base.py:154
↓ 3 callersMethodcreate_joint
(self,body_id,joint_type)
diff_simulation/simulator.py:200
↓ 3 callersMethodcreate_mesh_body
(self,mesh,physical_parameters,urdf,requires_grad,world_position=None,world_rotation=None)
diff_simulation/simulator.py:193
↓ 3 callersFunctionfactor_solve_kkt
(Q, D, G, A, C_tilde, rx, rs, rz, ry)
diff_simulation/solver/batch.py:358
↓ 3 callersFunctionfetchPly
(path)
diff_rendering/gaussian_splatting_2d/scene/dataset_readers.py:109
↓ 3 callersFunctiongetNerfppNorm
(cam_info)
diff_rendering/gaussian_splatting_2d/scene/dataset_readers.py:47
↓ 3 callersMethodgetTestCameras
(self, scale=1.0)
diff_rendering/gaussian_splatting_2d/scene/__init__.py:110
↓ 3 callersMethodget_pose_cpu
(self)
diff_simulation/body/base.py:94
↓ 3 callersFunctionpsnr
(img1, img2)
diff_rendering/gaussian_splatting_2d/utils/image_utils.py:17
↓ 3 callersMethodset_material
(self,material_name,value)
diff_simulation/physical_parameters.py:16
↓ 3 callersMethodset_requires_grad
(self, state: bool)
diff_rendering/gaussian_splatting_2d/lpipsPyTorch/modules/networks.py:46
↓ 3 callersMethodshot
(self, cam_id, frame_id, frame_dt)
envs/data_collect/camera.py:42
↓ 3 callersFunctionsolve_kkt
Solve KKT equations for the affine step
diff_simulation/solver/batch.py:422
↓ 3 callersFunctionsolve_kkt_ir
Inefficient iterative refinement.
diff_simulation/solver/batch.py:278
↓ 3 callersMethodstep
(self)
diff_simulation/simulator.py:240
↓ 2 callersMethodJe
(self)
diff_simulation/solver/lcp_solver.py:25
↓ 2 callersMethodM
(self)
diff_simulation/solver/lcp_solver.py:85
↓ 2 callersFunctionRGB2SH
(rgb)
diff_rendering/gaussian_splatting_2d/utils/sh_utils.py:114
↓ 2 callersMethodadd_body
(self, body_position, body_rotation, body_path, tex_path=None)
envs/base.py:63
↓ 2 callersMethodadd_gravity
(self)
diff_simulation/body/base.py:84
↓ 2 callersMethodapply
(self, cur_time)
diff_simulation/force/base.py:12
↓ 2 callersMethodbuild_data_collect_env
(self)
builder.py:90
↓ 2 callersFunctionbuild_rotation
(r)
diff_rendering/gaussian_splatting_2d/utils/general_utils.py:78
↓ 2 callersMethodbuild_static_2dgs
(self,all_train_cam_infos=None,mesh=None,logger=None)
builder.py:48
↓ 2 callersMethodcamera_surround_shoot
(self, frame_id, surround_ee_positon_batch)
envs/data_collect/data_collect_env.py:73
↓ 2 callersMethodclean
(self)
diff_rendering/gaussian_splatting_2d/utils/mesh_utils.py:81
↓ 2 callersFunctioncolormap
(img, cmap='jet')
diff_rendering/gaussian_splatting_2d/utils/general_utils.py:163
↓ 2 callersFunctionconfigure_logger
Configure the current logger. :param folder: the save location :param format_strings: the output logging format :return: The logger o
utils/sys_utils.py:538
↓ 2 callersFunctioncpu_deep_copy_tuple
(input_tuple)
diff_rendering/gaussian_splatting_2d/submodules/diff-surfel-rasterization/diff_surfel_rasterization/__init__.py:17
↓ 2 callersMethoddensification_postfix
(self, new_xyz, new_features_dc, new_features_rest, new_opacities, new_scaling, new_rotation)
diff_rendering/gaussian_splatting_2d/scene/gaussian_model.py:359
↓ 2 callersFunctioneval_sh
Evaluate spherical harmonics at unit directions using hardcoded SH polynomials. Works with torch/np/jnp. ... Can be 0 or more batch d
diff_rendering/gaussian_splatting_2d/utils/sh_utils.py:57
↓ 2 callersMethodextract
(self, args)
diff_rendering/gaussian_splatting_2d/arguments/__init__.py:60
↓ 2 callersFunctionextract_nBatch
(Q, p, G, h, A, b, F)
diff_simulation/solver/util.py:77
↓ 2 callersFunctionfactor_kkt
Factor the U22 block that we can only do after we know D.
diff_simulation/solver/batch.py:573
↓ 2 callersFunctionfactor_solve_kkt_reg
(Q_tilde, D, G, A, C_tilde, rx, rs, rz, ry, eps)
diff_simulation/solver/batch.py:313
↓ 2 callersFunctionfocus_point_fn
Calculate nearest point to all focal axes in poses.
diff_rendering/gaussian_splatting_2d/utils/render_utils.py:68
↓ 2 callersFunctiongetWorld2View2
(R, t, translate=np.array([.0, .0, .0]), scale=1.0)
diff_rendering/gaussian_splatting_2d/utils/graphics_utils.py:38
↓ 2 callersMethodget_all_physical_parameters
(self)
diff_simulation/simulator.py:218
↓ 2 callersMethodget_body_from_pybullet_id
(self,pybullet_id)
diff_simulation/simulator.py:55
↓ 2 callersMethodget_body_pose_clone
(self,body_id)
diff_simulation/simulator.py:168
↓ 2 callersFunctionget_combined_args
(args1,args2)
utils/cfg_utils.py:19
↓ 2 callersMethodget_physical_parameters
(self)
diff_simulation/body/base.py:50
↓ 2 callersMethodget_vel_vec
(self)
diff_simulation/body/base.py:179
↓ 2 callersFunctionkkt_resid_reg
(Q_tilde, D_tilde, G, A, F_tilde, eps, dx, ds, dz, dy, rx, rs, rz, ry)
diff_simulation/solver/batch.py:262
↓ 2 callersFunctionl1_loss
(network_output, gt)
diff_rendering/gaussian_splatting_2d/utils/loss_utils.py:17
↓ 2 callersFunctionload_img
Load an image and cast to float32.
diff_rendering/gaussian_splatting_2d/utils/render_utils.py:196
↓ 2 callersMethodload_ply
(self, path,requires_grad=True)
diff_rendering/gaussian_splatting_2d/scene/gaussian_model.py:245
↓ 2 callersFunctionmultiply
r"""Multiply two quaternions `q1`, `q2`. Args: q1 (torch.Tensor): First quaternion (shape: :math:`(4)`) (Assumes (r, i, j, k)
utils/quaternion_utils.py:148
↓ 2 callersFunctionprepare_output_and_logger
(all_args,need_logger = False)
utils/sys_utils.py:29
↓ 2 callersMethodprune_points
(self, mask)
diff_rendering/gaussian_splatting_2d/scene/gaussian_model.py:321
↓ 2 callersFunctionquat_to_rot
(q)
diff_rendering/gaussian_splatting_2d/scripts/eval_dtu/render_utils.py:112
↓ 2 callersFunctionquaternion_standardize_np
(q)
utils/quaternion_utils.py:212
↓ 2 callersFunctionquaternion_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:90
↓ 2 callersFunctionqvec2rotmat
(qvec)
diff_rendering/gaussian_splatting_2d/scene/colmap_loader.py:43
↓ 2 callersFunctionreadCamerasFromTransforms
(path, transformsfile, white_background, extension=".png")
diff_rendering/gaussian_splatting_2d/scene/dataset_readers.py:181
↓ 2 callersFunctionreadDataFromJson
(path, json_file, white_background,extension)
utils/data_utils.py:16
↓ 2 callersMethodreset
(self)
envs/data_collect/data_collect_env.py:49
next →1–100 of 473, ranked by callers