Functions — github.com/MegaScenes/nvs

hub / github.com/MegaScenes/nvs / functions

Functions1,032 in github.com/MegaScenes/nvs

⨍ Functions 1,032◇ Types & classes 170

↓ 1 callersMethodp_losses

(self, x_start, t, noise=None)

ldm/models/diffusion/ddpm.py:345

↓ 1 callersMethodp_losses

(self, x_start, cond, t, original_img=None, noise=None)

ldm/models/diffusion/ddpm.py:1171

↓ 1 callersMethodp_losses

(self, x_start, t, noise=None)

ldm/models/diffusion/backup_ddpm.py:345

↓ 1 callersMethodp_losses

(self, x_start, cond, t, noise=None)

ldm/models/diffusion/backup_ddpm.py:1014

↓ 1 callersMethodp_losses

(self, x_start, t, noise=None)

ldm/models/diffusion/all_functions_ddpm.py:345

↓ 1 callersMethodp_losses

(self, x_start, cond, t, original_img=None, noise=None)

ldm/models/diffusion/all_functions_ddpm.py:1072

↓ 1 callersMethodp_mean_variance

(self, x, t, clip_denoised: bool)

ldm/models/diffusion/ddpm.py:282

↓ 1 callersMethodp_mean_variance

(self, x, c, t, clip_denoised: bool, return_codebook_ids=False, quantize_denoised=False,

ldm/models/diffusion/ddpm.py:1227

↓ 1 callersMethodp_mean_variance

(self, x, t, clip_denoised: bool)

ldm/models/diffusion/backup_ddpm.py:282

↓ 1 callersMethodp_mean_variance

(self, x, c, t, clip_denoised: bool, return_codebook_ids=False, quantize_denoised=False,

ldm/models/diffusion/backup_ddpm.py:1051

↓ 1 callersMethodp_mean_variance

(self, x, t, clip_denoised: bool)

ldm/models/diffusion/all_functions_ddpm.py:282

↓ 1 callersMethodp_mean_variance

(self, x, c, t, clip_denoised: bool, return_codebook_ids=False, quantize_denoised=False,

ldm/models/diffusion/all_functions_ddpm.py:1128

↓ 1 callersMethodp_sample

(self, x, t, clip_denoised=True, repeat_noise=False)

ldm/models/diffusion/ddpm.py:295

↓ 1 callersMethodp_sample

(self, x, t, clip_denoised=True, repeat_noise=False)

ldm/models/diffusion/backup_ddpm.py:295

↓ 1 callersMethodp_sample

(self, x, t, clip_denoised=True, repeat_noise=False)

ldm/models/diffusion/all_functions_ddpm.py:295

↓ 1 callersMethodp_sample_loop

(self, shape, return_intermediates=False)

ldm/models/diffusion/ddpm.py:304

↓ 1 callersMethodp_sample_loop

(self, cond, shape, return_intermediates=False, x_T=None, verbose=True, callback=None, t

ldm/models/diffusion/ddpm.py:1371

↓ 1 callersMethodp_sample_loop

(self, shape, return_intermediates=False)

ldm/models/diffusion/backup_ddpm.py:304

↓ 1 callersMethodp_sample_loop

(self, cond, shape, return_intermediates=False, x_T=None, verbose=True, callback=None, t

ldm/models/diffusion/backup_ddpm.py:1170

↓ 1 callersMethodp_sample_loop

(self, shape, return_intermediates=False)

ldm/models/diffusion/all_functions_ddpm.py:304

↓ 1 callersMethodp_sample_loop

(self, cond, shape, return_intermediates=False, x_T=None, verbose=True, callback=None, t

ldm/models/diffusion/all_functions_ddpm.py:1247

↓ 1 callersMethodp_sample_plms

(self, x, c, t, index, repeat_noise=False, use_original_steps=False, quantize_denoised=False,

ldm/models/diffusion/plms.py:180

↓ 1 callersFunctionparams_to_K

(params)

dataloader/data_helpers.py:109

↓ 1 callersFunctionparams_to_K

(params)

ldm/data/data_helpers.py:54

↓ 1 callersFunctionpatches_from_image

(img, p_size=512, p_overlap=64, p_max=800)

ldm/modules/image_degradation/utils_image.py:93

↓ 1 callersFunctionpath_planning

(num_frames, x, y, z, path_type="")

dataloader/util_3dphoto.py:294

↓ 1 callersMethodplms_sampling

(self, cond, shape, x_T=None, ddim_use_original_steps=False, callb

ldm/models/diffusion/plms.py:120

↓ 1 callersMethodpreprocess

(self, x_img)

ldm/guidance.py:15

↓ 1 callersMethodpreprocess

(self, x)

ldm/modules/encoders/modules.py:328

↓ 1 callersMethodpreprocess

(self, x)

ldm/modules/encoders/modules.py:384

↓ 1 callersMethodpreprocess

(self, x)

ldm/modules/encoders/modules.py:430

↓ 1 callersMethodprogressive_denoising

(self, cond, shape, verbose=True, callback=None, quantize_denoised=False, img_ca

ldm/models/diffusion/all_functions_ddpm.py:1191

↓ 1 callersMethodpt2np

(self, x)

ldm/data/laion.py:231

↓ 1 callersMethodq_mean_variance

Get the distribution q(x_t | x_0). :param x_start: the [N x C x ...] tensor of noiseless inputs. :param t: the number of diff

ldm/models/diffusion/ddpm.py:255

↓ 1 callersMethodq_mean_variance

Get the distribution q(x_t | x_0). :param x_start: the [N x C x ...] tensor of noiseless inputs. :param t: the number of diff

ldm/models/diffusion/backup_ddpm.py:255

↓ 1 callersMethodq_mean_variance

Get the distribution q(x_t | x_0). :param x_start: the [N x C x ...] tensor of noiseless inputs. :param t: the number of diff

ldm/models/diffusion/all_functions_ddpm.py:255

↓ 1 callersMethodq_sample

(self, x_start, t, noise=None)

ldm/modules/encoders/modules.py:538

↓ 1 callersFunctionquaternion_to_rotation_matrix

(qw, qx, qy, qz)

dataloader/data_helpers.py:305

↓ 1 callersFunctionrandom_crop

(lq, hq, sf=4, lq_patchsize=64)

ldm/modules/image_degradation/bsrgan_light.py:431

↓ 1 callersMethodread_batch

(self, batch_size: int)

ldm/modules/evaluate/adm_evaluator.py:457

↓ 1 callersMethodread_batches

(self, batch_size: int)

ldm/modules/evaluate/adm_evaluator.py:464

↓ 1 callersMethodregister_schedule

(self, beta_schedule="linear", timesteps=1000, linear_start=1e-4, linear_end=2e-2, c

ldm/modules/encoders/modules.py:511

↓ 1 callersMethodremaining

(self)

ldm/modules/evaluate/adm_evaluator.py:461

↓ 1 callersMethodreset_noise_accs

(self)

ldm/models/diffusion/classifier.py:202

↓ 1 callersFunctionresize_rgb

(rgb, trgt_H, trgt_W)

dataloader/util_3dphoto.py:31

↓ 1 callersFunctionrotation_x

Compute the rotation matrices for a batch of x-axis (roll) rotations.

dataloader/data_helpers.py:393

↓ 1 callersFunctionrotation_y

Compute the rotation matrices for a batch of y-axis (elevation) rotations.

dataloader/data_helpers.py:377

↓ 1 callersFunctionrotation_z

Compute the rotation matrices for a batch of z-axis (azimuth) rotations.

dataloader/data_helpers.py:361

↓ 1 callersMethodsample

(self, cond, batch_size=16, return_intermediates=False, x_T=None, verbose=True, timesteps=None,

ldm/models/diffusion/ddpm.py:1422

↓ 1 callersMethodsample

(self, cond, batch_size=16, return_intermediates=False, x_T=None, verbose=True, timesteps=None,

ldm/models/diffusion/backup_ddpm.py:1221

↓ 1 callersMethodsample

(self, cond, batch_size=16, return_intermediates=False, x_T=None, verbose=True, timesteps=None,

ldm/models/diffusion/all_functions_ddpm.py:1298

↓ 1 callersMethodschedule

(self, n, **kwargs)

ldm/lr_scheduler.py:17

↓ 1 callersMethodschedule

(self, n, **kwargs)

ldm/lr_scheduler.py:59

↓ 1 callersFunctionsetup_model

()

video_script.py:40

↓ 1 callersFunctionsetup_paths

()

video_script.py:62

↓ 1 callersFunctionsetup_scene_and_poses

()

video_script.py:70

↓ 1 callersFunctionshowmultimgs

(imgs)

evaluation/test_data/helpers.py:24

↓ 1 callersFunctionssim

(img1, img2, window_size=11, mask=None, size_average=True)

ldm/modules/evaluate/ssim.py:116

↓ 1 callersFunctionsynset2idx

(path_to_yaml="data/index_synset.yaml")

ldm/data/imagenet.py:20

↓ 1 callersMethodto_rgb

(self, x)

ldm/models/diffusion/all_functions_ddpm.py:1562

↓ 1 callersMethodtrain_dataloader

(self)

ldm/data/laion.py:211

↓ 1 callersMethodwarmup

(self)

ldm/modules/evaluate/adm_evaluator.py:156

↓ 1 callersMethodwrite_logs

(self, loss, logits, targets)

ldm/models/diffusion/classifier.py:162

↓ 1 callersFunctionzero_module

Zero out the parameters of a module and return it.

ldm/modules/attention.py:68

FunctionFOV2K

(fov, W, H, is_hfov=True)

dataloader/util_3dphoto.py:281

FunctionIR_101

Constructs a ir-101 model.

ldm/thirdp/psp/model_irse.py:59

FunctionIR_152

Constructs a ir-152 model.

ldm/thirdp/psp/model_irse.py:65

FunctionIR_50

Constructs a ir-50 model.

ldm/thirdp/psp/model_irse.py:53

FunctionIR_SE_101

Constructs a ir_se-101 model.

ldm/thirdp/psp/model_irse.py:77

FunctionIR_SE_152

Constructs a ir_se-152 model.

ldm/thirdp/psp/model_irse.py:83

FunctionIR_SE_50

Constructs a ir_se-50 model.

ldm/thirdp/psp/model_irse.py:71

Method__call__

(self, n, **kwargs)

ldm/lr_scheduler.py:32

Method__call__

(self, n, **kwargs)

ldm/lr_scheduler.py:77

Method__call__

(self, sample)

ldm/data/laion.py:239

Method__call__

(self, sample)

ldm/data/laion.py:262

Method__call__

(self, sample)

ldm/data/laion.py:280

Method__call__

(self, sample)

ldm/data/laion.py:304

Method__getitem__

(self, idx)

dataloader/paired_dataset.py:96

Method__getitem__

(self, index)

ldm/data/simple.py:163

Method__getitem__

(self, i)

ldm/data/imagenet.py:42

Method__getitem__

(self, i)

ldm/data/imagenet.py:339

ldm/data/backup_simple.py:65

Method__getitem__

(self, index)

ldm/data/backup_simple.py:155

Method__getitem__

(self, index)

ldm/data/backup_simple.py:287

Method__getitem__

(self, index)

ldm/data/backup_simple.py:379

Method__getitem__

(self, index)

ldm/data/backup_simple.py:427

Method__getitem__

(self, index)

ldm/data/backup_simple.py:514

Method__getitem__

(self, idx)

ldm/data/nerf_like.py:55

Method__getitem__

(self, idx)

ldm/data/nerf_like.py:98

Method__getitem__

(self, idx)

ldm/data/nerf_like.py:139

Method__init__

(self, target_res=256, split='train', pose_cond='warp_plus_pose')

dataloader/paired_dataset.py:22

(self, log_directory, no_recon_cond=True, only_sample_output=False, grid_input_sample=False, batch_frequency=2

ldm/logger.py:12

Method__init__

(self, warm_up_steps, lr_min, lr_max, lr_start, max_decay_steps, verbosity_interval=0)

ldm/lr_scheduler.py:8

Method__init__

(self, warm_up_steps, f_min, f_max, f_start, cycle_lengths, verbosity_interval=0)

ldm/lr_scheduler.py:41

Method__init__

AdamW that saves EMA versions of the parameters.

ldm/util.py:170

← previous next →401–500 of 1,032, ranked by callers