Functions — github.com/XuAdventurer/PIN-WM

CodeHub MCPcopy Create free account

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

Functions473 in github.com/XuAdventurer/PIN-WM

⨍ Functions 473◇ Types & classes 60

diff_rendering/gaussian_splatting_2d/utils/general_utils.py:115

a class that extracts attributes a scene presented by 2DGS Usage example: >>> gaussExtrator = GaussianExtractor(gaussians, r

diff_rendering/gaussian_splatting_2d/utils/mesh_utils.py:64

(self, parser, sentinel=False)

diff_rendering/gaussian_splatting_2d/arguments/__init__.py:48

diff_rendering/gaussian_splatting_2d/arguments/__init__.py:66

diff_rendering/gaussian_splatting_2d/arguments/__init__.py:74

(self, raster_settings)

diff_rendering/gaussian_splatting_2d/submodules/diff-surfel-rasterization/diff_surfel_rasterization/__init__.py:173

(self, n_channels_list: Sequence[int])

diff_rendering/gaussian_splatting_2d/lpipsPyTorch/modules/networks.py:24

diff_rendering/gaussian_splatting_2d/lpipsPyTorch/modules/networks.py:37

diff_rendering/gaussian_splatting_2d/lpipsPyTorch/modules/networks.py:67

diff_rendering/gaussian_splatting_2d/lpipsPyTorch/modules/networks.py:78

(self, net_type: str = 'alex', version: str = '0.1')

diff_rendering/gaussian_splatting_2d/lpipsPyTorch/modules/lpips.py:17

Method_prune_optimizer

diff_rendering/gaussian_splatting_2d/scene/mesh_gaussian_model.py:116

Methodadd_noise

diff_simulation/physical_parameters.py:64

Methodapply_external_forces

(self, cur_time)

diff_simulation/simulator.py:128

Methodapply_impulse

(self,contact_pos,j)

diff_simulation/body/base.py:165

Methodbuild_covariance_from_scaling_rotation

(center, scaling, scaling_modifier, rotation)

diff_rendering/gaussian_splatting_2d/scene/gaussian_model.py:34

Functioncal_MassProperties_np

(mesh:trimesh.base.Trimesh)

utils/mesh_utils.py:82

Methodcat_tensors_to_optimizer

(self, tensors_dict)

diff_rendering/gaussian_splatting_2d/scene/mesh_gaussian_model.py:93

Methodchange_body_pose

(self, body_id, body_position, body_rotation)

envs/base.py:60

Methodchange_body_pose

(self,body_id,world_position=None,world_rotation=None)

diff_simulation/simulator.py:185

Methodchange_body_vel

(self,body_id,linear_velocity=None,angular_velocity=None)

diff_simulation/simulator.py:189

Methodclear_force

diff_simulation/body/base.py:150

Close owned resources

utils/sys_utils.py:133

closes the file

utils/sys_utils.py:255

closes the file

utils/sys_utils.py:313

envs/base.py:47

Methodcompute_angular_velocity_from_angular_momentum

(inertia_world_inv, angular_momentum,)

diff_simulation/body/base.py:128

Methodcompute_center_of_mass

(vertices, masses)

diff_simulation/body/base.py:134

Methodcompute_linear_velocity_from_linear_momentum

(linear_momentum, mass,)

diff_simulation/body/base.py:131

Methodcompute_sdf_perframe

compute per frame sdf

diff_rendering/gaussian_splatting_2d/utils/mesh_utils.py:187

Methodcompute_state_derivatives

(self, cur_time)

diff_simulation/body/base.py:138

Methodcompute_unbounded_tsdf

Fusion all frames, perform adaptive sdf_funcation on the contract spaces.

diff_rendering/gaussian_splatting_2d/utils/mesh_utils.py:201

Functionconfig_parser

Define command line arguments

utils/cfg_utils.py:9

diff_rendering/gaussian_splatting_2d/utils/mesh_utils.py:179

Methodcreate_from_pcd

(self, pcd : BasicPointCloud, spatial_lr_scale : float)

diff_rendering/gaussian_splatting_2d/scene/mesh_gaussian_model.py:46

Functioncreate_gif_with_rgbs

utils/vis_utils.py:30

Functioncreate_q_x

utils/quaternion_utils.py:185

Functioncreate_q_y

utils/quaternion_utils.py:201

Functioncreate_q_z

utils/quaternion_utils.py:193

Functioncreate_rotation_matrix_from_direction_vector_batch

(direction_vectors)

diff_rendering/gaussian_splatting_2d/utils/general_utils.py:138

Functioncreate_sample_points

(sample_num,mesh_local)

utils/mesh_utils.py:186

Functioncreate_videos_with_rgbs

utils/vis_utils.py:14

diff_rendering/gaussian_splatting_2d/submodules/diff-surfel-rasterization/cuda_rasterizer/auxiliary.h:154

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:466

Functiondnormvdv

diff_rendering/gaussian_splatting_2d/submodules/diff-surfel-rasterization/cuda_rasterizer/auxiliary.h:127

Functiondnormvdz

diff_rendering/gaussian_splatting_2d/submodules/diff-surfel-rasterization/cuda_rasterizer/auxiliary.h:119

Functionefficient_btriunpack

More efficient version of torch.btriunpack. From https://github.com/pytorch/pytorch/issues/15182

diff_simulation/solver/util.py:94

Functionextract_batch_size

(Q, p, G, h, A, b)

diff_simulation/solver/util.py:85

Methodforce_function

diff_simulation/force/constant_force.py:7

Functionforward

Q_LU, S_LU, R = pre_factor_kkt(Q, G, A)

diff_simulation/solver/batch.py:71

Solve a batch of QPs. This function solves a batch of QPs, each optimizing over `nz` variables and having `nineq` ine

diff_simulation/solver/lcp_solver.py:185

( ctx, means3D, means2D, sh, colors_precomp, opacities,

diff_rendering/gaussian_splatting_2d/submodules/diff-surfel-rasterization/diff_surfel_rasterization/__init__.py:46

(self, means3D, means2D, opacities, shs = None, colors_precomp = None, scales = None, rotations = None, cov3D_

diff_rendering/gaussian_splatting_2d/submodules/diff-surfel-rasterization/diff_surfel_rasterization/__init__.py:188

(self, x: torch.Tensor)

diff_rendering/gaussian_splatting_2d/lpipsPyTorch/modules/networks.py:53

(self, x: torch.Tensor, y: torch.Tensor)

diff_rendering/gaussian_splatting_2d/lpipsPyTorch/modules/lpips.py:30

Functionforward_single_np

(Q, p, G, h, A, b)

diff_simulation/solver/cvxpy.py:29

Functiongeom_transform_points

(points, transf_matrix)

diff_rendering/gaussian_splatting_2d/utils/graphics_utils.py:22

FunctiongetRect

diff_rendering/gaussian_splatting_2d/submodules/diff-surfel-rasterization/cuda_rasterizer/auxiliary.h:66

FunctiongetWorld2View

diff_rendering/gaussian_splatting_2d/utils/graphics_utils.py:31

Functionget_all_frames_train_camera

(cam_id,all_train_cam_infos,dataset)

utils/data_utils.py:138

Functionget_ang_inertia

(verts, faces, mass)

utils/mesh_utils.py:328

Methodget_body_physical_parameters

diff_simulation/simulator.py:214

Methodget_body_pose

diff_simulation/simulator.py:164

Methodget_body_pose_cpu

diff_simulation/simulator.py:172

Methodget_body_pose_np

diff_simulation/simulator.py:176

Methodget_body_vel_np

diff_simulation/simulator.py:180

Methodget_body_vel_np

diff_simulation/body/base.py:103

envs/data_collect/camera.py:33

Functionget_camera_for_plot

diff_rendering/gaussian_splatting_2d/scripts/eval_dtu/render_utils.py:85

Functionget_camera_params

(uv, pose, intrinsics)

diff_rendering/gaussian_splatting_2d/scripts/eval_dtu/render_utils.py:55

Methodget_collision_geom_local

diff_simulation/body/base.py:186

Methodget_collision_geom_world

diff_simulation/body/base.py:183

Functionget_contact_info

(action,sample_points,obj_position,obj_rotation)

utils/mesh_utils.py:196

Get directory that log files are being written to. will be None if there is no output directory (i.e., if you didn't call start)

utils/sys_utils.py:511

Methodget_features

diff_rendering/gaussian_splatting_2d/scene/gaussian_model.py:134

Methodget_joint

(self,joint_id)

diff_simulation/simulator.py:67

Methodget_joint_list_index

(self,joint_id)

diff_simulation/simulator.py:73

Methodget_material_names

diff_simulation/physical_parameters.py:46

Methodget_material_num

diff_simulation/physical_parameters.py:43

Functionget_mesh_world

(mesh,position,rotation)

utils/mesh_utils.py:176

Methodget_opacity

diff_rendering/gaussian_splatting_2d/scene/gaussian_model.py:140

Functionget_psnr

(img1, img2, normalize_rgb=False)

diff_rendering/gaussian_splatting_2d/scripts/eval_dtu/render_utils.py:9

Methodget_rotation

diff_rendering/gaussian_splatting_2d/scene/gaussian_model.py:124

Methodget_scaling

diff_rendering/gaussian_splatting_2d/scene/gaussian_model.py:120

Functionget_sphere_intersections

(cam_loc, ray_directions, r = 1.0)

diff_rendering/gaussian_splatting_2d/scripts/eval_dtu/render_utils.py:153

Methodget_vel_vec

diff_simulation/simulator.py:131

Methodget_visual_geom_local

diff_simulation/body/base.py:192

Methodget_visual_geom_local

diff_simulation/body/body_mesh.py:69

Methodget_visual_geom_world

diff_simulation/body/body_mesh.py:41

diff_rendering/gaussian_splatting_2d/scene/gaussian_model.py:130

diff_rendering/gaussian_splatting_2d/utils/general_utils.py:47

Functionin_frustum

diff_rendering/gaussian_splatting_2d/submodules/diff-surfel-rasterization/cuda_rasterizer/auxiliary.h:184

Functioninertia_diagonalize

utils/mesh_utils.py:345

Functioninertia_diagonalize_np

utils/mesh_utils.py:357

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:475

(wish_host, wish_port)

diff_rendering/gaussian_splatting_2d/gaussian_renderer/network_gui.py:26

Functioninit_s_z

Solve KKT equations for the affine step

diff_simulation/solver/batch.py:397

Functioninverse_sigmoid

diff_rendering/gaussian_splatting_2d/utils/general_utils.py:18

Functionl2_loss

(network_output, gt)

diff_rendering/gaussian_splatting_2d/utils/loss_utils.py:20

Functionload_K_Rt_from_P

(filename, P=None)

diff_rendering/gaussian_splatting_2d/scripts/eval_dtu/render_utils.py:31

← previous next →301–400 of 473, ranked by callers