SYMBOL INDEX (1427 symbols across 154 files) FILE: physdreamer/field_components/encoding.py class TemporalKplanesEncoding (line 12) | class TemporalKplanesEncoding(nn.Module): method __init__ (line 19) | def __init__( method forward (line 60) | def forward(self, inp: Float[Tensor, "*bs 4"]): method compute_temporal_smoothness (line 86) | def compute_temporal_smoothness( method compute_plane_tv (line 96) | def compute_plane_tv( method visualize (line 106) | def visualize( method functional_forward (line 117) | def functional_forward( class TriplanesEncoding (line 152) | class TriplanesEncoding(nn.Module): method __init__ (line 159) | def __init__( method forward (line 199) | def forward(self, inp: Float[Tensor, "*bs 3"]): method compute_plane_tv (line 225) | def compute_plane_tv( class PlaneEncoding (line 236) | class PlaneEncoding(nn.Module): method __init__ (line 243) | def __init__( method forward (line 276) | def forward(self, inp: Float[Tensor, "*bs 2"]): method compute_plane_tv (line 291) | def compute_plane_tv( class TemporalNeRFEncoding (line 302) | class TemporalNeRFEncoding(nn.Module): method __init__ (line 303) | def __init__( method get_out_dim (line 320) | def get_out_dim(self) -> int: method forward (line 328) | def forward( FILE: physdreamer/field_components/mlp.py class MLP (line 11) | class MLP(nn.Module): method __init__ (line 12) | def __init__( method build_nn_modules (line 40) | def build_nn_modules(self) -> None: method pytorch_fwd (line 65) | def pytorch_fwd( method forward (line 88) | def forward( FILE: physdreamer/fields/mul_offset_field.py class MulTemporalKplanesOffsetfields (line 15) | class MulTemporalKplanesOffsetfields(nn.Module): method __init__ (line 28) | def __init__( method forward (line 74) | def forward( method compute_smoothess_loss (line 98) | def compute_smoothess_loss( method compute_loss (line 117) | def compute_loss( method arap_loss (line 133) | def arap_loss(self, inp): FILE: physdreamer/fields/mul_se3_field.py class MulTemporalKplanesSE3fields (line 15) | class MulTemporalKplanesSE3fields(nn.Module): method __init__ (line 26) | def __init__( method forward (line 75) | def forward( method compute_smoothess_loss (line 111) | def compute_smoothess_loss( method compute_loss (line 130) | def compute_loss( FILE: physdreamer/fields/offset_field.py class TemporalKplanesOffsetfields (line 15) | class TemporalKplanesOffsetfields(nn.Module): method __init__ (line 26) | def __init__( method forward (line 66) | def forward( method compute_smoothess_loss (line 87) | def compute_smoothess_loss( method compute_loss (line 100) | def compute_loss( method arap_loss (line 116) | def arap_loss(self, inp): method forward_with_plane_coefs (line 119) | def forward_with_plane_coefs( FILE: physdreamer/fields/se3_field.py class TemporalKplanesSE3fields (line 15) | class TemporalKplanesSE3fields(nn.Module): method __init__ (line 26) | def __init__( method forward (line 69) | def forward( method compute_smoothess_loss (line 120) | def compute_smoothess_loss( method compute_loss (line 133) | def compute_loss( FILE: physdreamer/fields/triplane_field.py class TriplaneFields (line 11) | class TriplaneFields(nn.Module): method __init__ (line 22) | def __init__( method forward (line 57) | def forward( method compute_smoothess_loss (line 72) | def compute_smoothess_loss( function compute_entropy (line 80) | def compute_entropy(p): class TriplaneFieldsWithEntropy (line 86) | class TriplaneFieldsWithEntropy(nn.Module): method __init__ (line 97) | def __init__( method forward (line 134) | def forward( method compute_smoothess_loss (line 155) | def compute_smoothess_loss( FILE: physdreamer/gaussian_3d/arguments/__init__.py class GroupParams (line 16) | class GroupParams: class ParamGroup (line 19) | class ParamGroup: method __init__ (line 20) | def __init__(self, parser: ArgumentParser, name : str, fill_none = Fal... method extract (line 40) | def extract(self, args): class ModelParams (line 47) | class ModelParams(ParamGroup): method __init__ (line 48) | def __init__(self, parser, sentinel=False): method extract (line 59) | def extract(self, args): class PipelineParams (line 64) | class PipelineParams(ParamGroup): method __init__ (line 65) | def __init__(self, parser): class OptimizationParams (line 71) | class OptimizationParams(ParamGroup): method __init__ (line 72) | def __init__(self, parser): function get_combined_args (line 91) | def get_combined_args(parser : ArgumentParser): FILE: physdreamer/gaussian_3d/gaussian_renderer/depth_uv_render.py function render_uv_depth_w_gaussian (line 12) | def render_uv_depth_w_gaussian( FILE: physdreamer/gaussian_3d/gaussian_renderer/feat_render.py function render_feat_gaussian (line 12) | def render_feat_gaussian( FILE: physdreamer/gaussian_3d/gaussian_renderer/flow_depth_render.py function render_flow_depth_w_gaussian (line 12) | def render_flow_depth_w_gaussian( FILE: physdreamer/gaussian_3d/gaussian_renderer/render.py function render_gaussian (line 21) | def render_gaussian( function gaussian_intrin_scale (line 127) | def gaussian_intrin_scale(x_or_y: torch.Tensor, w_or_h: float): function render_arrow_in_screen (line 134) | def render_arrow_in_screen(viewpoint_camera, points_3d): function render_arrow_in_screen_back (line 190) | def render_arrow_in_screen_back(viewpoint_camera, points_3d): function eval_sh (line 261) | def eval_sh(deg, sh, dirs): function RGB2SH (line 327) | def RGB2SH(rgb): function SH2RGB (line 331) | def SH2RGB(sh): FILE: physdreamer/gaussian_3d/scene/__init__.py class Scene (line 26) | class Scene: method __init__ (line 29) | def __init__( method save (line 116) | def save(self, iteration): method getTrainCameras (line 122) | def getTrainCameras(self, scale=1.0): method getTestCameras (line 125) | def getTestCameras(self, scale=1.0): FILE: physdreamer/gaussian_3d/scene/cameras.py class Camera (line 21) | class Camera(nn.Module): method __init__ (line 22) | def __init__( class MiniCam (line 91) | class MiniCam: method __init__ (line 92) | def __init__( FILE: physdreamer/gaussian_3d/scene/colmap_loader.py function qvec2rotmat (line 43) | def qvec2rotmat(qvec): function rotmat2qvec (line 55) | def rotmat2qvec(R): class Image (line 68) | class Image(BaseImage): method qvec2rotmat (line 69) | def qvec2rotmat(self): function read_next_bytes (line 72) | def read_next_bytes(fid, num_bytes, format_char_sequence, endian_charact... function read_points3D_text (line 83) | def read_points3D_text(path): function read_points3D_binary (line 113) | def read_points3D_binary(path_to_model_file): function read_intrinsics_text (line 144) | def read_intrinsics_text(path): function read_extrinsics_binary (line 168) | def read_extrinsics_binary(path_to_model_file): function read_intrinsics_binary (line 203) | def read_intrinsics_binary(path_to_model_file): function read_extrinsics_text (line 232) | def read_extrinsics_text(path): function read_colmap_bin_array (line 261) | def read_colmap_bin_array(path): FILE: physdreamer/gaussian_3d/scene/dataset_readers.py class CameraInfo (line 45) | class CameraInfo(NamedTuple): class SceneInfo (line 58) | class SceneInfo(NamedTuple): function getNerfppNorm (line 66) | def getNerfppNorm(cam_info): function readColmapCameras (line 90) | def readColmapCameras(cam_extrinsics, cam_intrinsics, images_folder): function fetchPly (line 142) | def fetchPly(path): function storePly (line 151) | def storePly(path, xyz, rgb): function readColmapSceneInfo (line 177) | def readColmapSceneInfo(path, images, eval, llffhold=8): function readCamerasFromTransforms (line 233) | def readCamerasFromTransforms(path, transformsfile, white_background, ex... function readNerfSyntheticInfo (line 297) | def readNerfSyntheticInfo(path, white_background, eval, extension=".png"): class NoImageCamera (line 349) | class NoImageCamera(nn.Module): method __init__ (line 350) | def __init__( function fast_read_cameras_from_transform_file (line 427) | def fast_read_cameras_from_transform_file(file_path, width=1080, height=... FILE: physdreamer/gaussian_3d/scene/gaussian_model.py class GaussianModel (line 37) | class GaussianModel: method setup_functions (line 38) | def setup_functions(self): method __init__ (line 55) | def __init__(self, sh_degree: int = 3): method capture (line 74) | def capture(self): method restore (line 95) | def restore(self, model_args, training_args): method capture_training_args (line 118) | def capture_training_args( method get_scaling (line 124) | def get_scaling(self): method get_rotation (line 128) | def get_rotation(self): method get_xyz (line 132) | def get_xyz(self): method get_features (line 136) | def get_features(self): method get_opacity (line 142) | def get_opacity(self): method get_covariance (line 145) | def get_covariance(self, scaling_modifier=1): method oneupSHdegree (line 150) | def oneupSHdegree(self): method create_from_pcd (line 154) | def create_from_pcd(self, pcd: BasicPointCloud, spatial_lr_scale: float): method training_setup (line 196) | def training_setup(self, training_args): method update_learning_rate (line 242) | def update_learning_rate(self, iteration): method construct_list_of_attributes (line 250) | def construct_list_of_attributes(self): method save_ply (line 264) | def save_ply(self, path): method reset_opacity (line 301) | def reset_opacity(self): method load_ply (line 308) | def load_ply(self, path): method replace_tensor_to_optimizer (line 388) | def replace_tensor_to_optimizer(self, tensor, name): method _prune_optimizer (line 403) | def _prune_optimizer(self, mask): method prune_points (line 425) | def prune_points(self, mask): method cat_tensors_to_optimizer (line 441) | def cat_tensors_to_optimizer(self, tensors_dict): method densification_postfix (line 475) | def densification_postfix( method densify_and_split (line 505) | def densify_and_split(self, grads, grad_threshold, scene_extent, N=2): method densify_and_clone (line 549) | def densify_and_clone(self, grads, grad_threshold, scene_extent): method densify_and_prune (line 576) | def densify_and_prune(self, max_grad, min_opacity, extent, max_screen_... method add_densification_stats (line 594) | def add_densification_stats(self, viewspace_point_tensor, update_filter): method apply_discrete_offset_filds (line 600) | def apply_discrete_offset_filds(self, origin_points, offsets): method apply_discrete_offset_filds_with_R (line 646) | def apply_discrete_offset_filds_with_R(self, origin_points, offsets, t... method apply_se3_fields (line 711) | def apply_se3_fields( method apply_offset_fields (line 772) | def apply_offset_fields(self, offset_field, timestamp: float): method apply_offset_fields_with_R (line 812) | def apply_offset_fields_with_R(self, offset_field, timestamp: float, e... method init_from_mesh (line 879) | def init_from_mesh( method detach_grad (line 943) | def detach_grad( method apply_mask (line 953) | def apply_mask(self, mask): method extract_fields (line 982) | def extract_fields(self, resolution=128, num_blocks=16, relax_ratio=1.5): method extract_mesh (line 1073) | def extract_mesh(self, path, density_thresh=1, resolution=128, decimat... function gaussian_3d_coeff (line 1109) | def gaussian_3d_coeff(xyzs, covs): FILE: physdreamer/gaussian_3d/scene/mesh.py function dot (line 8) | def dot(x, y): function length (line 12) | def length(x, eps=1e-20): function safe_normalize (line 16) | def safe_normalize(x, eps=1e-20): class Mesh (line 20) | class Mesh: method __init__ (line 21) | def __init__( method load (line 49) | def load( method load_obj (line 146) | def load_obj(cls, path, albedo_path=None, device=None): method load_trimesh (line 296) | def load_trimesh(cls, path, device=None): method aabb (line 390) | def aabb(self): method auto_size (line 395) | def auto_size(self): method auto_normal (line 401) | def auto_normal(self): method auto_uv (line 424) | def auto_uv(self, cache_path=None, vmap=True): method align_v_to_vt (line 459) | def align_v_to_vt(self, vmapping=None): method to (line 476) | def to(self, device): method write (line 484) | def write(self, path): method write_ply (line 495) | def write_ply(self, path): method write_glb (line 503) | def write_glb(self, path): method write_obj (line 653) | def write_obj(self, path): FILE: physdreamer/gaussian_3d/scene/mesh_utils.py function poisson_mesh_reconstruction (line 5) | def poisson_mesh_reconstruction(points, normals=None): function decimate_mesh (line 44) | def decimate_mesh( function clean_mesh (line 88) | def clean_mesh( FILE: physdreamer/gaussian_3d/utils/camera_utils.py function loadCam (line 21) | def loadCam(args, id, cam_info, resolution_scale): function cameraList_from_camInfos (line 69) | def cameraList_from_camInfos(cam_infos, resolution_scale, args): function camera_to_JSON (line 78) | def camera_to_JSON(id, camera: Camera): function look_at (line 101) | def look_at(from_point, to_point, up_vector=(0, 1, 0)): function create_cameras_around_sphere (line 149) | def create_cameras_around_sphere( FILE: physdreamer/gaussian_3d/utils/general_utils.py function inverse_sigmoid (line 18) | def inverse_sigmoid(x): function PILtoTorch (line 21) | def PILtoTorch(pil_image, resolution): function get_expon_lr_func (line 29) | def get_expon_lr_func( function strip_lowerdiag (line 64) | def strip_lowerdiag(L): function strip_symmetric (line 75) | def strip_symmetric(sym): function build_rotation (line 78) | def build_rotation(r): function build_scaling_rotation (line 101) | def build_scaling_rotation(s, r): function safe_state (line 112) | def safe_state(silent): FILE: physdreamer/gaussian_3d/utils/graphics_utils.py class BasicPointCloud (line 17) | class BasicPointCloud(NamedTuple): function geom_transform_points (line 22) | def geom_transform_points(points, transf_matrix): function getWorld2View (line 31) | def getWorld2View(R, t): function getWorld2View2 (line 38) | def getWorld2View2(R, t, translate=np.array([.0, .0, .0]), scale=1.0): function getProjectionMatrix (line 51) | def getProjectionMatrix(znear, zfar, fovX, fovY): function fov2focal (line 73) | def fov2focal(fov, pixels): function focal2fov (line 76) | def focal2fov(focal, pixels): FILE: physdreamer/gaussian_3d/utils/image_utils.py function mse (line 14) | def mse(img1, img2): function psnr (line 17) | def psnr(img1, img2): FILE: physdreamer/gaussian_3d/utils/loss_utils.py function l1_loss (line 17) | def l1_loss(network_output, gt): function l2_loss (line 20) | def l2_loss(network_output, gt): function gaussian (line 23) | def gaussian(window_size, sigma): function create_window (line 27) | def create_window(window_size, channel): function ssim (line 33) | def ssim(img1, img2, window_size=11, size_average=True): function _ssim (line 43) | def _ssim(img1, img2, window, window_size, channel, size_average=True): FILE: physdreamer/gaussian_3d/utils/rigid_body_utils.py function get_rigid_transform (line 5) | def get_rigid_transform(A, B): function _test_rigid_transform (line 56) | def _test_rigid_transform(): function _sqrt_positive_part (line 72) | def _sqrt_positive_part(x: torch.Tensor) -> torch.Tensor: function matrix_to_quaternion (line 83) | def matrix_to_quaternion(matrix: torch.Tensor) -> torch.Tensor: function quternion_to_matrix (line 146) | def quternion_to_matrix(r): function standardize_quaternion (line 172) | def standardize_quaternion(quaternions: torch.Tensor) -> torch.Tensor: function quaternion_multiply (line 188) | def quaternion_multiply(a: torch.Tensor, b: torch.Tensor) -> torch.Tensor: function _test_matrix_to_quaternion (line 212) | def _test_matrix_to_quaternion(): function _test_matrix_to_quaternion_2 (line 241) | def _test_matrix_to_quaternion_2(): FILE: physdreamer/gaussian_3d/utils/sh_utils.py function eval_sh (line 57) | def eval_sh(deg, sh, dirs): function RGB2SH (line 114) | def RGB2SH(rgb): function SH2RGB (line 117) | def SH2RGB(sh): FILE: physdreamer/gaussian_3d/utils/system_utils.py function mkdir_p (line 16) | def mkdir_p(folder_path): function searchForMaxIteration (line 26) | def searchForMaxIteration(folder): FILE: physdreamer/losses/smoothness_loss.py function compute_plane_tv (line 5) | def compute_plane_tv(t: torch.Tensor, only_w: bool = False) -> float: function compute_plane_smoothness (line 26) | def compute_plane_smoothness(t: torch.Tensor) -> float: FILE: physdreamer/operators/dct.py function dct1_rfft_impl (line 12) | def dct1_rfft_impl(x): function dct_fft_impl (line 16) | def dct_fft_impl(v): function idct_irfft_impl (line 20) | def idct_irfft_impl(V): function dct (line 24) | def dct(x, norm=None): function idct (line 63) | def idct(X, norm=None): function dct_3d (line 110) | def dct_3d(x, norm=None): function idct_3d (line 127) | def idct_3d(X, norm=None): function code_test_dct3d (line 146) | def code_test_dct3d(): FILE: physdreamer/operators/np_operators.py function feature_map_to_rgb_pca (line 7) | def feature_map_to_rgb_pca(feature_map): FILE: physdreamer/operators/rotation.py function rotation_6d_to_matrix (line 7) | def rotation_6d_to_matrix(d6: torch.Tensor) -> torch.Tensor: function matrix_to_rotation_6d (line 31) | def matrix_to_rotation_6d(matrix: torch.Tensor) -> torch.Tensor: function quaternion_to_matrix (line 50) | def quaternion_to_matrix(quaternions: torch.Tensor) -> torch.Tensor: function _sqrt_positive_part (line 82) | def _sqrt_positive_part(x: torch.Tensor) -> torch.Tensor: function matrix_to_quaternion (line 93) | def matrix_to_quaternion(matrix: torch.Tensor) -> torch.Tensor: FILE: physdreamer/utils/camera_utils.py function normalize (line 4) | def normalize(x: np.ndarray) -> np.ndarray: function viewmatrix (line 9) | def viewmatrix(lookdir: np.ndarray, up: np.ndarray, position: np.ndarray... function generate_spiral_path (line 18) | def generate_spiral_path( FILE: physdreamer/utils/colmap_utils.py function qvec2rotmat (line 43) | def qvec2rotmat(qvec): function rotmat2qvec (line 55) | def rotmat2qvec(R): class Image (line 68) | class Image(BaseImage): method qvec2rotmat (line 69) | def qvec2rotmat(self): function read_next_bytes (line 72) | def read_next_bytes(fid, num_bytes, format_char_sequence, endian_charact... function read_points3D_text (line 83) | def read_points3D_text(path): function read_points3D_binary (line 125) | def read_points3D_binary(path_to_model_file): function read_intrinsics_text (line 156) | def read_intrinsics_text(path): function read_extrinsics_binary (line 180) | def read_extrinsics_binary(path_to_model_file): function read_intrinsics_binary (line 215) | def read_intrinsics_binary(path_to_model_file): function read_extrinsics_text (line 244) | def read_extrinsics_text(path): function read_colmap_bin_array (line 273) | def read_colmap_bin_array(path): FILE: physdreamer/utils/config.py function load_config_with_merge (line 4) | def load_config_with_merge(config_path: str): function merge_without_none (line 17) | def merge_without_none(base_cfg, override_cfg): function create_config (line 26) | def create_config(config_path, args, cli_args: list = []): FILE: physdreamer/utils/img_utils.py function make_grid (line 10) | def make_grid(imgs: torch.Tensor, scale=0.5): function compute_psnr (line 37) | def compute_psnr(img1, img2, mask=None): function torch_rgb_to_gray (line 64) | def torch_rgb_to_gray(image): function compute_gradient_loss (line 76) | def compute_gradient_loss(pred, gt, mask=None): function mark_image_with_red_squares (line 144) | def mark_image_with_red_squares(img): function gaussian (line 157) | def gaussian(window_size, sigma): function create_window (line 168) | def create_window(window_size, channel): function compute_ssim (line 177) | def compute_ssim(img1, img2, window_size=11, size_average=True): function _ssim (line 188) | def _ssim(img1, img2, window, window_size, channel, size_average=True): function compute_low_res_psnr (line 223) | def compute_low_res_psnr(img1, img2, scale_factor): function compute_low_res_mse (line 239) | def compute_low_res_mse(img1, img2, scale_factor): FILE: physdreamer/utils/io_utils.py function read_video_cv2 (line 9) | def read_video_cv2(video_path, rgb=True): function save_video_cv2 (line 31) | def save_video_cv2(video_path, img_list, fps): function save_video_imageio (line 47) | def save_video_imageio(video_path, img_list, fps): function save_gif_imageio (line 60) | def save_gif_imageio(video_path, img_list, fps): function save_video_mediapy (line 71) | def save_video_mediapy(video_frames, output_video_path: str = None, fps:... FILE: physdreamer/utils/optimizer.py function get_linear_schedule_with_warmup (line 5) | def get_linear_schedule_with_warmup( FILE: physdreamer/utils/print_utils.py function print_if_zero_rank (line 4) | def print_if_zero_rank(s): FILE: physdreamer/utils/pytorch_mssim.py function gaussian (line 8) | def gaussian(window_size, sigma): function create_window (line 13) | def create_window(window_size, channel=1): function create_window_3d (line 19) | def create_window_3d(window_size, channel=1): function ssim (line 27) | def ssim(img1, img2, window_size=11, window=None, size_average=True, ful... function ssim_matlab (line 81) | def ssim_matlab(img1, img2, window_size=11, window=None, size_average=Tr... function msssim (line 141) | def msssim(img1, img2, window_size=11, size_average=True, val_range=None... class SSIM (line 171) | class SSIM(torch.nn.Module): method __init__ (line 172) | def __init__(self, window_size=11, size_average=True, val_range=None): method forward (line 182) | def forward(self, img1, img2): class MSSSIM (line 196) | class MSSSIM(torch.nn.Module): method __init__ (line 197) | def __init__(self, window_size=11, size_average=True, channel=3): method forward (line 203) | def forward(self, img1, img2): FILE: physdreamer/utils/svd_helpper.py function init_st (line 18) | def init_st(version_dict, load_ckpt=True, load_filter=True): function load_model_from_config (line 38) | def load_model_from_config(config, ckpt=None, verbose=True): function load_model (line 72) | def load_model(model): function set_lowvram_mode (line 79) | def set_lowvram_mode(mode): function initial_model_load (line 84) | def initial_model_load(model): function unload_model (line 93) | def unload_model(model): function get_unique_embedder_keys_from_conditioner (line 100) | def get_unique_embedder_keys_from_conditioner(conditioner): function get_batch (line 104) | def get_batch(keys, value_dict, N, T, device): FILE: physdreamer/utils/torch_utils.py function get_sync_time (line 5) | def get_sync_time(): FILE: physdreamer/warp_mpm/gaussian_sim_utils.py function get_volume (line 4) | def get_volume(xyzs: np.ndarray, resolution=128) -> np.ndarray: FILE: physdreamer/warp_mpm/mpm_data_structure.py class MPMStateStruct (line 14) | class MPMStateStruct(object): method init (line 41) | def init( method init_grid (line 97) | def init_grid( method from_torch (line 119) | def from_torch( method reset_state (line 174) | def reset_state( method continue_from_torch (line 257) | def continue_from_torch( method set_require_grad (line 294) | def set_require_grad(self, requires_grad=True): method reset_density (line 305) | def reset_density( method partial_clone (line 356) | def partial_clone(self, device="cuda:0", requires_grad=True): class MPMModelStruct (line 391) | class MPMModelStruct(object): method init (line 424) | def init( method finalize_mu_lam (line 448) | def finalize_mu_lam(self, n_particles, device="cuda:0"): method init_other_params (line 456) | def init_other_params(self, n_grid=100, grid_lim=1.0, device="cuda:0"): method from_torch (line 486) | def from_torch( method set_require_grad (line 494) | def set_require_grad(self, requires_grad=True): class Dirichlet_collider (line 503) | class Dirichlet_collider: class GridCollider (line 537) | class GridCollider: class Impulse_modifier (line 548) | class Impulse_modifier: class MPMtailoredStruct (line 564) | class MPMtailoredStruct: class MaterialParamsModifier (line 590) | class MaterialParamsModifier: class ParticleVelocityModifier (line 599) | class ParticleVelocityModifier: function compute_mu_lam_from_E_nu_clean (line 621) | def compute_mu_lam_from_E_nu_clean( function set_vec3_to_zero (line 633) | def set_vec3_to_zero(target_array: wp.array(dtype=wp.vec3)): function set_vec3_to_vec3 (line 639) | def set_vec3_to_vec3( function set_float_vec_to_vec_wmask (line 647) | def set_float_vec_to_vec_wmask( function set_float_vec_to_vec (line 658) | def set_float_vec_to_vec( function set_mat33_to_identity (line 666) | def set_mat33_to_identity(target_array: wp.array(dtype=wp.mat33)): function set_mat33_to_zero (line 672) | def set_mat33_to_zero(target_array: wp.array(dtype=wp.mat33)): function add_identity_to_mat33 (line 678) | def add_identity_to_mat33(target_array: wp.array(dtype=wp.mat33)): function subtract_identity_to_mat33 (line 686) | def subtract_identity_to_mat33(target_array: wp.array(dtype=wp.mat33)): function add_vec3_to_vec3 (line 694) | def add_vec3_to_vec3( function set_value_to_float_array (line 702) | def set_value_to_float_array(target_array: wp.array(dtype=float), value:... function set_warpvalue_to_float_array (line 708) | def set_warpvalue_to_float_array( function get_float_array_product (line 716) | def get_float_array_product( function torch2warp_quat (line 725) | def torch2warp_quat(t, copy=False, dtype=warp.types.float32, dvc="cuda:0"): function torch2warp_float (line 746) | def torch2warp_float(t, copy=False, dtype=warp.types.float32, dvc="cuda:... function torch2warp_vec3 (line 766) | def torch2warp_vec3(t, copy=False, dtype=warp.types.float32, dvc="cuda:0"): function torch2warp_mat33 (line 787) | def torch2warp_mat33(t, copy=False, dtype=warp.types.float32, dvc="cuda:... FILE: physdreamer/warp_mpm/mpm_solver_diff.py class MPMWARPDiff (line 13) | class MPMWARPDiff(object): method __init__ (line 18) | def __init__(self, n_particles, n_grid=100, grid_lim=1.0, device="cuda... method initialize (line 22) | def initialize(self, n_particles, n_grid=100, grid_lim=1.0, device="cu... method set_parameters (line 39) | def set_parameters(self, device="cuda:0", **kwargs): method set_parameters_dict (line 42) | def set_parameters_dict(self, mpm_model, mpm_state, kwargs={}, device=... method set_E_nu (line 110) | def set_E_nu(self, mpm_model, E: float, nu: float, device="cuda:0"): method set_E_nu_from_torch (line 141) | def set_E_nu_from_torch( method prepare_mu_lam (line 160) | def prepare_mu_lam(self, mpm_model, mpm_state, device="cuda:0"): method p2g2p_differentiable (line 169) | def p2g2p_differentiable( method p2g2p (line 297) | def p2g2p(self, mpm_model, mpm_state, step, dt, device="cuda:0"): method print_time_profile (line 435) | def print_time_profile(self): method add_surface_collider (line 441) | def add_surface_collider( method set_velocity_on_cuboid (line 545) | def set_velocity_on_cuboid( method add_bounding_box (line 602) | def add_bounding_box(self, start_time=0.0, end_time=999.0): method add_impulse_on_particles (line 674) | def add_impulse_on_particles( method enforce_particle_velocity_translation (line 724) | def enforce_particle_velocity_translation( method enforce_particle_velocity_rotation (line 772) | def enforce_particle_velocity_rotation( method release_particles_sequentially (line 877) | def release_particles_sequentially( method enforce_particle_velocity_by_mask (line 904) | def enforce_particle_velocity_by_mask( method restart_and_compute_F_C (line 945) | def restart_and_compute_F_C(self, mpm_model, mpm_state, target_pos, de... method enforce_grid_velocity_by_mask (line 995) | def enforce_grid_velocity_by_mask( method add_impulse_on_particles_with_mask (line 1025) | def add_impulse_on_particles_with_mask( FILE: physdreamer/warp_mpm/mpm_utils.py function kirchoff_stress_FCR (line 9) | def kirchoff_stress_FCR( function kirchoff_stress_neoHookean (line 19) | def kirchoff_stress_neoHookean( function kirchoff_stress_StVK (line 51) | def kirchoff_stress_StVK( function kirchoff_stress_drucker_prager (line 70) | def kirchoff_stress_drucker_prager( function von_mises_return_mapping (line 88) | def von_mises_return_mapping(F_trial: wp.mat33, model: MPMModelStruct, p... function von_mises_return_mapping_with_damage (line 134) | def von_mises_return_mapping_with_damage( function viscoplasticity_return_mapping_with_StVK (line 191) | def viscoplasticity_return_mapping_with_StVK( function sand_return_mapping (line 238) | def sand_return_mapping( function compute_mu_lam_from_E_nu (line 278) | def compute_mu_lam_from_E_nu(state: MPMStateStruct, model: MPMModelStruct): function zero_grid (line 287) | def zero_grid(state: MPMStateStruct, model: MPMModelStruct): function compute_dweight (line 295) | def compute_dweight( function update_cov (line 307) | def update_cov(state: MPMStateStruct, p: int, grad_v: wp.mat33, dt: float): function update_cov_differentiable (line 330) | def update_cov_differentiable( function p2g_apic_with_stress (line 359) | def p2g_apic_with_stress(state: MPMStateStruct, model: MPMModelStruct, d... function grid_normalization_and_gravity (line 428) | def grid_normalization_and_gravity( function g2p (line 442) | def g2p(state: MPMStateStruct, model: MPMModelStruct, dt: float): function g2p_differentiable (line 503) | def g2p_differentiable( function clip_particle_x (line 583) | def clip_particle_x(state: MPMStateStruct, model: MPMModelStruct): function compute_stress_from_F_trial (line 604) | def compute_stress_from_F_trial( function add_damping_via_grid (line 723) | def add_damping_via_grid(state: MPMStateStruct, scale: float): function apply_additional_params (line 738) | def apply_additional_params( function selection_add_impulse_on_particles (line 759) | def selection_add_impulse_on_particles( function selection_enforce_particle_velocity_translation (line 775) | def selection_enforce_particle_velocity_translation( function selection_enforce_particle_velocity_cylinder (line 791) | def selection_enforce_particle_velocity_cylinder( function compute_position_l2_loss (line 812) | def compute_position_l2_loss( function aggregate_grad (line 829) | def aggregate_grad(x: wp.array(dtype=float), grad: wp.array(dtype=float)): function set_F_C_p2g (line 837) | def set_F_C_p2g( function set_F_C_g2p (line 874) | def set_F_C_g2p(state: MPMStateStruct, model: MPMModelStruct): function compute_posloss_with_grad (line 923) | def compute_posloss_with_grad( function compute_veloloss_with_grad (line 943) | def compute_veloloss_with_grad( function compute_Floss_with_grad (line 964) | def compute_Floss_with_grad( function compute_Closs_with_grad (line 997) | def compute_Closs_with_grad( FILE: physdreamer/warp_mpm/warp_utils.py function from_torch_safe (line 13) | def from_torch_safe(t, dtype=None, requires_grad=None, grad=None): class MyTape (line 93) | class MyTape(wp.Tape): method get_adjoint (line 95) | def get_adjoint(self, a): class CondTape (line 131) | class CondTape(object): method __init__ (line 132) | def __init__(self, tape: Optional[MyTape], cond: bool = True) -> None: method __enter__ (line 136) | def __enter__(self): method __exit__ (line 140) | def __exit__(self, exc_type, exc_value, traceback): FILE: projects/inference/config_demo.py class DemoParams (line 17) | class DemoParams(object): method __init__ (line 18) | def __init__(self, scene_name): method get_cfg (line 59) | def get_cfg( FILE: projects/inference/demo.py function create_dataset (line 61) | def create_dataset(args): class Trainer (line 94) | class Trainer: method __init__ (line 95) | def __init__(self, args): method init_trainable_params (line 181) | def init_trainable_params( method setup_simulation (line 203) | def setup_simulation(self, dataset_dir, grid_size=100): method add_constant_force (line 412) | def add_constant_force(self, center_point, radius, force, dt, start_ti... method get_simulation_input (line 429) | def get_simulation_input(self, device): method get_material_params (line 471) | def get_material_params(self, device): method load (line 499) | def load(self, checkpoint_dir): method setup_render (line 513) | def setup_render(self, args, gaussian_path, white_background=True): method demo (line 580) | def demo( method compute_metric (line 788) | def compute_metric(self, exp_name, result_dir): function parse_args (line 893) | def parse_args(): FILE: projects/inference/local_utils.py function cycle (line 29) | def cycle(dl: torch.utils.data.DataLoader): function load_motion_model (line 35) | def load_motion_model(model, checkpoint_path): function create_spatial_fields (line 42) | def create_spatial_fields( function create_motion_model (line 68) | def create_motion_model( function create_velocity_model (line 99) | def create_velocity_model( function create_svd_model (line 125) | def create_svd_model(model_name="svd_full", ckpt_path=None): class LinearStepAnneal (line 151) | class LinearStepAnneal(object): method __init__ (line 153) | def __init__( method compute_state (line 175) | def compute_state(self, cur_iter): function setup_boundary_condition (line 187) | def setup_boundary_condition( function setup_plannar_boundary_condition (line 216) | def setup_plannar_boundary_condition( function find_far_points (line 259) | def find_far_points(xyzs, selected_points, thres=0.05): function setup_boundary_condition_with_points (line 289) | def setup_boundary_condition_with_points( function setup_bottom_boundary_condition (line 311) | def setup_bottom_boundary_condition(xyzs, mpm_solver, mpm_state, percent... function render_single_view_video (line 334) | def render_single_view_video( function render_gaussian_seq (line 403) | def render_gaussian_seq(cam, render_params, gaussian_pos_list, gaussian_... function render_gaussian_seq_w_mask (line 430) | def render_gaussian_seq_w_mask( function render_gaussian_seq_w_mask_with_disp (line 470) | def render_gaussian_seq_w_mask_with_disp( function render_gaussian_seq_w_mask_with_disp_for_figure (line 524) | def render_gaussian_seq_w_mask_with_disp_for_figure( function render_gaussian_seq_w_mask_cam_seq (line 591) | def render_gaussian_seq_w_mask_cam_seq( function apply_grid_bc_w_freeze_pts (line 631) | def apply_grid_bc_w_freeze_pts(grid_size, grid_lim, freeze_pts, mpm_solv... function add_constant_force (line 661) | def add_constant_force( function render_force_2d (line 703) | def render_force_2d(cam, render_params, center_point, force): function render_gaussian_seq_w_mask_cam_seq_with_force (line 750) | def render_gaussian_seq_w_mask_cam_seq_with_force( function render_gaussian_seq_w_mask_cam_seq_with_force_with_disp (line 828) | def render_gaussian_seq_w_mask_cam_seq_with_force_with_disp( function downsample_with_kmeans (line 909) | def downsample_with_kmeans(points_array: np.ndarray, num_points: int): function downsample_with_kmeans_gpu (line 935) | def downsample_with_kmeans_gpu(points_array: torch.Tensor, num_points: i... function downsample_with_kmeans_gpu_with_chunk (line 971) | def downsample_with_kmeans_gpu_with_chunk(points_array: torch.Tensor, nu... function interpolate_points (line 1029) | def interpolate_points(query_points, drive_displacement, top_k_index): function interpolate_points_w_R (line 1046) | def interpolate_points_w_R( function create_camera_path (line 1078) | def create_camera_path( function get_camera_trajectory (line 1137) | def get_camera_trajectory(cam, num_pos, camera_cfg: dict, dataset): FILE: projects/uncleaned_train/exp_motion/train/config_demo.py class DemoParams (line 15) | class DemoParams(object): method __init__ (line 16) | def __init__(self): method get_cfg (line 41) | def get_cfg( FILE: projects/uncleaned_train/exp_motion/train/convert_gaussian_to_mesh.py function convert_gaussian_to_mesh (line 9) | def convert_gaussian_to_mesh(gaussian_path, thresh=0.1, save_path=None): function internal_filling (line 33) | def internal_filling(gaussian_path, thresh=2.0, save_path=None, resolut... FILE: projects/uncleaned_train/exp_motion/train/fast_train_velocity.py function create_dataset (line 76) | def create_dataset(args): class Trainer (line 110) | class Trainer: method __init__ (line 111) | def __init__(self, args): method init_trainable_params (line 308) | def init_trainable_params( method setup_simulation (line 338) | def setup_simulation(self, dataset_dir, grid_size=100): method set_simulation_state (line 553) | def set_simulation_state( method get_density_velocity (line 591) | def get_density_velocity(self, time_stamp: float, device, requires_gra... method train_one_step (line 618) | def train_one_step(self): method train (line 875) | def train(self): method inference (line 889) | def inference( method save (line 977) | def save( method load (line 1000) | def load(self, checkpoint_dir): method setup_eval (line 1013) | def setup_eval(self, args, gaussian_path, white_background=True): method eval (line 1087) | def eval( function parse_args (line 1126) | def parse_args(): FILE: projects/uncleaned_train/exp_motion/train/interface.py class MPMDifferentiableSimulation (line 18) | class MPMDifferentiableSimulation(autograd.Function): method forward (line 21) | def forward( method backward (line 155) | def backward(ctx, out_pos_grad: Float[Tensor, "n 3"]): class MPMDifferentiableSimulationWCheckpoint (line 241) | class MPMDifferentiableSimulationWCheckpoint(autograd.Function): method forward (line 249) | def forward( method backward (line 344) | def backward(ctx, out_pos_grad: Float[Tensor, "n 3"], out_velo_grad: F... class MPMDifferentiableSimulationClean (line 543) | class MPMDifferentiableSimulationClean(autograd.Function): method forward (line 551) | def forward( method backward (line 665) | def backward(ctx, out_pos_grad: Float[Tensor, "n 3"], out_velo_grad: F... FILE: projects/uncleaned_train/exp_motion/train/local_utils.py function cycle (line 30) | def cycle(dl: torch.utils.data.DataLoader): function load_motion_model (line 36) | def load_motion_model(model, checkpoint_path): function create_spatial_fields (line 43) | def create_spatial_fields( function create_motion_model (line 85) | def create_motion_model( function create_velocity_model (line 116) | def create_velocity_model( function create_svd_model (line 142) | def create_svd_model(model_name="svd_full", ckpt_path=None): class LinearStepAnneal (line 166) | class LinearStepAnneal(object): method __init__ (line 168) | def __init__( method compute_state (line 190) | def compute_state(self, cur_iter): function setup_boundary_condition (line 202) | def setup_boundary_condition( function setup_plannar_boundary_condition (line 231) | def setup_plannar_boundary_condition( function find_far_points (line 274) | def find_far_points(xyzs, selected_points, thres=0.05): function setup_boundary_condition_with_points (line 304) | def setup_boundary_condition_with_points( function setup_bottom_boundary_condition (line 326) | def setup_bottom_boundary_condition(xyzs, mpm_solver, mpm_state, percent... function render_single_view_video (line 349) | def render_single_view_video( function render_gaussian_seq (line 418) | def render_gaussian_seq(cam, render_params, gaussian_pos_list, gaussian_... function render_gaussian_seq_w_mask (line 445) | def render_gaussian_seq_w_mask( function render_gaussian_seq_w_mask_with_disp (line 485) | def render_gaussian_seq_w_mask_with_disp( function render_gaussian_seq_w_mask_with_disp_for_figure (line 539) | def render_gaussian_seq_w_mask_with_disp_for_figure( function render_gaussian_seq_w_mask_cam_seq (line 606) | def render_gaussian_seq_w_mask_cam_seq( function apply_grid_bc_w_freeze_pts (line 646) | def apply_grid_bc_w_freeze_pts(grid_size, grid_lim, freeze_pts, mpm_solv... function add_constant_force (line 676) | def add_constant_force( function render_force_2d (line 718) | def render_force_2d(cam, render_params, center_point, force): function render_gaussian_seq_w_mask_cam_seq_with_force (line 765) | def render_gaussian_seq_w_mask_cam_seq_with_force( function render_gaussian_seq_w_mask_cam_seq_with_force_with_disp (line 843) | def render_gaussian_seq_w_mask_cam_seq_with_force_with_disp( function downsample_with_kmeans (line 924) | def downsample_with_kmeans(points_array: np.ndarray, num_points: int): function downsample_with_kmeans_gpu (line 950) | def downsample_with_kmeans_gpu(points_array: torch.Tensor, num_points: i... function interpolate_points (line 985) | def interpolate_points(query_points, drive_displacement, top_k_index): function interpolate_points_w_R (line 1002) | def interpolate_points_w_R( function create_camera_path (line 1034) | def create_camera_path( function get_camera_trajectory (line 1093) | def get_camera_trajectory(cam, num_pos, camera_cfg: dict, dataset): FILE: projects/uncleaned_train/exp_motion/train/train_material.py function create_dataset (line 94) | def create_dataset(args): class Trainer (line 132) | class Trainer: method __init__ (line 133) | def __init__(self, args): method init_trainable_params (line 363) | def init_trainable_params( method setup_simulation (line 394) | def setup_simulation(self, dataset_dir, grid_size=100): method get_simulation_input (line 618) | def get_simulation_input(self, device): method get_material_params (line 661) | def get_material_params(self, device): method train_one_step (line 691) | def train_one_step(self): method train (line 1052) | def train(self): method inference (line 1066) | def inference( method save (line 1162) | def save( method load (line 1185) | def load(self, checkpoint_dir): method setup_eval (line 1198) | def setup_eval(self, args, gaussian_path, white_background=True): method demo (line 1272) | def demo( function parse_args (line 1417) | def parse_args(): FILE: projects/uncleaned_train/motionrep/datatools/_convert_fbx_to_mesh.py function convert (line 8) | def convert(fbx_path): function convert2 (line 85) | def convert2(fbx_path): function main (line 191) | def main(): FILE: projects/uncleaned_train/motionrep/datatools/blender_deforming_things4d.py function anime_read (line 9) | def anime_read(filename): FILE: projects/uncleaned_train/motionrep/datatools/blender_render_imgs.py function focal2fov (line 12) | def focal2fov(focal, pixels): function create_camera (line 16) | def create_camera(location, rotation): function set_camera_look_at (line 22) | def set_camera_look_at(camera, target_point): function setup_alpha_mask (line 32) | def setup_alpha_mask(obj_name, pass_index=1): function render_scene (line 68) | def render_scene(camera, output_path): function setup_light (line 84) | def setup_light(): function create_mesh_from_data (line 110) | def create_mesh_from_data(vertices, faces): function write_next_bytes (line 198) | def write_next_bytes(fid, data, format_char_sequence, endian_character="... function write_cameras_binary (line 214) | def write_cameras_binary(cameras, path_to_model_file): function write_images_binary (line 231) | def write_images_binary(images, path_to_model_file): function write_points3D_binary (line 252) | def write_points3D_binary(points3D, path_to_model_file): function get_colmap_camera (line 271) | def get_colmap_camera(camera_obj, render_resolution): function main (line 320) | def main(): FILE: projects/uncleaned_train/motionrep/datatools/deforming_things4d.py function anime_read (line 5) | def anime_read(filename): function extract_trajectory (line 32) | def extract_trajectory( function main (line 55) | def main(): FILE: projects/uncleaned_train/motionrep/datatools/fbx_to_mesh.py function convert_to_mesh (line 6) | def convert_to_mesh(fbx_path, output_dir): function convert_obj_to_traj (line 50) | def convert_obj_to_traj(meshes_dir): function main (line 78) | def main(): FILE: projects/uncleaned_train/motionrep/datatools/fbx_to_mesh_flag.py function convert_to_mesh (line 6) | def convert_to_mesh(fbx_path, output_dir): function subdivde_mesh (line 53) | def subdivde_mesh(mesh_directory, output_directory): function convert_obj_to_traj (line 96) | def convert_obj_to_traj(meshes_dir): function main (line 124) | def main(): FILE: projects/uncleaned_train/motionrep/datatools/render_blender_annimations.py function focal2fov (line 12) | def focal2fov(focal, pixels): function create_camera (line 16) | def create_camera(location, rotation): function set_camera_look_at (line 22) | def set_camera_look_at(camera, target_point): function setup_alpha_mask (line 32) | def setup_alpha_mask(obj_name, pass_index=1): function render_scene (line 68) | def render_scene(camera, output_path): function setup_light (line 84) | def setup_light(): function create_mesh_from_data (line 110) | def create_mesh_from_data(vertices, faces): function write_next_bytes (line 198) | def write_next_bytes(fid, data, format_char_sequence, endian_character="... function write_cameras_binary (line 214) | def write_cameras_binary(cameras, path_to_model_file): function write_images_binary (line 231) | def write_images_binary(images, path_to_model_file): function write_points3D_binary (line 252) | def write_points3D_binary(points3D, path_to_model_file): function get_colmap_camera (line 271) | def get_colmap_camera(camera_obj, render_resolution): function main (line 320) | def main(): FILE: projects/uncleaned_train/motionrep/datatools/render_fbx_first_frame.py function focal2fov (line 12) | def focal2fov(focal, pixels): function create_camera (line 16) | def create_camera(location, rotation): function set_camera_look_at (line 22) | def set_camera_look_at(camera, target_point): function setup_alpha_mask (line 32) | def setup_alpha_mask(obj_name, pass_index=1): function render_scene (line 68) | def render_scene(camera, output_path, mask_name="U3DMesh"): function normalize_mesh (line 84) | def normalize_mesh(obj): function setup_light (line 115) | def setup_light(): function create_mesh_from_fpx (line 153) | def create_mesh_from_fpx(fbx_path): function get_colmap_camera (line 173) | def get_colmap_camera(camera_obj, render_resolution): function get_textures (line 222) | def get_textures( function main (line 303) | def main(): function find_material (line 398) | def find_material(): FILE: projects/uncleaned_train/motionrep/datatools/render_obj.py function focal2fov (line 12) | def focal2fov(focal, pixels): function create_camera (line 16) | def create_camera(location, rotation): function set_camera_look_at (line 22) | def set_camera_look_at(camera, target_point): function setup_alpha_mask (line 32) | def setup_alpha_mask(obj_name, pass_index=1): function render_scene (line 68) | def render_scene(camera, output_path, mask_name="U3DMesh"): function setup_light (line 84) | def setup_light(): function create_mesh_from_obj (line 122) | def create_mesh_from_obj(obj_file_path): function get_focal_length (line 143) | def get_focal_length(camera_obj, render_resolution): function normalize_mesh (line 176) | def normalize_mesh(transform_meta_path, mesh_objects): function apply_rotation (line 196) | def apply_rotation(mesh_objects): function main (line 209) | def main(): FILE: projects/uncleaned_train/motionrep/datatools/render_obj_external_texture.py function focal2fov (line 12) | def focal2fov(focal, pixels): function create_camera (line 16) | def create_camera(location, rotation): function set_camera_look_at (line 22) | def set_camera_look_at(camera, target_point): function setup_alpha_mask (line 32) | def setup_alpha_mask(obj_name, pass_index=1): function render_scene (line 68) | def render_scene(camera, output_path, mask_name="U3DMesh"): function setup_light (line 84) | def setup_light(): function create_mesh_from_obj (line 122) | def create_mesh_from_obj(obj_file_path): function get_focal_length (line 143) | def get_focal_length(camera_obj, render_resolution): function normalize_mesh (line 176) | def normalize_mesh(transform_meta_path, mesh_objects): function apply_rotation (line 196) | def apply_rotation(mesh_objects): function get_textures (line 209) | def get_textures( function main (line 290) | def main(): FILE: projects/uncleaned_train/motionrep/datatools/test_colmap_camera.py function qvec2rotmat (line 42) | def qvec2rotmat(qvec): function rotmat2qvec (line 64) | def rotmat2qvec(R): function fov2focal (line 84) | def fov2focal(fov, pixels): class Image (line 88) | class Image(BaseImage): method qvec2rotmat (line 89) | def qvec2rotmat(self): function read_next_bytes (line 93) | def read_next_bytes(fid, num_bytes, format_char_sequence, endian_charact... function read_points3D_text (line 105) | def read_points3D_text(path): function read_points3D_binary (line 147) | def read_points3D_binary(path_to_model_file): function read_intrinsics_text (line 182) | def read_intrinsics_text(path): function read_extrinsics_binary (line 209) | def read_extrinsics_binary(path_to_model_file): function read_intrinsics_binary (line 255) | def read_intrinsics_binary(path_to_model_file): function read_extrinsics_text (line 288) | def read_extrinsics_text(path): function read_colmap_bin_array (line 323) | def read_colmap_bin_array(path): class CameraInfo (line 348) | class CameraInfo(NamedTuple): function focal2fov (line 361) | def focal2fov(focal, pixels): function readColmapCameras (line 365) | def readColmapCameras(cam_extrinsics, cam_intrinsics, images_folder): function read_camera_points (line 417) | def read_camera_points(dir_path): function extract_projection_matrix (line 436) | def extract_projection_matrix(cam_info): function main (line 462) | def main(): FILE: projects/uncleaned_train/motionrep/datatools/transform_obj_for_blender.py function transform_vertex (line 8) | def transform_vertex(vertex: np.ndarray, transform_dict): function colmap_to_blender_transform (line 26) | def colmap_to_blender_transform(vertex: np.ndarray): function copy_mtl_file (line 35) | def copy_mtl_file(obj_path, transformed_obj_path): function main (line 46) | def main(): FILE: projects/uncleaned_train/motionrep/diffusion/builder.py function create_gaussian_diffusion (line 5) | def create_gaussian_diffusion( FILE: projects/uncleaned_train/motionrep/diffusion/discretizer.py class EDMResShiftedDiscretization (line 6) | class EDMResShiftedDiscretization(Discretization): method __init__ (line 7) | def __init__( method get_sigmas (line 15) | def get_sigmas(self, n, device="cpu"): FILE: projects/uncleaned_train/motionrep/diffusion/draft.py function latent_sds (line 5) | def latent_sds(input_x, schduler, unet, t_range=[0.02, 0.98]): FILE: projects/uncleaned_train/motionrep/diffusion/gaussian_diffusion.py function mean_flat (line 19) | def mean_flat(tensor): function get_named_beta_schedule (line 26) | def get_named_beta_schedule(schedule_name, num_diffusion_timesteps): function betas_for_alpha_bar (line 53) | def betas_for_alpha_bar(num_diffusion_timesteps, alpha_bar, max_beta=0.9... class ModelMeanType (line 73) | class ModelMeanType(enum.Enum): class ModelVarType (line 83) | class ModelVarType(enum.Enum): class LossType (line 97) | class LossType(enum.Enum): method is_vb (line 105) | def is_vb(self): class GaussianDiffusion (line 109) | class GaussianDiffusion: method __init__ (line 126) | def __init__( method q_mean_variance (line 179) | def q_mean_variance(self, x_start, t): method q_sample (line 196) | def q_sample(self, x_start, t, noise=None): method q_posterior_mean_variance (line 216) | def q_posterior_mean_variance(self, x_start, x_t, t): method p_mean_variance (line 240) | def p_mean_variance( method _predict_xstart_from_eps (line 336) | def _predict_xstart_from_eps(self, x_t, t, eps): method _predict_xstart_from_xprev (line 343) | def _predict_xstart_from_xprev(self, x_t, t, xprev): method _predict_eps_from_xstart (line 353) | def _predict_eps_from_xstart(self, x_t, t, pred_xstart): method _scale_timesteps (line 359) | def _scale_timesteps(self, t): method condition_mean (line 364) | def condition_mean(self, cond_fn, p_mean_var, x, t, model_kwargs=None): method condition_score (line 379) | def condition_score(self, cond_fn, p_mean_var, x, t, model_kwargs=None): method p_sample (line 403) | def p_sample( method p_sample_loop (line 449) | def p_sample_loop( method p_sample_loop_progressive (line 495) | def p_sample_loop_progressive( method ddim_sample (line 545) | def ddim_sample( method ddim_reverse_sample (line 611) | def ddim_reverse_sample( method ddim_sample_loop (line 649) | def ddim_sample_loop( method ddim_sample_loop_progressive (line 689) | def ddim_sample_loop_progressive( method _vb_terms_bpd (line 745) | def _vb_terms_bpd( method training_losses (line 780) | def training_losses(self, model, x_start, t, model_kwargs=None, noise=... method _prior_bpd (line 876) | def _prior_bpd(self, x_start): method calc_bpd_loop (line 894) | def calc_bpd_loop(self, model, x_start, clip_denoised=True, model_kwar... function _extract_into_tensor (line 952) | def _extract_into_tensor(arr, timesteps, broadcast_shape): FILE: projects/uncleaned_train/motionrep/diffusion/losses.py function normal_kl (line 12) | def normal_kl(mean1, logvar1, mean2, logvar2): function approx_standard_normal_cdf (line 42) | def approx_standard_normal_cdf(x): function discretized_gaussian_log_likelihood (line 50) | def discretized_gaussian_log_likelihood(x, *, means, log_scales): FILE: projects/uncleaned_train/motionrep/diffusion/resample.py function create_named_schedule_sampler (line 12) | def create_named_schedule_sampler(name, diffusion): class ScheduleSampler (line 27) | class ScheduleSampler(ABC): method weights (line 39) | def weights(self): method sample (line 46) | def sample(self, batch_size, device): class UniformSampler (line 65) | class UniformSampler(ScheduleSampler): method __init__ (line 66) | def __init__(self, diffusion): method weights (line 70) | def weights(self): class LossAwareSampler (line 74) | class LossAwareSampler(ScheduleSampler): method update_with_local_losses (line 75) | def update_with_local_losses(self, local_ts, local_losses): method update_with_all_losses (line 111) | def update_with_all_losses(self, ts, losses): class LossSecondMomentResampler (line 128) | class LossSecondMomentResampler(LossAwareSampler): method __init__ (line 129) | def __init__(self, diffusion, history_per_term=10, uniform_prob=0.001): method weights (line 138) | def weights(self): method update_with_all_losses (line 147) | def update_with_all_losses(self, ts, losses): method _warmed_up (line 157) | def _warmed_up(self): FILE: projects/uncleaned_train/motionrep/diffusion/respace.py function space_timesteps (line 7) | def space_timesteps(num_timesteps, section_counts): class SpacedDiffusion (line 63) | class SpacedDiffusion(GaussianDiffusion): method __init__ (line 72) | def __init__(self, use_timesteps, **kwargs): method p_mean_variance (line 88) | def p_mean_variance( method training_losses (line 93) | def training_losses( method condition_mean (line 98) | def condition_mean(self, cond_fn, *args, **kwargs): method condition_score (line 101) | def condition_score(self, cond_fn, *args, **kwargs): method _wrap_model (line 104) | def _wrap_model(self, model): method _scale_timesteps (line 111) | def _scale_timesteps(self, t): class _WrappedModel (line 116) | class _WrappedModel: method __init__ (line 117) | def __init__(self, model, timestep_map, rescale_timesteps, original_nu... method __call__ (line 123) | def __call__(self, x, ts, **kwargs): FILE: projects/uncleaned_train/motionrep/diffusion/sigma_sampling.py function exists (line 7) | def exists(x): function default (line 11) | def default(val, d): class EDMSamplingWithResShift (line 17) | class EDMSamplingWithResShift: method __init__ (line 18) | def __init__(self, p_mean=-1.2, p_std=1.2, scale_shift=320.0 / 576): method __call__ (line 23) | def __call__(self, n_samples, rand=None): FILE: projects/uncleaned_train/motionrep/diffusion/sv_diffusion_engine.py class SVDiffusionEngine (line 25) | class SVDiffusionEngine(pl.LightningModule): method __init__ (line 30) | def __init__( method init_from_ckpt (line 103) | def init_from_ckpt( method _init_first_stage (line 126) | def _init_first_stage(self, config): method get_input (line 133) | def get_input(self, batch): method decode_first_stage (line 139) | def decode_first_stage(self, z): method encode_first_stage (line 159) | def encode_first_stage(self, x): method forward (line 173) | def forward(self, batch, training=True): method shared_step (line 186) | def shared_step(self, batch: Dict) -> Any: method training_step (line 193) | def training_step(self, batch, batch_idx): method on_train_start (line 217) | def on_train_start(self, *args, **kwargs): method on_train_batch_end (line 221) | def on_train_batch_end(self, *args, **kwargs): method ema_scope (line 226) | def ema_scope(self, context=None): method instantiate_optimizer_from_config (line 240) | def instantiate_optimizer_from_config(self, params, lr, cfg): method configure_optimizers (line 245) | def configure_optimizers(self): method get_trainable_parameters (line 265) | def get_trainable_parameters(self): method sample (line 280) | def sample( method log_conditionings (line 297) | def log_conditionings(self, batch: Dict, n: int) -> Dict: method log_images (line 336) | def log_images( FILE: projects/uncleaned_train/motionrep/diffusion/svd_conditioner.py class SVDConditioner (line 19) | class SVDConditioner(GeneralConditioner): method __init__ (line 23) | def __init__(self, emb_models: Union[List, ListConfig]): method forward (line 26) | def forward( FILE: projects/uncleaned_train/motionrep/diffusion/svd_sds_engine.py class SVDSDSEngine (line 32) | class SVDSDSEngine(pl.LightningModule): method __init__ (line 37) | def __init__( method init_from_ckpt (line 98) | def init_from_ckpt( method _init_first_stage (line 122) | def _init_first_stage(self, config): method get_input (line 132) | def get_input(self, batch): method encode_first_stage (line 137) | def encode_first_stage(self, x): method forward (line 151) | def forward(self, batch, sample_time_range=[0.02, 0.98]): method forward_with_encoder_chunk (line 177) | def forward_with_encoder_chunk( method edm_sds (line 227) | def edm_sds(self, input_x, extra_input, sample_time_range=[0.02, 0.98]): method edm_sds_multistep (line 327) | def edm_sds_multistep(self, input_x, extra_input, sample_time_range=[0... method sampler_step (line 433) | def sampler_step(self, sigma, noised_input, c, uc=None, num_frames=Non... FILE: projects/uncleaned_train/motionrep/diffusion/svd_sds_engine_backup.py class SVDSDSEngine (line 30) | class SVDSDSEngine(pl.LightningModule): method __init__ (line 35) | def __init__( method init_from_ckpt (line 86) | def init_from_ckpt( method _init_first_stage (line 110) | def _init_first_stage(self, config): method get_input (line 120) | def get_input(self, batch): method encode_first_stage (line 125) | def encode_first_stage(self, x): method forward (line 139) | def forward(self, batch, training=True): method forward_with_encoder_chunk (line 159) | def forward_with_encoder_chunk(self, batch, chunk_size=2): method emd_sds (line 203) | def emd_sds(self, input_x, extra_input): FILE: projects/uncleaned_train/motionrep/diffusion/svd_sds_wdecoder_engine.py class SVDWDecSDSEngine (line 33) | class SVDWDecSDSEngine(pl.LightningModule): method __init__ (line 38) | def __init__( method init_from_ckpt (line 99) | def init_from_ckpt( method _init_first_stage (line 123) | def _init_first_stage(self, config): method get_input (line 130) | def get_input(self, batch): method encode_first_stage (line 135) | def encode_first_stage(self, x): method decode_first_stage (line 150) | def decode_first_stage(self, z): method forward (line 169) | def forward(self, batch, sample_time_range=[0.02, 0.98]): method forward_with_encoder_chunk (line 195) | def forward_with_encoder_chunk( method edm_sds (line 245) | def edm_sds(self, input_x, extra_input, sample_time_range=[0.02, 0.98]): method edm_sds_multistep (line 346) | def edm_sds_multistep(self, input_x, extra_input, sample_time_range=[0... method resample_multistep (line 453) | def resample_multistep(self, input_x, extra_input, sample_time_range=[... method sampler_step (line 558) | def sampler_step(self, sigma, noised_input, c, uc=None, num_frames=Non... FILE: projects/uncleaned_train/motionrep/diffusion/video_diffusion_loss.py class StandardVideoDiffusionLoss (line 13) | class StandardVideoDiffusionLoss(nn.Module): method __init__ (line 14) | def __init__( method get_noised_input (line 43) | def get_noised_input( method forward (line 49) | def forward( method _forward (line 65) | def _forward( method get_loss (line 109) | def get_loss(self, model_output, target, w): FILE: projects/uncleaned_train/motionrep/field_components/encoding.py class TemporalKplanesEncoding (line 9) | class TemporalKplanesEncoding(nn.Module): method __init__ (line 16) | def __init__( method forward (line 57) | def forward(self, inp: Float[Tensor, "*bs 4"]): method compute_temporal_smoothness (line 83) | def compute_temporal_smoothness( method compute_plane_tv (line 93) | def compute_plane_tv( method visualize (line 103) | def visualize( method functional_forward (line 114) | def functional_forward( class TriplanesEncoding (line 149) | class TriplanesEncoding(nn.Module): method __init__ (line 156) | def __init__( method forward (line 196) | def forward(self, inp: Float[Tensor, "*bs 3"]): method compute_plane_tv (line 222) | def compute_plane_tv( class PlaneEncoding (line 233) | class PlaneEncoding(nn.Module): method __init__ (line 240) | def __init__( method forward (line 273) | def forward(self, inp: Float[Tensor, "*bs 2"]): method compute_plane_tv (line 288) | def compute_plane_tv( class TemporalNeRFEncoding (line 299) | class TemporalNeRFEncoding(nn.Module): method __init__ (line 300) | def __init__( method get_out_dim (line 317) | def get_out_dim(self) -> int: method forward (line 325) | def forward( FILE: projects/uncleaned_train/motionrep/field_components/mlp.py class MLP (line 11) | class MLP(nn.Module): method __init__ (line 12) | def __init__( method build_nn_modules (line 40) | def build_nn_modules(self) -> None: method pytorch_fwd (line 65) | def pytorch_fwd( method forward (line 88) | def forward( FILE: projects/uncleaned_train/motionrep/fields/dct_trajectory_field.py class DCTTrajctoryField (line 7) | class DCTTrajctoryField(nn.Module): method __init__ (line 8) | def __init__( method forward (line 14) | def forward(self, x): method query_points_at_time (line 17) | def query_points_at_time(self, x, t): FILE: projects/uncleaned_train/motionrep/fields/discrete_field.py class PointSetMotionSE3 (line 15) | class PointSetMotionSE3(nn.Module): method __init__ (line 26) | def __init__( method construct_knn (line 70) | def construct_knn(self, inpx: Float[Tensor, "*bs 3"], topk=10, chunk_s... method prepare_isometry (line 94) | def prepare_isometry(self, points, knn_ind): method _forward_single_time (line 105) | def _forward_single_time(self, time_ind: int): method forward (line 121) | def forward( method compute_smoothess_loss (line 137) | def compute_smoothess_loss( method compute_arap_loss (line 146) | def compute_arap_loss( method compute_isometry_loss (line 181) | def compute_isometry_loss( method compute_loss (line 199) | def compute_loss( FILE: projects/uncleaned_train/motionrep/fields/mul_offset_field.py class MulTemporalKplanesOffsetfields (line 15) | class MulTemporalKplanesOffsetfields(nn.Module): method __init__ (line 28) | def __init__( method forward (line 74) | def forward( method compute_smoothess_loss (line 98) | def compute_smoothess_loss( method compute_loss (line 117) | def compute_loss( method arap_loss (line 133) | def arap_loss(self, inp): FILE: projects/uncleaned_train/motionrep/fields/mul_se3_field.py class MulTemporalKplanesSE3fields (line 15) | class MulTemporalKplanesSE3fields(nn.Module): method __init__ (line 26) | def __init__( method forward (line 75) | def forward( method compute_smoothess_loss (line 111) | def compute_smoothess_loss( method compute_loss (line 130) | def compute_loss( FILE: projects/uncleaned_train/motionrep/fields/offset_field.py class TemporalKplanesOffsetfields (line 15) | class TemporalKplanesOffsetfields(nn.Module): method __init__ (line 26) | def __init__( method forward (line 66) | def forward( method compute_smoothess_loss (line 87) | def compute_smoothess_loss( method compute_loss (line 100) | def compute_loss( method arap_loss (line 116) | def arap_loss(self, inp): method forward_with_plane_coefs (line 119) | def forward_with_plane_coefs( FILE: projects/uncleaned_train/motionrep/fields/se3_field.py class TemporalKplanesSE3fields (line 15) | class TemporalKplanesSE3fields(nn.Module): method __init__ (line 26) | def __init__( method forward (line 69) | def forward( method compute_smoothess_loss (line 120) | def compute_smoothess_loss( method compute_loss (line 133) | def compute_loss( FILE: projects/uncleaned_train/motionrep/fields/triplane_field.py class TriplaneFields (line 11) | class TriplaneFields(nn.Module): method __init__ (line 22) | def __init__( method forward (line 55) | def forward( method compute_smoothess_loss (line 70) | def compute_smoothess_loss( function compute_entropy (line 78) | def compute_entropy(p): class TriplaneFieldsWithEntropy (line 82) | class TriplaneFieldsWithEntropy(nn.Module): method __init__ (line 93) | def __init__( method forward (line 130) | def forward( method compute_smoothess_loss (line 152) | def compute_smoothess_loss( FILE: projects/uncleaned_train/motionrep/fields/video_triplane_disp_field.py class TriplaneDispFields (line 16) | class TriplaneDispFields(nn.Module): method __init__ (line 27) | def __init__( method forward (line 82) | def forward( method compute_smoothess_loss (line 102) | def compute_smoothess_loss( method get_canonical (line 110) | def get_canonical( method sample_canonical (line 127) | def sample_canonical( class PlaneDynamicDispFields (line 180) | class PlaneDynamicDispFields(nn.Module): method __init__ (line 191) | def __init__( method forward (line 255) | def forward( method compute_smoothess_loss (line 281) | def compute_smoothess_loss( method get_canonical (line 289) | def get_canonical( method sample_canonical (line 306) | def sample_canonical( FILE: projects/uncleaned_train/motionrep/gaussian_3d/arguments/__init__.py class GroupParams (line 16) | class GroupParams: class ParamGroup (line 19) | class ParamGroup: method __init__ (line 20) | def __init__(self, parser: ArgumentParser, name : str, fill_none = Fal... method extract (line 40) | def extract(self, args): class ModelParams (line 47) | class ModelParams(ParamGroup): method __init__ (line 48) | def __init__(self, parser, sentinel=False): method extract (line 59) | def extract(self, args): class PipelineParams (line 64) | class PipelineParams(ParamGroup): method __init__ (line 65) | def __init__(self, parser): class OptimizationParams (line 71) | class OptimizationParams(ParamGroup): method __init__ (line 72) | def __init__(self, parser): function get_combined_args (line 91) | def get_combined_args(parser : ArgumentParser): FILE: projects/uncleaned_train/motionrep/gaussian_3d/gaussian_renderer/depth_uv_render.py function render_uv_depth_w_gaussian (line 12) | def render_uv_depth_w_gaussian( FILE: projects/uncleaned_train/motionrep/gaussian_3d/gaussian_renderer/feat_render.py function render_feat_gaussian (line 12) | def render_feat_gaussian( FILE: projects/uncleaned_train/motionrep/gaussian_3d/gaussian_renderer/flow_depth_render.py function render_flow_depth_w_gaussian (line 12) | def render_flow_depth_w_gaussian( FILE: projects/uncleaned_train/motionrep/gaussian_3d/gaussian_renderer/motion_renderer.py function render_motion_w_gaussian (line 9) | def render_motion_w_gaussian( FILE: projects/uncleaned_train/motionrep/gaussian_3d/gaussian_renderer/render.py function render_gaussian (line 21) | def render_gaussian( function gaussian_intrin_scale (line 127) | def gaussian_intrin_scale(x_or_y: torch.Tensor, w_or_h: float): function render_arrow_in_screen (line 134) | def render_arrow_in_screen(viewpoint_camera, points_3d): function render_arrow_in_screen_back (line 190) | def render_arrow_in_screen_back(viewpoint_camera, points_3d): function eval_sh (line 261) | def eval_sh(deg, sh, dirs): function RGB2SH (line 327) | def RGB2SH(rgb): function SH2RGB (line 331) | def SH2RGB(sh): FILE: projects/uncleaned_train/motionrep/gaussian_3d/scene/__init__.py class Scene (line 26) | class Scene: method __init__ (line 29) | def __init__( method save (line 116) | def save(self, iteration): method getTrainCameras (line 122) | def getTrainCameras(self, scale=1.0): method getTestCameras (line 125) | def getTestCameras(self, scale=1.0): FILE: projects/uncleaned_train/motionrep/gaussian_3d/scene/cameras.py class Camera (line 21) | class Camera(nn.Module): method __init__ (line 22) | def __init__( class MiniCam (line 91) | class MiniCam: method __init__ (line 92) | def __init__( FILE: projects/uncleaned_train/motionrep/gaussian_3d/scene/colmap_loader.py function qvec2rotmat (line 43) | def qvec2rotmat(qvec): function rotmat2qvec (line 55) | def rotmat2qvec(R): class Image (line 68) | class Image(BaseImage): method qvec2rotmat (line 69) | def qvec2rotmat(self): function read_next_bytes (line 72) | def read_next_bytes(fid, num_bytes, format_char_sequence, endian_charact... function read_points3D_text (line 83) | def read_points3D_text(path): function read_points3D_binary (line 113) | def read_points3D_binary(path_to_model_file): function read_intrinsics_text (line 144) | def read_intrinsics_text(path): function read_extrinsics_binary (line 168) | def read_extrinsics_binary(path_to_model_file): function read_intrinsics_binary (line 203) | def read_intrinsics_binary(path_to_model_file): function read_extrinsics_text (line 232) | def read_extrinsics_text(path): function read_colmap_bin_array (line 261) | def read_colmap_bin_array(path): FILE: projects/uncleaned_train/motionrep/gaussian_3d/scene/dataset_readers.py class CameraInfo (line 45) | class CameraInfo(NamedTuple): class SceneInfo (line 58) | class SceneInfo(NamedTuple): function getNerfppNorm (line 66) | def getNerfppNorm(cam_info): function readColmapCameras (line 90) | def readColmapCameras(cam_extrinsics, cam_intrinsics, images_folder): function fetchPly (line 142) | def fetchPly(path): function storePly (line 151) | def storePly(path, xyz, rgb): function readColmapSceneInfo (line 177) | def readColmapSceneInfo(path, images, eval, llffhold=8): function readCamerasFromTransforms (line 233) | def readCamerasFromTransforms(path, transformsfile, white_background, ex... function readNerfSyntheticInfo (line 297) | def readNerfSyntheticInfo(path, white_background, eval, extension=".png"): class NoImageCamera (line 349) | class NoImageCamera(nn.Module): method __init__ (line 350) | def __init__( function fast_read_cameras_from_transform_file (line 425) | def fast_read_cameras_from_transform_file(file_path, width=1080, height=... FILE: projects/uncleaned_train/motionrep/gaussian_3d/scene/gaussian_model.py class GaussianModel (line 37) | class GaussianModel: method setup_functions (line 38) | def setup_functions(self): method __init__ (line 55) | def __init__(self, sh_degree: int = 3): method capture (line 74) | def capture(self): method restore (line 95) | def restore(self, model_args, training_args): method capture_training_args (line 118) | def capture_training_args( method get_scaling (line 124) | def get_scaling(self): method get_rotation (line 128) | def get_rotation(self): method get_xyz (line 132) | def get_xyz(self): method get_features (line 136) | def get_features(self): method get_opacity (line 142) | def get_opacity(self): method get_covariance (line 145) | def get_covariance(self, scaling_modifier=1): method oneupSHdegree (line 150) | def oneupSHdegree(self): method create_from_pcd (line 154) | def create_from_pcd(self, pcd: BasicPointCloud, spatial_lr_scale: float): method training_setup (line 196) | def training_setup(self, training_args): method update_learning_rate (line 242) | def update_learning_rate(self, iteration): method construct_list_of_attributes (line 250) | def construct_list_of_attributes(self): method save_ply (line 264) | def save_ply(self, path): method reset_opacity (line 301) | def reset_opacity(self): method load_ply (line 308) | def load_ply(self, path): method replace_tensor_to_optimizer (line 388) | def replace_tensor_to_optimizer(self, tensor, name): method _prune_optimizer (line 403) | def _prune_optimizer(self, mask): method prune_points (line 425) | def prune_points(self, mask): method cat_tensors_to_optimizer (line 441) | def cat_tensors_to_optimizer(self, tensors_dict): method densification_postfix (line 475) | def densification_postfix( method densify_and_split (line 505) | def densify_and_split(self, grads, grad_threshold, scene_extent, N=2): method densify_and_clone (line 549) | def densify_and_clone(self, grads, grad_threshold, scene_extent): method densify_and_prune (line 576) | def densify_and_prune(self, max_grad, min_opacity, extent, max_screen_... method add_densification_stats (line 594) | def add_densification_stats(self, viewspace_point_tensor, update_filter): method apply_discrete_offset_filds (line 600) | def apply_discrete_offset_filds(self, origin_points, offsets): method apply_discrete_offset_filds_with_R (line 646) | def apply_discrete_offset_filds_with_R(self, origin_points, offsets, t... method apply_se3_fields (line 711) | def apply_se3_fields( method apply_offset_fields (line 772) | def apply_offset_fields(self, offset_field, timestamp: float): method apply_offset_fields_with_R (line 812) | def apply_offset_fields_with_R(self, offset_field, timestamp: float, e... method init_from_mesh (line 879) | def init_from_mesh( method detach_grad (line 943) | def detach_grad( method apply_mask (line 953) | def apply_mask(self, mask): method extract_fields (line 982) | def extract_fields(self, resolution=128, num_blocks=16, relax_ratio=1.5): method extract_mesh (line 1073) | def extract_mesh(self, path, density_thresh=1, resolution=128, decimat... function gaussian_3d_coeff (line 1109) | def gaussian_3d_coeff(xyzs, covs): FILE: projects/uncleaned_train/motionrep/gaussian_3d/scene/mesh.py function dot (line 8) | def dot(x, y): function length (line 12) | def length(x, eps=1e-20): function safe_normalize (line 16) | def safe_normalize(x, eps=1e-20): class Mesh (line 20) | class Mesh: method __init__ (line 21) | def __init__( method load (line 49) | def load( method load_obj (line 146) | def load_obj(cls, path, albedo_path=None, device=None): method load_trimesh (line 296) | def load_trimesh(cls, path, device=None): method aabb (line 390) | def aabb(self): method auto_size (line 395) | def auto_size(self): method auto_normal (line 401) | def auto_normal(self): method auto_uv (line 424) | def auto_uv(self, cache_path=None, vmap=True): method align_v_to_vt (line 459) | def align_v_to_vt(self, vmapping=None): method to (line 476) | def to(self, device): method write (line 484) | def write(self, path): method write_ply (line 495) | def write_ply(self, path): method write_glb (line 503) | def write_glb(self, path): method write_obj (line 653) | def write_obj(self, path): FILE: projects/uncleaned_train/motionrep/gaussian_3d/scene/mesh_utils.py function poisson_mesh_reconstruction (line 5) | def poisson_mesh_reconstruction(points, normals=None): function decimate_mesh (line 44) | def decimate_mesh( function clean_mesh (line 88) | def clean_mesh( FILE: projects/uncleaned_train/motionrep/gaussian_3d/utils/camera_utils.py function loadCam (line 21) | def loadCam(args, id, cam_info, resolution_scale): function cameraList_from_camInfos (line 69) | def cameraList_from_camInfos(cam_infos, resolution_scale, args): function camera_to_JSON (line 78) | def camera_to_JSON(id, camera: Camera): function look_at (line 101) | def look_at(from_point, to_point, up_vector=(0, 1, 0)): function create_cameras_around_sphere (line 149) | def create_cameras_around_sphere( FILE: projects/uncleaned_train/motionrep/gaussian_3d/utils/general_utils.py function inverse_sigmoid (line 18) | def inverse_sigmoid(x): function PILtoTorch (line 21) | def PILtoTorch(pil_image, resolution): function get_expon_lr_func (line 29) | def get_expon_lr_func( function strip_lowerdiag (line 64) | def strip_lowerdiag(L): function strip_symmetric (line 75) | def strip_symmetric(sym): function build_rotation (line 78) | def build_rotation(r): function build_scaling_rotation (line 101) | def build_scaling_rotation(s, r): function safe_state (line 112) | def safe_state(silent): FILE: projects/uncleaned_train/motionrep/gaussian_3d/utils/graphics_utils.py class BasicPointCloud (line 17) | class BasicPointCloud(NamedTuple): function geom_transform_points (line 22) | def geom_transform_points(points, transf_matrix): function getWorld2View (line 31) | def getWorld2View(R, t): function getWorld2View2 (line 38) | def getWorld2View2(R, t, translate=np.array([.0, .0, .0]), scale=1.0): function getProjectionMatrix (line 51) | def getProjectionMatrix(znear, zfar, fovX, fovY): function fov2focal (line 73) | def fov2focal(fov, pixels): function focal2fov (line 76) | def focal2fov(focal, pixels): FILE: projects/uncleaned_train/motionrep/gaussian_3d/utils/image_utils.py function mse (line 14) | def mse(img1, img2): function psnr (line 17) | def psnr(img1, img2): FILE: projects/uncleaned_train/motionrep/gaussian_3d/utils/loss_utils.py function l1_loss (line 17) | def l1_loss(network_output, gt): function l2_loss (line 20) | def l2_loss(network_output, gt): function gaussian (line 23) | def gaussian(window_size, sigma): function create_window (line 27) | def create_window(window_size, channel): function ssim (line 33) | def ssim(img1, img2, window_size=11, size_average=True): function _ssim (line 43) | def _ssim(img1, img2, window, window_size, channel, size_average=True): FILE: projects/uncleaned_train/motionrep/gaussian_3d/utils/rigid_body_utils.py function get_rigid_transform (line 5) | def get_rigid_transform(A, B): function _test_rigid_transform (line 56) | def _test_rigid_transform(): function _sqrt_positive_part (line 72) | def _sqrt_positive_part(x: torch.Tensor) -> torch.Tensor: function matrix_to_quaternion (line 83) | def matrix_to_quaternion(matrix: torch.Tensor) -> torch.Tensor: function quternion_to_matrix (line 146) | def quternion_to_matrix(r): function standardize_quaternion (line 172) | def standardize_quaternion(quaternions: torch.Tensor) -> torch.Tensor: function quaternion_multiply (line 188) | def quaternion_multiply(a: torch.Tensor, b: torch.Tensor) -> torch.Tensor: function _test_matrix_to_quaternion (line 212) | def _test_matrix_to_quaternion(): function _test_matrix_to_quaternion_2 (line 241) | def _test_matrix_to_quaternion_2(): FILE: projects/uncleaned_train/motionrep/gaussian_3d/utils/sh_utils.py function eval_sh (line 57) | def eval_sh(deg, sh, dirs): function RGB2SH (line 114) | def RGB2SH(rgb): function SH2RGB (line 117) | def SH2RGB(sh): FILE: projects/uncleaned_train/motionrep/gaussian_3d/utils/system_utils.py function mkdir_p (line 16) | def mkdir_p(folder_path): function searchForMaxIteration (line 26) | def searchForMaxIteration(folder): FILE: projects/uncleaned_train/motionrep/losses/smoothness_loss.py function compute_plane_tv (line 5) | def compute_plane_tv(t: torch.Tensor, only_w: bool = False) -> float: function compute_plane_smoothness (line 26) | def compute_plane_smoothness(t: torch.Tensor) -> float: FILE: projects/uncleaned_train/motionrep/operators/dct.py function dct1_rfft_impl (line 12) | def dct1_rfft_impl(x): function dct_fft_impl (line 16) | def dct_fft_impl(v): function idct_irfft_impl (line 20) | def idct_irfft_impl(V): function dct (line 24) | def dct(x, norm=None): function idct (line 63) | def idct(X, norm=None): function dct_3d (line 110) | def dct_3d(x, norm=None): function idct_3d (line 127) | def idct_3d(X, norm=None): function code_test_dct3d (line 146) | def code_test_dct3d(): FILE: projects/uncleaned_train/motionrep/operators/np_operators.py function feature_map_to_rgb_pca (line 7) | def feature_map_to_rgb_pca(feature_map): FILE: projects/uncleaned_train/motionrep/operators/rotation.py function rotation_6d_to_matrix (line 7) | def rotation_6d_to_matrix(d6: torch.Tensor) -> torch.Tensor: function matrix_to_rotation_6d (line 31) | def matrix_to_rotation_6d(matrix: torch.Tensor) -> torch.Tensor: function quaternion_to_matrix (line 50) | def quaternion_to_matrix(quaternions: torch.Tensor) -> torch.Tensor: function _sqrt_positive_part (line 82) | def _sqrt_positive_part(x: torch.Tensor) -> torch.Tensor: function matrix_to_quaternion (line 93) | def matrix_to_quaternion(matrix: torch.Tensor) -> torch.Tensor: FILE: projects/uncleaned_train/motionrep/utils/camera_utils.py function normalize (line 4) | def normalize(x: np.ndarray) -> np.ndarray: function viewmatrix (line 9) | def viewmatrix(lookdir: np.ndarray, up: np.ndarray, position: np.ndarray... function generate_spiral_path (line 18) | def generate_spiral_path( FILE: projects/uncleaned_train/motionrep/utils/colmap_utils.py function qvec2rotmat (line 43) | def qvec2rotmat(qvec): function rotmat2qvec (line 55) | def rotmat2qvec(R): class Image (line 68) | class Image(BaseImage): method qvec2rotmat (line 69) | def qvec2rotmat(self): function read_next_bytes (line 72) | def read_next_bytes(fid, num_bytes, format_char_sequence, endian_charact... function read_points3D_text (line 83) | def read_points3D_text(path): function read_points3D_binary (line 125) | def read_points3D_binary(path_to_model_file): function read_intrinsics_text (line 156) | def read_intrinsics_text(path): function read_extrinsics_binary (line 180) | def read_extrinsics_binary(path_to_model_file): function read_intrinsics_binary (line 215) | def read_intrinsics_binary(path_to_model_file): function read_extrinsics_text (line 244) | def read_extrinsics_text(path): function read_colmap_bin_array (line 273) | def read_colmap_bin_array(path): FILE: projects/uncleaned_train/motionrep/utils/config.py function load_config_with_merge (line 4) | def load_config_with_merge(config_path: str): function merge_without_none (line 17) | def merge_without_none(base_cfg, override_cfg): function create_config (line 26) | def create_config(config_path, args, cli_args: list = []): FILE: projects/uncleaned_train/motionrep/utils/dct.py function dct1_rfft_impl (line 12) | def dct1_rfft_impl(x): function dct_fft_impl (line 16) | def dct_fft_impl(v): function idct_irfft_impl (line 20) | def idct_irfft_impl(V): function dct (line 24) | def dct(x, norm=None): function idct (line 63) | def idct(X, norm=None): function dct_3d (line 110) | def dct_3d(x, norm=None): function idct_3d (line 127) | def idct_3d(X, norm=None): function code_test_dct3d (line 146) | def code_test_dct3d(): function unwarp_phase (line 164) | def unwarp_phase(phase, frequency_array): function get_mag_phase (line 177) | def get_mag_phase(fft_weights, s=3.0 / 16.0): function get_fft_from_mag_phase (line 214) | def get_fft_from_mag_phase(mag_phase, s=3.0 / 16.0): function get_displacements_from_fft_coeffs (line 245) | def get_displacements_from_fft_coeffs(fft_coe, t, s=3.0 / 16.0): function bandpass_filter (line 278) | def bandpass_filter(signal: torch.Tensor, low_cutoff, high_cutoff, fs: i... function bandpass_filter_numpy (line 299) | def bandpass_filter_numpy(signal: np.ndarray, low_cutoff, high_cutoff, fs): FILE: projects/uncleaned_train/motionrep/utils/flow_utils.py function flow_to_image (line 4) | def flow_to_image(flow, display=False): function make_color_wheel (line 48) | def make_color_wheel(): function compute_color (line 98) | def compute_color(u, v): FILE: projects/uncleaned_train/motionrep/utils/img_utils.py function make_grid (line 10) | def make_grid(imgs: torch.Tensor, scale=0.5): function compute_psnr (line 37) | def compute_psnr(img1, img2, mask=None): function torch_rgb_to_gray (line 64) | def torch_rgb_to_gray(image): function compute_gradient_loss (line 76) | def compute_gradient_loss(pred, gt, mask=None): function mark_image_with_red_squares (line 144) | def mark_image_with_red_squares(img): function gaussian (line 157) | def gaussian(window_size, sigma): function create_window (line 168) | def create_window(window_size, channel): function compute_ssim (line 177) | def compute_ssim(img1, img2, window_size=11, size_average=True): function _ssim (line 188) | def _ssim(img1, img2, window, window_size, channel, size_average=True): function compute_low_res_psnr (line 223) | def compute_low_res_psnr(img1, img2, scale_factor): function compute_low_res_mse (line 239) | def compute_low_res_mse(img1, img2, scale_factor): FILE: projects/uncleaned_train/motionrep/utils/io_utils.py function read_video_cv2 (line 9) | def read_video_cv2(video_path, rgb=True): function save_video_cv2 (line 31) | def save_video_cv2(video_path, img_list, fps): function save_video_imageio (line 47) | def save_video_imageio(video_path, img_list, fps): function save_gif_imageio (line 60) | def save_gif_imageio(video_path, img_list, fps): function save_video_mediapy (line 71) | def save_video_mediapy(video_frames, output_video_path: str = None, fps:... FILE: projects/uncleaned_train/motionrep/utils/optimizer.py function get_linear_schedule_with_warmup (line 5) | def get_linear_schedule_with_warmup( FILE: projects/uncleaned_train/motionrep/utils/peft_utils.py function save_peft_adaptor (line 7) | def save_peft_adaptor(model: peft.PeftModel, dir, save_base_model=False): function load_peft_adaptor_and_merge (line 15) | def load_peft_adaptor_and_merge(adaptor_path, base_model): function _code_test_peft_load_save (line 22) | def _code_test_peft_load_save(): FILE: projects/uncleaned_train/motionrep/utils/print_utils.py function print_if_zero_rank (line 4) | def print_if_zero_rank(s): FILE: projects/uncleaned_train/motionrep/utils/pytorch_mssim.py function gaussian (line 8) | def gaussian(window_size, sigma): function create_window (line 13) | def create_window(window_size, channel=1): function create_window_3d (line 19) | def create_window_3d(window_size, channel=1): function ssim (line 27) | def ssim(img1, img2, window_size=11, window=None, size_average=True, ful... function ssim_matlab (line 81) | def ssim_matlab(img1, img2, window_size=11, window=None, size_average=Tr... function msssim (line 141) | def msssim(img1, img2, window_size=11, size_average=True, val_range=None... class SSIM (line 171) | class SSIM(torch.nn.Module): method __init__ (line 172) | def __init__(self, window_size=11, size_average=True, val_range=None): method forward (line 182) | def forward(self, img1, img2): class MSSSIM (line 196) | class MSSSIM(torch.nn.Module): method __init__ (line 197) | def __init__(self, window_size=11, size_average=True, channel=3): method forward (line 203) | def forward(self, img1, img2): FILE: projects/uncleaned_train/motionrep/utils/svd_helpper.py function init_st (line 18) | def init_st(version_dict, load_ckpt=True, load_filter=True): function load_model_from_config (line 38) | def load_model_from_config(config, ckpt=None, verbose=True): function load_model (line 72) | def load_model(model): function set_lowvram_mode (line 79) | def set_lowvram_mode(mode): function initial_model_load (line 84) | def initial_model_load(model): function unload_model (line 93) | def unload_model(model): function get_unique_embedder_keys_from_conditioner (line 100) | def get_unique_embedder_keys_from_conditioner(conditioner): function get_batch (line 104) | def get_batch(keys, value_dict, N, T, device): FILE: projects/uncleaned_train/motionrep/utils/torch_utils.py function get_sync_time (line 5) | def get_sync_time(): FILE: projects/uncleaned_train/thirdparty_code/warp_mpm/backup/convert_gaussian_to_mesh.py function convert_gaussian_to_mesh (line 9) | def convert_gaussian_to_mesh(gaussian_path, save_path=None): function internal_filling (line 33) | def internal_filling(gaussian_path, save_path=None, resolution=64): FILE: projects/uncleaned_train/thirdparty_code/warp_mpm/backup/diff_warp_utils.py class MPMStateStruct (line 10) | class MPMStateStruct(object): method init (line 40) | def init( method init_grid (line 104) | def init_grid( method from_torch (line 126) | def from_torch( method reset_state (line 182) | def reset_state( method set_require_grad (line 250) | def set_require_grad(self, requires_grad=True): class ParticleStateStruct (line 262) | class ParticleStateStruct(object): method init (line 280) | def init( method from_torch (line 322) | def from_torch( method set_require_grad (line 392) | def set_require_grad(self, requires_grad=True): class MPMModelStruct (line 401) | class MPMModelStruct(object): method init (line 434) | def init( method finalize_mu_lam (line 458) | def finalize_mu_lam(self, n_particles, device="cuda:0"): method init_other_params (line 466) | def init_other_params(self, n_grid=100, grid_lim=1.0, device="cuda:0"): method from_torch (line 496) | def from_torch( method set_require_grad (line 504) | def set_require_grad(self, requires_grad=True): class Dirichlet_collider (line 513) | class Dirichlet_collider: class Impulse_modifier (line 547) | class Impulse_modifier: class MPMtailoredStruct (line 563) | class MPMtailoredStruct: class MaterialParamsModifier (line 589) | class MaterialParamsModifier: class ParticleVelocityModifier (line 598) | class ParticleVelocityModifier: function compute_mu_lam_from_E_nu_clean (line 620) | def compute_mu_lam_from_E_nu_clean( function set_vec3_to_zero (line 632) | def set_vec3_to_zero(target_array: wp.array(dtype=wp.vec3)): function set_mat33_to_identity (line 638) | def set_mat33_to_identity(target_array: wp.array(dtype=wp.mat33)): function set_mat33_to_zero (line 644) | def set_mat33_to_zero(target_array: wp.array(dtype=wp.mat33)): function add_identity_to_mat33 (line 650) | def add_identity_to_mat33(target_array: wp.array(dtype=wp.mat33)): function subtract_identity_to_mat33 (line 658) | def subtract_identity_to_mat33(target_array: wp.array(dtype=wp.mat33)): function add_vec3_to_vec3 (line 666) | def add_vec3_to_vec3( function set_value_to_float_array (line 674) | def set_value_to_float_array(target_array: wp.array(dtype=float), value:... function set_warpvalue_to_float_array (line 680) | def set_warpvalue_to_float_array( function get_float_array_product (line 688) | def get_float_array_product( function torch2warp_quat (line 697) | def torch2warp_quat(t, copy=False, dtype=warp.types.float32, dvc="cuda:0"): function torch2warp_float (line 718) | def torch2warp_float(t, copy=False, dtype=warp.types.float32, dvc="cuda:... function torch2warp_vec3 (line 738) | def torch2warp_vec3(t, copy=False, dtype=warp.types.float32, dvc="cuda:0"): function torch2warp_mat33 (line 759) | def torch2warp_mat33(t, copy=False, dtype=warp.types.float32, dvc="cuda:... FILE: projects/uncleaned_train/thirdparty_code/warp_mpm/backup/engine_utils.py function save_data_at_frame (line 8) | def save_data_at_frame(mpm_solver, dir_name, frame, save_to_ply = True, ... function particle_position_to_ply (line 38) | def particle_position_to_ply(mpm_solver, filename): function particle_position_tensor_to_ply (line 58) | def particle_position_tensor_to_ply(position_tensor, filename): FILE: projects/uncleaned_train/thirdparty_code/warp_mpm/backup/grad_test.py function test (line 16) | def test(input_dir, output_dir=None, fps=6, device=0): function position_loss_kernel (line 183) | def position_loss_kernel(mpm_state: MPMStateStruct, loss: wp.array(dtype... function g2p_test (line 192) | def g2p_test(state: MPMStateStruct, model: MPMModelStruct, dt: float): FILE: projects/uncleaned_train/thirdparty_code/warp_mpm/backup/mpm_solver_warp.py class MPM_Simulator_WARP (line 12) | class MPM_Simulator_WARP: method __init__ (line 13) | def __init__(self, n_particles, n_grid=100, grid_lim=1.0, device="cuda... method initialize (line 17) | def initialize(self, n_particles, n_grid=100, grid_lim=1.0, device="cu... method load_from_sampling (line 145) | def load_from_sampling( method load_initial_data_from_torch (line 196) | def load_initial_data_from_torch( method set_parameters (line 251) | def set_parameters(self, device="cuda:0", **kwargs): method set_parameters_dict (line 254) | def set_parameters_dict(self, kwargs={}, device="cuda:0"): method finalize_mu_lam (line 389) | def finalize_mu_lam(self, device="cuda:0"): method p2g2p (line 397) | def p2g2p(self, step, dt, device="cuda:0"): method reset_densities_and_update_masses (line 520) | def reset_densities_and_update_masses( method import_particle_x_from_torch (line 539) | def import_particle_x_from_torch(self, tensor_x, clone=True, device="c... method import_particle_v_from_torch (line 546) | def import_particle_v_from_torch(self, tensor_v, clone=True, device="c... method import_particle_F_from_torch (line 553) | def import_particle_F_from_torch(self, tensor_F, clone=True, device="c... method import_particle_C_from_torch (line 561) | def import_particle_C_from_torch(self, tensor_C, clone=True, device="c... method export_particle_x_to_torch (line 568) | def export_particle_x_to_torch(self): method export_particle_v_to_torch (line 571) | def export_particle_v_to_torch(self): method export_particle_F_to_torch (line 574) | def export_particle_F_to_torch(self): method export_particle_R_to_torch (line 579) | def export_particle_R_to_torch(self, device="cuda:0"): method export_particle_C_to_torch (line 597) | def export_particle_C_to_torch(self): method export_particle_cov_to_torch (line 602) | def export_particle_cov_to_torch(self, device="cuda:0"): method print_time_profile (line 620) | def print_time_profile(self): method add_surface_collider (line 626) | def add_surface_collider( method set_velocity_on_cuboid (line 730) | def set_velocity_on_cuboid( method add_bounding_box (line 787) | def add_bounding_box(self, start_time=0.0, end_time=999.0): method add_impulse_on_particles (line 859) | def add_impulse_on_particles( method enforce_particle_velocity_translation (line 908) | def enforce_particle_velocity_translation( method enforce_particle_velocity_rotation (line 956) | def enforce_particle_velocity_rotation( method release_particles_sequentially (line 1060) | def release_particles_sequentially( FILE: projects/uncleaned_train/thirdparty_code/warp_mpm/backup/mpm_solver_warp_diff.py class MPM_Simulator_WARPDiff (line 13) | class MPM_Simulator_WARPDiff(object): method __init__ (line 18) | def __init__(self, n_particles, n_grid=100, grid_lim=1.0, device="cuda... method initialize (line 22) | def initialize(self, n_particles, n_grid=100, grid_lim=1.0, device="cu... method set_parameters (line 39) | def set_parameters(self, device="cuda:0", **kwargs): method set_parameters_dict (line 42) | def set_parameters_dict(self, mpm_model, mpm_state, kwargs={}, device=... method set_E_nu (line 108) | def set_E_nu(self, mpm_model, E: float, nu: float, device="cuda:0"): method p2g2p (line 123) | def p2g2p(self, mpm_model, mpm_state, step, dt, device="cuda:0"): method print_time_profile (line 254) | def print_time_profile(self): method add_surface_collider (line 260) | def add_surface_collider( method set_velocity_on_cuboid (line 364) | def set_velocity_on_cuboid( method add_bounding_box (line 421) | def add_bounding_box(self, start_time=0.0, end_time=999.0): method add_impulse_on_particles (line 493) | def add_impulse_on_particles( method enforce_particle_velocity_translation (line 543) | def enforce_particle_velocity_translation( method enforce_particle_velocity_rotation (line 591) | def enforce_particle_velocity_rotation( method release_particles_sequentially (line 696) | def release_particles_sequentially( method enforce_particle_velocity_by_mask (line 723) | def enforce_particle_velocity_by_mask( FILE: projects/uncleaned_train/thirdparty_code/warp_mpm/backup/mpm_utils.py function kirchoff_stress_FCR (line 9) | def kirchoff_stress_FCR( function kirchoff_stress_neoHookean (line 19) | def kirchoff_stress_neoHookean( function kirchoff_stress_StVK (line 41) | def kirchoff_stress_StVK( function kirchoff_stress_drucker_prager (line 60) | def kirchoff_stress_drucker_prager( function von_mises_return_mapping (line 78) | def von_mises_return_mapping(F_trial: wp.mat33, model: MPMModelStruct, p... function von_mises_return_mapping_with_damage (line 124) | def von_mises_return_mapping_with_damage( function viscoplasticity_return_mapping_with_StVK (line 181) | def viscoplasticity_return_mapping_with_StVK( function sand_return_mapping (line 228) | def sand_return_mapping( function compute_mu_lam_from_E_nu (line 268) | def compute_mu_lam_from_E_nu(state: MPMStateStruct, model: MPMModelStruct): function zero_grid (line 277) | def zero_grid(state: MPMStateStruct, model: MPMModelStruct): function compute_dweight (line 285) | def compute_dweight( function update_cov (line 297) | def update_cov(state: MPMStateStruct, p: int, grad_v: wp.mat33, dt: float): function p2g_apic_with_stress (line 320) | def p2g_apic_with_stress(state: MPMStateStruct, model: MPMModelStruct, d... function grid_normalization_and_gravity (line 389) | def grid_normalization_and_gravity( function g2p (line 403) | def g2p(state: MPMStateStruct, model: MPMModelStruct, dt: float): function compute_stress_from_F_trial (line 457) | def compute_stress_from_F_trial( function compute_cov_from_F (line 512) | def compute_cov_from_F(state: MPMStateStruct, model: MPMModelStruct): function compute_R_from_F (line 539) | def compute_R_from_F(state: MPMStateStruct, model: MPMModelStruct): function add_damping_via_grid (line 566) | def add_damping_via_grid(state: MPMStateStruct, scale: float): function apply_additional_params (line 574) | def apply_additional_params( function selection_add_impulse_on_particles (line 595) | def selection_add_impulse_on_particles( function selection_enforce_particle_velocity_translation (line 611) | def selection_enforce_particle_velocity_translation( function selection_enforce_particle_velocity_cylinder (line 627) | def selection_enforce_particle_velocity_cylinder( FILE: projects/uncleaned_train/thirdparty_code/warp_mpm/backup/run_gaussian.py function load_gaussians (line 12) | def load_gaussians(input_dir: str = None): function init_volume (line 42) | def init_volume(xyz, grid=[-1, 1], num_grid=20): function run_mpm_gaussian (line 46) | def run_mpm_gaussian(input_dir, output_dir=None, fps=6, device=0): function code_test (line 165) | def code_test(input_dir, device=0): function render_gaussians (line 177) | def render_gaussians( FILE: projects/uncleaned_train/thirdparty_code/warp_mpm/backup/run_gaussian_static.py function load_gaussians (line 13) | def load_gaussians(input_dir: str = None): function get_volume (line 68) | def get_volume(xyzs: np.ndarray, resolution=128) -> np.ndarray: function run_mpm_gaussian (line 99) | def run_mpm_gaussian(input_dir, output_dir=None, fps=6, device=0): function code_test (line 260) | def code_test(input_dir, device=0): function render_gaussians (line 272) | def render_gaussians( FILE: projects/uncleaned_train/thirdparty_code/warp_mpm/backup/sim_grad.py class MyTape (line 16) | class MyTape(wp.Tape): method get_adjoint (line 19) | def get_adjoint(self, a): function test (line 56) | def test(input_dir, output_dir=None, fps=6, device=0): function g2p_test (line 171) | def g2p_test(state: MPMStateStruct, model: MPMModelStruct, dt: float): function position_loss_kernel (line 224) | def position_loss_kernel(mpm_state: MPMStateStruct, loss: wp.array(dtype... function position_loss_kernel_raw (line 233) | def position_loss_kernel_raw(particle_x: wp.array(dtype=wp.vec3), loss: ... FILE: projects/uncleaned_train/thirdparty_code/warp_mpm/backup/solver_grad_test.py function test (line 17) | def test(input_dir, output_dir=None, fps=6, device=0): function position_loss_kernel (line 152) | def position_loss_kernel(mpm_state: MPMStateStruct, loss: wp.array(dtype... FILE: projects/uncleaned_train/thirdparty_code/warp_mpm/backup/test_inverse_sim.py function test (line 18) | def test( function position_loss_kernel (line 211) | def position_loss_kernel( function step_kernel (line 228) | def step_kernel(x: wp.array(dtype=float), grad: wp.array(dtype=float), a... function aggregate_grad (line 236) | def aggregate_grad(x: wp.array(dtype=float), grad: wp.array(dtype=float)): FILE: projects/uncleaned_train/thirdparty_code/warp_mpm/backup/test_sim.py class MPMStateStruct (line 6) | class MPMStateStruct: class MPM_Simulator_WARPDiff (line 18) | class MPM_Simulator_WARPDiff: method __init__ (line 19) | def __init__(self, x, v, vol, device): function vec3_add (line 30) | def vec3_add(mpm_state: MPMStateStruct, selection: wp.array(dtype=wp.flo... function loss_kernel (line 47) | def loss_kernel(mpm_state: MPMStateStruct, loss: wp.array(dtype=float)): function main (line 56) | def main(): FILE: projects/uncleaned_train/thirdparty_code/warp_mpm/backup/warp_rewrite.py function from_torch_safe (line 13) | def from_torch_safe(t, dtype=None, requires_grad=None, grad=None): class MyTape (line 93) | class MyTape(wp.Tape): method get_adjoint (line 95) | def get_adjoint(self, a): FILE: projects/uncleaned_train/thirdparty_code/warp_mpm/backup/warp_utils.py class MPMModelStruct (line 7) | class MPMModelStruct: class MPMStateStruct (line 42) | class MPMStateStruct: class Dirichlet_collider (line 73) | class Dirichlet_collider: class Impulse_modifier (line 108) | class Impulse_modifier: class MPMtailoredStruct (line 124) | class MPMtailoredStruct: class MaterialParamsModifier (line 149) | class MaterialParamsModifier: class ParticleVelocityModifier (line 157) | class ParticleVelocityModifier: function set_vec3_to_zero (line 181) | def set_vec3_to_zero(target_array: wp.array(dtype=wp.vec3)): function set_mat33_to_identity (line 187) | def set_mat33_to_identity(target_array: wp.array(dtype=wp.mat33)): function add_identity_to_mat33 (line 193) | def add_identity_to_mat33(target_array: wp.array(dtype=wp.mat33)): function subtract_identity_to_mat33 (line 201) | def subtract_identity_to_mat33(target_array: wp.array(dtype=wp.mat33)): function add_vec3_to_vec3 (line 209) | def add_vec3_to_vec3( function set_value_to_float_array (line 217) | def set_value_to_float_array(target_array: wp.array(dtype=float), value:... function set_warpvalue_to_float_array (line 223) | def set_warpvalue_to_float_array(target_array: wp.array(dtype=float), va... function get_float_array_product (line 229) | def get_float_array_product( function torch2warp_quat (line 238) | def torch2warp_quat(t, copy=False, dtype=warp.types.float32, dvc="cuda:0"): function torch2warp_float (line 258) | def torch2warp_float(t, copy=False, dtype=warp.types.float32, dvc="cuda:... function torch2warp_vec3 (line 277) | def torch2warp_vec3(t, copy=False, dtype=warp.types.float32, dvc="cuda:0"): function torch2warp_mat33 (line 298) | def torch2warp_mat33(t, copy=False, dtype=warp.types.float32, dvc="cuda:... FILE: projects/uncleaned_train/thirdparty_code/warp_mpm/backup_jan10/gaussian_sim_utils.py function get_volume (line 3) | def get_volume(xyzs: np.ndarray, resolution=128) -> np.ndarray: FILE: projects/uncleaned_train/thirdparty_code/warp_mpm/backup_jan10/mpm_data_structure.py class MPMStateStruct (line 13) | class MPMStateStruct(object): method init (line 43) | def init( method init_grid (line 107) | def init_grid( method from_torch (line 129) | def from_torch( method reset_state (line 185) | def reset_state( method set_require_grad (line 274) | def set_require_grad(self, requires_grad=True): method reset_density (line 284) | def reset_density(self, tensor_density: Tensor, class ParticleStateStruct (line 311) | class ParticleStateStruct(object): method init (line 329) | def init( method from_torch (line 371) | def from_torch( method set_require_grad (line 441) | def set_require_grad(self, requires_grad=True): class MPMModelStruct (line 450) | class MPMModelStruct(object): method init (line 483) | def init( method finalize_mu_lam (line 507) | def finalize_mu_lam(self, n_particles, device="cuda:0"): method init_other_params (line 515) | def init_other_params(self, n_grid=100, grid_lim=1.0, device="cuda:0"): method from_torch (line 545) | def from_torch( method set_require_grad (line 553) | def set_require_grad(self, requires_grad=True): class Dirichlet_collider (line 562) | class Dirichlet_collider: class Impulse_modifier (line 596) | class Impulse_modifier: class MPMtailoredStruct (line 612) | class MPMtailoredStruct: class MaterialParamsModifier (line 638) | class MaterialParamsModifier: class ParticleVelocityModifier (line 647) | class ParticleVelocityModifier: function compute_mu_lam_from_E_nu_clean (line 669) | def compute_mu_lam_from_E_nu_clean( function set_vec3_to_zero (line 681) | def set_vec3_to_zero(target_array: wp.array(dtype=wp.vec3)): function set_vec3_to_vec3 (line 686) | def set_vec3_to_vec3(source_array: wp.array(dtype=wp.vec3), target_array... function set_float_vec_to_vec_wmask (line 691) | def set_float_vec_to_vec_wmask(source_array: wp.array(dtype=float), targ... function set_float_vec_to_vec (line 697) | def set_float_vec_to_vec(source_array: wp.array(dtype=float), target_arr... function set_mat33_to_identity (line 704) | def set_mat33_to_identity(target_array: wp.array(dtype=wp.mat33)): function set_mat33_to_zero (line 710) | def set_mat33_to_zero(target_array: wp.array(dtype=wp.mat33)): function add_identity_to_mat33 (line 716) | def add_identity_to_mat33(target_array: wp.array(dtype=wp.mat33)): function subtract_identity_to_mat33 (line 724) | def subtract_identity_to_mat33(target_array: wp.array(dtype=wp.mat33)): function add_vec3_to_vec3 (line 732) | def add_vec3_to_vec3( function set_value_to_float_array (line 740) | def set_value_to_float_array(target_array: wp.array(dtype=float), value:... function set_warpvalue_to_float_array (line 746) | def set_warpvalue_to_float_array( function get_float_array_product (line 754) | def get_float_array_product( function torch2warp_quat (line 763) | def torch2warp_quat(t, copy=False, dtype=warp.types.float32, dvc="cuda:0"): function torch2warp_float (line 784) | def torch2warp_float(t, copy=False, dtype=warp.types.float32, dvc="cuda:... function torch2warp_vec3 (line 804) | def torch2warp_vec3(t, copy=False, dtype=warp.types.float32, dvc="cuda:0"): function torch2warp_mat33 (line 825) | def torch2warp_mat33(t, copy=False, dtype=warp.types.float32, dvc="cuda:... FILE: projects/uncleaned_train/thirdparty_code/warp_mpm/backup_jan10/mpm_solver_diff.py class MPMWARPDiff (line 12) | class MPMWARPDiff(object): method __init__ (line 17) | def __init__(self, n_particles, n_grid=100, grid_lim=1.0, device="cuda... method initialize (line 21) | def initialize(self, n_particles, n_grid=100, grid_lim=1.0, device="cu... method set_parameters (line 38) | def set_parameters(self, device="cuda:0", **kwargs): method set_parameters_dict (line 41) | def set_parameters_dict(self, mpm_model, mpm_state, kwargs={}, device=... method set_E_nu (line 107) | def set_E_nu(self, mpm_model, E: float, nu: float, device="cuda:0"): method p2g2p (line 139) | def p2g2p(self, mpm_model, mpm_state, step, dt, device="cuda:0"): method print_time_profile (line 284) | def print_time_profile(self): method add_surface_collider (line 290) | def add_surface_collider( method set_velocity_on_cuboid (line 394) | def set_velocity_on_cuboid( method add_bounding_box (line 451) | def add_bounding_box(self, start_time=0.0, end_time=999.0): method add_impulse_on_particles (line 523) | def add_impulse_on_particles( method enforce_particle_velocity_translation (line 573) | def enforce_particle_velocity_translation( method enforce_particle_velocity_rotation (line 621) | def enforce_particle_velocity_rotation( method release_particles_sequentially (line 726) | def release_particles_sequentially( method enforce_particle_velocity_by_mask (line 753) | def enforce_particle_velocity_by_mask( method restart_and_compute_F_C (line 792) | def restart_and_compute_F_C(self, mpm_model, mpm_state, target_pos, de... FILE: projects/uncleaned_train/thirdparty_code/warp_mpm/backup_jan10/mpm_utils.py function kirchoff_stress_FCR (line 9) | def kirchoff_stress_FCR( function kirchoff_stress_neoHookean (line 19) | def kirchoff_stress_neoHookean( function kirchoff_stress_StVK (line 41) | def kirchoff_stress_StVK( function kirchoff_stress_drucker_prager (line 60) | def kirchoff_stress_drucker_prager( function von_mises_return_mapping (line 78) | def von_mises_return_mapping(F_trial: wp.mat33, model: MPMModelStruct, p... function von_mises_return_mapping_with_damage (line 124) | def von_mises_return_mapping_with_damage( function viscoplasticity_return_mapping_with_StVK (line 181) | def viscoplasticity_return_mapping_with_StVK( function sand_return_mapping (line 228) | def sand_return_mapping( function compute_mu_lam_from_E_nu (line 268) | def compute_mu_lam_from_E_nu(state: MPMStateStruct, model: MPMModelStruct): function zero_grid (line 277) | def zero_grid(state: MPMStateStruct, model: MPMModelStruct): function compute_dweight (line 285) | def compute_dweight( function update_cov (line 297) | def update_cov(state: MPMStateStruct, p: int, grad_v: wp.mat33, dt: float): function p2g_apic_with_stress (line 320) | def p2g_apic_with_stress(state: MPMStateStruct, model: MPMModelStruct, d... function grid_normalization_and_gravity (line 389) | def grid_normalization_and_gravity( function g2p (line 403) | def g2p(state: MPMStateStruct, model: MPMModelStruct, dt: float): function clip_particle_x (line 464) | def clip_particle_x(state: MPMStateStruct, model: MPMModelStruct): function compute_stress_from_F_trial (line 484) | def compute_stress_from_F_trial( function compute_cov_from_F (line 539) | def compute_cov_from_F(state: MPMStateStruct, model: MPMModelStruct): function compute_R_from_F (line 566) | def compute_R_from_F(state: MPMStateStruct, model: MPMModelStruct): function add_damping_via_grid (line 593) | def add_damping_via_grid(state: MPMStateStruct, scale: float): function apply_additional_params (line 602) | def apply_additional_params( function selection_add_impulse_on_particles (line 623) | def selection_add_impulse_on_particles( function selection_enforce_particle_velocity_translation (line 639) | def selection_enforce_particle_velocity_translation( function selection_enforce_particle_velocity_cylinder (line 655) | def selection_enforce_particle_velocity_cylinder( function compute_position_l2_loss (line 675) | def compute_position_l2_loss( function aggregate_grad (line 691) | def aggregate_grad(x: wp.array(dtype=float), grad: wp.array(dtype=float)): function set_F_C_p2g (line 699) | def set_F_C_p2g(state: MPMStateStruct, model: MPMModelStruct, target_pos... function set_F_C_g2p (line 735) | def set_F_C_g2p(state: MPMStateStruct, model: MPMModelStruct): function compute_posloss_with_grad (line 785) | def compute_posloss_with_grad( FILE: projects/uncleaned_train/thirdparty_code/warp_mpm/backup_jan10/warp_utils.py function from_torch_safe (line 13) | def from_torch_safe(t, dtype=None, requires_grad=None, grad=None): class MyTape (line 93) | class MyTape(wp.Tape): method get_adjoint (line 95) | def get_adjoint(self, a): class CondTape (line 131) | class CondTape(object): method __init__ (line 132) | def __init__(self, tape: Optional[MyTape], cond: bool = True) -> None: method __enter__ (line 136) | def __enter__(self): method __exit__ (line 140) | def __exit__(self, exc_type, exc_value, traceback): FILE: projects/uncleaned_train/thirdparty_code/warp_mpm/gaussian_sim_utils.py function get_volume (line 4) | def get_volume(xyzs: np.ndarray, resolution=128) -> np.ndarray: FILE: projects/uncleaned_train/thirdparty_code/warp_mpm/mpm_data_structure.py class MPMStateStruct (line 14) | class MPMStateStruct(object): method init (line 41) | def init( method init_grid (line 97) | def init_grid( method from_torch (line 119) | def from_torch( method reset_state (line 174) | def reset_state( method continue_from_torch (line 257) | def continue_from_torch( method set_require_grad (line 294) | def set_require_grad(self, requires_grad=True): method reset_density (line 305) | def reset_density( method partial_clone (line 356) | def partial_clone(self, device="cuda:0", requires_grad=True): class MPMModelStruct (line 391) | class MPMModelStruct(object): method init (line 424) | def init( method finalize_mu_lam (line 448) | def finalize_mu_lam(self, n_particles, device="cuda:0"): method init_other_params (line 456) | def init_other_params(self, n_grid=100, grid_lim=1.0, device="cuda:0"): method from_torch (line 486) | def from_torch( method set_require_grad (line 494) | def set_require_grad(self, requires_grad=True): class Dirichlet_collider (line 503) | class Dirichlet_collider: class GridCollider (line 537) | class GridCollider: class Impulse_modifier (line 548) | class Impulse_modifier: class MPMtailoredStruct (line 564) | class MPMtailoredStruct: class MaterialParamsModifier (line 590) | class MaterialParamsModifier: class ParticleVelocityModifier (line 599) | class ParticleVelocityModifier: function compute_mu_lam_from_E_nu_clean (line 621) | def compute_mu_lam_from_E_nu_clean( function set_vec3_to_zero (line 633) | def set_vec3_to_zero(target_array: wp.array(dtype=wp.vec3)): function set_vec3_to_vec3 (line 639) | def set_vec3_to_vec3( function set_float_vec_to_vec_wmask (line 647) | def set_float_vec_to_vec_wmask( function set_float_vec_to_vec (line 658) | def set_float_vec_to_vec( function set_mat33_to_identity (line 666) | def set_mat33_to_identity(target_array: wp.array(dtype=wp.mat33)): function set_mat33_to_zero (line 672) | def set_mat33_to_zero(target_array: wp.array(dtype=wp.mat33)): function add_identity_to_mat33 (line 678) | def add_identity_to_mat33(target_array: wp.array(dtype=wp.mat33)): function subtract_identity_to_mat33 (line 686) | def subtract_identity_to_mat33(target_array: wp.array(dtype=wp.mat33)): function add_vec3_to_vec3 (line 694) | def add_vec3_to_vec3( function set_value_to_float_array (line 702) | def set_value_to_float_array(target_array: wp.array(dtype=float), value:... function set_warpvalue_to_float_array (line 708) | def set_warpvalue_to_float_array( function get_float_array_product (line 716) | def get_float_array_product( function torch2warp_quat (line 725) | def torch2warp_quat(t, copy=False, dtype=warp.types.float32, dvc="cuda:0"): function torch2warp_float (line 746) | def torch2warp_float(t, copy=False, dtype=warp.types.float32, dvc="cuda:... function torch2warp_vec3 (line 766) | def torch2warp_vec3(t, copy=False, dtype=warp.types.float32, dvc="cuda:0"): function torch2warp_mat33 (line 787) | def torch2warp_mat33(t, copy=False, dtype=warp.types.float32, dvc="cuda:... FILE: projects/uncleaned_train/thirdparty_code/warp_mpm/mpm_solver_diff.py class MPMWARPDiff (line 13) | class MPMWARPDiff(object): method __init__ (line 18) | def __init__(self, n_particles, n_grid=100, grid_lim=1.0, device="cuda... method initialize (line 22) | def initialize(self, n_particles, n_grid=100, grid_lim=1.0, device="cu... method set_parameters (line 39) | def set_parameters(self, device="cuda:0", **kwargs): method set_parameters_dict (line 42) | def set_parameters_dict(self, mpm_model, mpm_state, kwargs={}, device=... method set_E_nu (line 110) | def set_E_nu(self, mpm_model, E: float, nu: float, device="cuda:0"): method set_E_nu_from_torch (line 141) | def set_E_nu_from_torch( method prepare_mu_lam (line 160) | def prepare_mu_lam(self, mpm_model, mpm_state, device="cuda:0"): method p2g2p_differentiable (line 169) | def p2g2p_differentiable( method p2g2p (line 297) | def p2g2p(self, mpm_model, mpm_state, step, dt, device="cuda:0"): method print_time_profile (line 435) | def print_time_profile(self): method add_surface_collider (line 441) | def add_surface_collider( method set_velocity_on_cuboid (line 545) | def set_velocity_on_cuboid( method add_bounding_box (line 602) | def add_bounding_box(self, start_time=0.0, end_time=999.0): method add_impulse_on_particles (line 674) | def add_impulse_on_particles( method enforce_particle_velocity_translation (line 724) | def enforce_particle_velocity_translation( method enforce_particle_velocity_rotation (line 772) | def enforce_particle_velocity_rotation( method release_particles_sequentially (line 877) | def release_particles_sequentially( method enforce_particle_velocity_by_mask (line 904) | def enforce_particle_velocity_by_mask( method restart_and_compute_F_C (line 945) | def restart_and_compute_F_C(self, mpm_model, mpm_state, target_pos, de... method enforce_grid_velocity_by_mask (line 995) | def enforce_grid_velocity_by_mask( method add_impulse_on_particles_with_mask (line 1025) | def add_impulse_on_particles_with_mask( FILE: projects/uncleaned_train/thirdparty_code/warp_mpm/mpm_utils.py function kirchoff_stress_FCR (line 9) | def kirchoff_stress_FCR( function kirchoff_stress_neoHookean (line 19) | def kirchoff_stress_neoHookean( function kirchoff_stress_StVK (line 51) | def kirchoff_stress_StVK( function kirchoff_stress_drucker_prager (line 70) | def kirchoff_stress_drucker_prager( function von_mises_return_mapping (line 88) | def von_mises_return_mapping(F_trial: wp.mat33, model: MPMModelStruct, p... function von_mises_return_mapping_with_damage (line 134) | def von_mises_return_mapping_with_damage( function viscoplasticity_return_mapping_with_StVK (line 191) | def viscoplasticity_return_mapping_with_StVK( function sand_return_mapping (line 238) | def sand_return_mapping( function compute_mu_lam_from_E_nu (line 278) | def compute_mu_lam_from_E_nu(state: MPMStateStruct, model: MPMModelStruct): function zero_grid (line 287) | def zero_grid(state: MPMStateStruct, model: MPMModelStruct): function compute_dweight (line 295) | def compute_dweight( function update_cov (line 307) | def update_cov(state: MPMStateStruct, p: int, grad_v: wp.mat33, dt: float): function update_cov_differentiable (line 330) | def update_cov_differentiable( function p2g_apic_with_stress (line 359) | def p2g_apic_with_stress(state: MPMStateStruct, model: MPMModelStruct, d... function grid_normalization_and_gravity (line 428) | def grid_normalization_and_gravity( function g2p (line 442) | def g2p(state: MPMStateStruct, model: MPMModelStruct, dt: float): function g2p_differentiable (line 503) | def g2p_differentiable( function clip_particle_x (line 583) | def clip_particle_x(state: MPMStateStruct, model: MPMModelStruct): function compute_stress_from_F_trial (line 604) | def compute_stress_from_F_trial( function add_damping_via_grid (line 723) | def add_damping_via_grid(state: MPMStateStruct, scale: float): function apply_additional_params (line 738) | def apply_additional_params( function selection_add_impulse_on_particles (line 759) | def selection_add_impulse_on_particles( function selection_enforce_particle_velocity_translation (line 775) | def selection_enforce_particle_velocity_translation( function selection_enforce_particle_velocity_cylinder (line 791) | def selection_enforce_particle_velocity_cylinder( function compute_position_l2_loss (line 812) | def compute_position_l2_loss( function aggregate_grad (line 829) | def aggregate_grad(x: wp.array(dtype=float), grad: wp.array(dtype=float)): function set_F_C_p2g (line 837) | def set_F_C_p2g( function set_F_C_g2p (line 874) | def set_F_C_g2p(state: MPMStateStruct, model: MPMModelStruct): function compute_posloss_with_grad (line 923) | def compute_posloss_with_grad( function compute_veloloss_with_grad (line 943) | def compute_veloloss_with_grad( function compute_Floss_with_grad (line 964) | def compute_Floss_with_grad( function compute_Closs_with_grad (line 997) | def compute_Closs_with_grad( FILE: projects/uncleaned_train/thirdparty_code/warp_mpm/warp_utils.py function from_torch_safe (line 13) | def from_torch_safe(t, dtype=None, requires_grad=None, grad=None): class MyTape (line 93) | class MyTape(wp.Tape): method get_adjoint (line 95) | def get_adjoint(self, a): class CondTape (line 131) | class CondTape(object): method __init__ (line 132) | def __init__(self, tape: Optional[MyTape], cond: bool = True) -> None: method __enter__ (line 136) | def __enter__(self): method __exit__ (line 140) | def __exit__(self, exc_type, exc_value, traceback):