MCPcopy Create free account

hub / github.com/OpenImagingLab/4DSloMo / functions

Functions291 in github.com/OpenImagingLab/4DSloMo

⨍ Functions 291◇ Types & classes 59
↓ 69 callersMethodcuda
(self)
scene/cameras.py:84
↓ 13 callersFunctionread_next_bytes
Read and unpack the next bytes from a binary file. :param fid: :param num_bytes: Sum of combination of {2, 4, 8}, e.g. 2, 6, 16, 30, etc.
scene/colmap_loader.py:72
↓ 10 callersMethodbackward
input: grad_out: (m, c, nsample) output: (n, c), None
pointops2/functions/pointops.py:68
↓ 10 callersMethodwrite
(self, x)
utils/general_utils.py:153
↓ 8 callersFunctiondo_system
(arg)
scripts/n3v2blender.py:190
↓ 6 callersMethodextract
(self, args)
arguments/__init__.py:40
↓ 5 callersFunctionfocal2fov
(focal, pixels)
utils/graphics_utils.py:97
↓ 5 callersMethodsave
(self, iteration)
scene/__init__.py:92
↓ 4 callersMethod__init__
(self)
lpipsPyTorch/modules/networks.py:89
↓ 4 callersMethodget_current_covariance_and_mean_offset
(self, scaling_modifier = 1, timestamp = 0.0)
scene/gaussian_model.py:242
↓ 4 callersMethodupdate
(self, hidden_states)
FixModel.py:357
↓ 3 callersFunctionPILtoTorch
(pil_image, resolution)
utils/general_utils.py:22
↓ 3 callersFunctionSH2RGB
(sh)
utils/sh_utils.py:228
↓ 3 callersMethod__init__
(self, parser: ArgumentParser, name : str, fill_none = False)
arguments/__init__.py:20
↓ 3 callersFunctioncrop_images
Crops images from the source folder and saves them to the output folder.
process_video.py:8
↓ 3 callersFunctionfov2focal
(fov, pixels)
utils/graphics_utils.py:94
↓ 3 callersMethodgetTestCameras
(self, scale=1.0)
scene/__init__.py:98
↓ 3 callersMethodgetTrainCameras
(self, scale=1.0)
scene/__init__.py:95
↓ 3 callersMethodset_requires_grad
(self, state: bool)
lpipsPyTorch/modules/networks.py:46
↓ 2 callersFunctionbuild_rotation
(r)
utils/general_utils.py:79
↓ 2 callersFunctionbuild_rotation_4d
(l, r)
utils/general_utils.py:113
↓ 2 callersFunctionbuild_scaling_rotation_4d
(s, l, r)
utils/general_utils.py:135
↓ 2 callersFunctioncameraList_from_camInfos
(cam_infos, resolution_scale, args)
utils/camera_utils.py:71
↓ 2 callersMethodcapture
(self)
scene/gaussian_model.py:96
↓ 2 callersFunctioncpu_deep_copy_tuple
(input_tuple)
gaussian_renderer/diff_gaussian_rasterization.py:30
↓ 2 callersFunctioncreate_videos_by_prefix
Creates videos from sequences of images, grouped by filename prefix.
process_video.py:49
↓ 2 callersMethoddensification_postfix
(self, new_xyz, new_features_dc, new_features_rest, new_opacities, new_scaling, new_rotation, new_t, new_scali
scene/gaussian_model.py:449
↓ 2 callersMethodencode_prompt
(self, prompt, positive=True)
FixModel.py:150
↓ 2 callersFunctionfetchPly
(path)
scene/dataset_readers.py:118
↓ 2 callersFunctiongetNerfppNorm
(cam_info)
scene/dataset_readers.py:56
↓ 2 callersFunctiongetWorld2View2
(R, t, translate=np.array([.0, .0, .0]), scale=1.0)
utils/graphics_utils.py:39
↓ 2 callersFunctioninverse_sigmoid
(x)
utils/general_utils.py:19
↓ 2 callersFunctionl1_loss
(network_output, gt)
utils/loss_utils.py:18
↓ 2 callersFunctionmodel_fn_wan_video
( dit: WanModel, x: torch.Tensor, timestep: torch.Tensor, context: torch.Tensor, clip_fea
FixModel.py:363
↓ 2 callersFunctionopt_n_threads
pointops2/src/cuda_utils.h:11
↓ 2 callersMethodprune_points
(self, mask)
scene/gaussian_model.py:404
↓ 2 callersFunctionpsnr
(img1, img2)
utils/image_utils.py:17
↓ 2 callersFunctionqvec2rotmat
(qvec)
scene/colmap_loader.py:43
↓ 2 callersFunctionreadCamerasFromTransforms
(path, transformsfile, white_background, extension=".png", time_duration=None, frame_ratio=1, dataloader=False
scene/dataset_readers.py:212
↓ 2 callersFunctionrender
Render the scene. Background tensor (bg_color) must be on GPU!
gaussian_renderer/__init__.py:19
↓ 2 callersMethodrestore
(self, model_args, training_args)
scene/gaussian_model.py:135
↓ 2 callersFunctionsafe_state
(silent)
utils/general_utils.py:147
↓ 2 callersFunctionssim
(img1, img2, window_size=11, size_average=True)
utils/loss_utils.py:34
↓ 2 callersFunctionstorePly
(path, xyz, rgb)
scene/dataset_readers.py:133
↓ 2 callersFunctionstrip_symmetric
(sym)
utils/general_utils.py:76
↓ 2 callersMethodtraining_setup
(self, training_args)
scene/gaussian_model.py:326
↓ 1 callersFunctionRGB2SH
(rgb)
utils/sh_utils.py:225
↓ 1 callersMethod__init__
(self, num_inference_steps, rel_l1_thresh, model_id)
FixModel.py:311
↓ 1 callersMethod_prune_optimizer
(self, mask)
scene/gaussian_model.py:386
↓ 1 callersFunction_ssim
(img1, img2, window, window_size, channel, size_average=True)
utils/loss_utils.py:44
↓ 1 callersMethodadd_densification_stats
(self, viewspace_point_tensor, update_filter, avg_t_grad=None)
scene/gaussian_model.py:574
↓ 1 callersMethodadd_densification_stats_grad
(self, viewspace_point_grad, update_filter, avg_t_grad=None)
scene/gaussian_model.py:580
↓ 1 callersFunctionarray_to_blob
(array)
scripts/n3v2blender.py:87
↓ 1 callersFunctionblob_to_array
(blob, dtype, shape=(-1,))
scripts/n3v2blender.py:93
↓ 1 callersFunctionbuild_scaling_rotation
(s, r)
utils/general_utils.py:102
↓ 1 callersFunctioncamTodatabase
(txtfile, database_path)
scripts/n3v2blender.py:130
↓ 1 callersFunctioncamera_to_JSON
(id, camera : Camera)
utils/camera_utils.py:79
↓ 1 callersMethodcat_tensors_to_optimizer
(self, tensors_dict)
scene/gaussian_model.py:427
↓ 1 callersMethodcheck
(self, dit: WanModel, x, t_mod)
FixModel.py:331
↓ 1 callersFunctionclosest_point_2_lines
(oa, da, ob, db)
scripts/n3v2blender.py:198
↓ 1 callersMethodconnect
(database_path)
scripts/n3v2blender.py:102
↓ 1 callersMethodcreate_from_pcd
(self, pcd : BasicPointCloud, spatial_lr_scale : float)
scene/gaussian_model.py:254
↓ 1 callersFunctioncreate_window
(window_size, channel)
utils/loss_utils.py:28
↓ 1 callersMethoddecode_video
(self, latents, tiled=True, tile_size=(34, 34), tile_stride=(18, 16))
FixModel.py:211
↓ 1 callersMethoddensify_and_clone
(self, grads, grad_threshold, scene_extent, grads_t, grad_t_threshold)
scene/gaussian_model.py:528
↓ 1 callersMethoddensify_and_prune
(self, max_grad, min_opacity, extent, max_screen_size, max_grad_t=None, prune_only=False)
scene/gaussian_model.py:552
↓ 1 callersMethoddensify_and_split
(self, grads, grad_threshold, scene_extent, grads_t, grad_t_threshold, N=2)
scene/gaussian_model.py:481
↓ 1 callersFunctioneasy_cmap
(x: torch.Tensor)
utils/image_utils.py:21
↓ 1 callersMethodenable_vram_management
(self, num_persistent_param_in_dit=None)
FixModel.py:38
↓ 1 callersMethodencode_image
(self, image, num_frames, height, width)
FixModel.py:155
↓ 1 callersMethodencode_image_condition
(self, image, inter_images, num_frames, height, width,cam_id)
FixModel.py:174
↓ 1 callersFunctioneval_sh
Evaluate spherical harmonics at unit directions using hardcoded SH polynomials. Works with torch/np/jnp. ... Can be 0 or more batch d
utils/sh_utils.py:58
↓ 1 callersFunctioneval_shfs_4d
Evaluate spherical harmonics at unit directions using hardcoded SH polynomials. Works with torch/np/jnp. ... Can be 0 or more batch d
utils/sh_utils.py:115
↓ 1 callersMethodfetch_models
(self, model_manager: ModelManager)
FixModel.py:126
↓ 1 callersMethodflush
(self)
utils/general_utils.py:160
↓ 1 callersMethodfrom_model_manager
(model_manager: ModelManager, torch_dtype=None, device=None)
FixModel.py:138
↓ 1 callersFunctiongaussian
(window_size, sigma)
utils/loss_utils.py:24
↓ 1 callersFunctiongetProjectionMatrix
(znear, zfar, fovX, fovY)
utils/graphics_utils.py:52
↓ 1 callersFunctiongetProjectionMatrixCenterShift
(znear, zfar, cx, cy, fl_x, fl_y, w, h)
utils/graphics_utils.py:74
↓ 1 callersFunctionget_center_and_diag
(cam_centers)
scene/dataset_readers.py:57
↓ 1 callersFunctionget_combined_args
(parser : ArgumentParser)
arguments/__init__.py:109
↓ 1 callersMethodget_cov_t
(self, scaling_modifier = 1)
scene/gaussian_model.py:227
↓ 1 callersMethodget_covariance
(self, scaling_modifier = 1)
scene/gaussian_model.py:239
↓ 1 callersFunctionget_expon_lr_func
Copied from Plenoxels Continuous learning rate decay function. Adapted from JaxNeRF The returned rate is lr_init when step=0 and lr_fina
utils/general_utils.py:30
↓ 1 callersMethodget_marginal_t
(self, timestamp, scaling_modifier = 1)
scene/gaussian_model.py:235
↓ 1 callersFunctionget_network
(net_type: str)
lpipsPyTorch/modules/networks.py:12
↓ 1 callersMethodget_rays
(self)
scene/cameras.py:75
↓ 1 callersFunctionget_state_dict
(net_type: str = 'alex', version: str = '0.1')
lpipsPyTorch/modules/utils.py:11
↓ 1 callersFunctionknn
(x, src, k, transpose=False)
utils/general_utils.py:170
↓ 1 callersFunctionloadCam
(args, id, cam_info, resolution_scale)
utils/camera_utils.py:19
↓ 1 callersFunctionmain
()
process_video.py:101
↓ 1 callersFunctionmain
()
test_lora.py:23
↓ 1 callersFunctionmsssim
(rgb, gts)
utils/loss_utils.py:68
↓ 1 callersFunctionnormalize_activation
(x, eps=1e-10)
lpipsPyTorch/modules/utils.py:6
↓ 1 callersMethodoneupSHdegree
(self)
scene/gaussian_model.py:248
↓ 1 callersFunctionplayVids
(videoId)
docs/js/video_comparison.js:5
↓ 1 callersMethodprepare_extra_input
(self, latents=None)
FixModel.py:202
↓ 1 callersFunctionprepare_output_and_logger
(args)
train.py:255
↓ 1 callersFunctionrasterize_gaussians
( means3D, means2D, sh, colors_precomp, flow_2d, opacities, ts, scales, sc
gaussian_renderer/diff_gaussian_rasterization.py:34
↓ 1 callersFunctionread
()
gaussian_renderer/network_gui.py:43
next →1–100 of 291, ranked by callers