Functions — github.com/LuPaoPao/RhythmGuassian

hub / github.com/LuPaoPao/RhythmGuassian / functions

Functions67 in github.com/LuPaoPao/RhythmGuassian

⨍ Functions 67 ◇ Types & classes 10

↓ 18 callersMethodwrite

(self, message, is_terminal=1, is_file=1)

models/utils.py:42

↓ 6 callersMethodopen

(self, file, mode=None)

models/utils.py:38

↓ 6 callersFunctiontime_to_str

(t, mode='min')

models/utils.py:19

↓ 2 callersMethod__init__

(self, in_channels, output_channels)

models/model.py:51

↓ 2 callersMethodflush

(self)

models/utils.py:52

↓ 2 callersFunctionfocal2fov

(focal, pixels)

models/graphics_utils.py:78

↓ 2 callersMethodgetLabel

(self, nowPath, Step_Index)

datasets/MyDataset.py:43

↓ 2 callersFunctionprepare_paths

Prepare file paths and related variables for given domain names.

train.py:39

↓ 2 callersFunctionprepare_paths

Prepare file paths and related variables for given domain names.

datasets/MMPD.py:39

↓ 2 callersFunctionprepare_paths

Prepare file paths and related variables for given domain names.

datasets/MR.py:39

↓ 2 callersFunctionprepare_paths

Prepare file paths and related variables for given domain names.

datasets/UCLA.py:39

↓ 2 callersFunctionprepare_paths

Prepare file paths and related variables for given domain names.

datasets/Phys.py:39

↓ 2 callersFunctionprepare_paths

Prepare file paths and related variables for given domain names.

datasets/VV100.py:39

↓ 2 callersMethodrender

Rasterize Gaussians into images. Args: gaussians: ``(B, N, 14)`` packed primitives — ``[xyz(3), opacity(1), scale

models/gs.py:29

↓ 1 callersFunctionMyEval

(HR_pr, HR_rel)

datasets/Eval.py:10

↓ 1 callersMethod__init__

(self)

models/MyLoss.py:25

↓ 1 callersMethoddepth2xyz

Lift a per-pixel depth map to world-space xyz. Args: Depth: ``(N, H, W)`` per-pixel depth values. intrinsics: ``(

models/model.py:165

↓ 1 callersMethodforward_gaussians

Split decoder output into Gaussian primitives and motion flow. Input ``x``: ``(B, 22, H=spatial_size, W=time_length)``. 22 channel

models/model.py:197

↓ 1 callersFunctiongetProjectionMatrix

(znear, zfar, fovX, fovY)

models/graphics_utils.py:53

↓ 1 callersFunctiongetWorld2View2

(R, t, translate=np.array([.0, .0, .0]), scale=1.0)

models/graphics_utils.py:39

↓ 1 callersFunctionload_datasets

Load datasets based on the prepared paths.

train.py:57

↓ 1 callersFunctionload_datasets

Load datasets based on the prepared paths.

datasets/MMPD.py:57

↓ 1 callersFunctionload_datasets

Load datasets based on the prepared paths.

datasets/MR.py:57

↓ 1 callersFunctionload_datasets

Load datasets based on the prepared paths.

datasets/UCLA.py:57

↓ 1 callersFunctionload_datasets

Load datasets based on the prepared paths.

datasets/Phys.py:57

↓ 1 callersFunctionload_datasets

Load datasets based on the prepared paths.

datasets/VV100.py:57

↓ 1 callersFunctionmain

()

train.py:99

↓ 1 callersFunctionmain

()

datasets/MMPD.py:82

↓ 1 callersFunctionmain

()

datasets/MR.py:82

↓ 1 callersFunctionmain

()

datasets/UCLA.py:82

↓ 1 callersFunctionmain

()

datasets/Phys.py:82

↓ 1 callersFunctionmain

()

datasets/VV100.py:82

↓ 1 callersFunctionsave_images

Save a batch of rendered images to ``output_dir`` as PNGs.

train.py:74

FunctionCrossValidation

(root_dir, fold_num=5,fold_index=0)

datasets/MyDataset.py:203

FunctionDrop_HR

(whole_max_idx, delNum=4)

models/utils.py:11

FunctionM_loss

Negative-Pearson between motion-flow modulus and ``v_n`` modulus over time. motion_flow: ``(B, N, T, 5)`` containing ``[Δh, Δw, Δs(3)]``. v_n

datasets/MyDataset.py:126

Method__init__

(self, root_dir, dataName, STMap, frames_num, args, transform = None)

datasets/MyDataset.py:18

(self, device, clip_length=256, delta=3, loss_type=1, use_wave=False)

models/MyLoss.py:47

Method__init__

(self, in_channels, out_channels, stride=2, downsample=False, residual=False)

models/model.py:15

Method__init__

(self, output_channels=22)

models/model.py:80

Method__len__

(self)

datasets/MyDataset.py:40

Functionapply_rotation

Applies a rotation to a quaternion. Parameters: q1 (Tensor): The original quaternion. q2 (Tensor): The rotation quaternion

models/graphics_utils.py:123

Functionbatch_quaternion_multiply

Multiply batches of quaternions. Args: - q1 (torch.Tensor): A tensor of shape [N, 4] representing the first batch of quaternions

models/graphics_utils.py:153

Methodforward

(self, gt_lable, pre_lable)

models/MyLoss.py:28

Methodforward

(self, wave, gt, pred=None, flag=None)

models/graphics_utils.py:75

Functiongeom_transform_points

(points, transf_matrix)

models/graphics_utils.py:23

FunctiongetIndex

(root_path, filesList, save_path, Pic_path, Step, frames_num)

datasets/MyDataset.py:213

FunctiongetWorld2View

(R, t)

models/graphics_utils.py:32

Per-dataset weighted physiological loss ``L_physio``. Different rPPG datasets provide different label quality (only HR vs HR+BVP), so the wei

models/MyLoss.py:180

Functionget_rays

Args: x: the horizontal coordinates of the pixels, shape: (num_rays,) y: the vertical coordinates of the pixels, shape: (num_r

models/graphics_utils.py:82

Functioniterate_data_loaders

Reset and get iterators for data loaders if needed.

train.py:90

Functioniterate_data_loaders

Reset and get iterators for data loaders if needed.

datasets/MMPD.py:73

Functioniterate_data_loaders

Reset and get iterators for data loaders if needed.

datasets/MR.py:73

Functioniterate_data_loaders

Reset and get iterators for data loaders if needed.

datasets/UCLA.py:73

Functioniterate_data_loaders

Reset and get iterators for data loaders if needed.

datasets/Phys.py:73

Functioniterate_data_loaders

Reset and get iterators for data loaders if needed.

datasets/VV100.py:73

Functionparam_size

Compute parameter size in MB

models/utils.py:154