Functions — github.com/ChenWu98/cycle-diffusion

hub / github.com/ChenWu98/cycle-diffusion / functions

Functions1,414 in github.com/ChenWu98/cycle-diffusion

⨍ Functions 1,414◇ Types & classes 263

↓ 4 callersMethodddpm_ddim_encoding

(self, S, batch_size, shape,

model/lib/latentdiff/ldm/models/diffusion/ddim.py:228

↓ 4 callersMethodema_scope

(self, context=None)

model/lib/stable_diffusion/ldm/models/autoencoder.py:64

↓ 4 callersFunctionget_condition

(model, class_label, bs)

model/gan_wrapper/latentdiff_wrapper.py:58

↓ 4 callersMethodget_fold_unfold

:param x: img of size (bs, c, h, w) :return: n img crops of size (n, bs, c, kernel_size[0], kernel_size[1])

model/lib/latentdiff/ldm/models/diffusion/ddpm.py:595

↓ 4 callersMethodget_fold_unfold

:param x: img of size (bs, c, h, w) :return: n img crops of size (n, bs, c, kernel_size[0], kernel_size[1])

model/lib/stable_diffusion/ldm/models/diffusion/ddpm.py:595

↓ 4 callersMethodget_input

(self, batch, k)

model/lib/latentdiff/ldm/models/autoencoder.py:124

↓ 4 callersMethodget_input

(self, batch, k)

model/lib/stable_diffusion/ldm/models/autoencoder.py:124

↓ 4 callersMethodget_last_layer

(self)

model/lib/latentdiff/ldm/models/autoencoder.py:230

↓ 4 callersMethodget_last_layer

(self)

model/lib/latentdiff/ldm/models/autoencoder.py:397

↓ 4 callersMethodget_last_layer

(self)

model/lib/stable_diffusion/ldm/models/autoencoder.py:230

↓ 4 callersMethodget_last_layer

(self)

model/lib/stable_diffusion/ldm/models/autoencoder.py:397

↓ 4 callersMethodmake_schedule

(self, ddim_num_steps, ddim_discretize="uniform", ddim_eta=0., verbose=True)

model/lib/latentdiff/ldm/models/diffusion/ddim.py:24

↓ 4 callersMethodmake_schedule

(self, ddim_num_steps, ddim_discretize="uniform", ddim_eta=0., verbose=True)

model/lib/stable_diffusion/ldm/models/diffusion/ddim.py:25

↓ 4 callersFunctionnormalization

Make a standard normalization layer. :param channels: number of input channels. :return: an nn.Module for normalization.

model/lib/ddpm_ddim/models/improved_ddpm/nn.py:93

↓ 4 callersFunctionrequires_grad

(model, flag=True)

model/model_utils.py:4

↓ 4 callersMethodsample

(self, batch_size=16, return_intermediates=False)

model/lib/latentdiff/ldm/models/diffusion/ddpm.py:264

↓ 4 callersMethodsample

(self, batch_size=16, return_intermediates=False)

model/lib/stable_diffusion/ldm/models/diffusion/ddpm.py:265

↓ 4 callersMethodsample_log

(self,cond,batch_size,ddim, ddim_steps,**kwargs)

model/lib/latentdiff/ldm/models/diffusion/ddpm.py:1221

↓ 4 callersMethodsample_log

(self,cond,batch_size,ddim, ddim_steps,**kwargs)

model/lib/stable_diffusion/ldm/models/diffusion/ddpm.py:1221

↓ 4 callersMethodsample_with_eps

(self, S, eps_list, batch_size,

model/lib/latentdiff/ldm/models/diffusion/ddim.py:168

↓ 4 callersMethodshared_step

(self, batch, t=None)

model/lib/latentdiff/ldm/models/diffusion/classifier.py:175

↓ 4 callersMethodshared_step

(self, batch, t=None)

model/lib/stable_diffusion/ldm/models/diffusion/classifier.py:175

↓ 4 callersMethodtrain

(self)

trainer/trainer.py:902

↓ 4 callersFunctionzero_module

Zero out the parameters of a module and return it.

model/lib/latentdiff/ldm/modules/diffusionmodules/util.py:175

↓ 4 callersFunctionzero_module

Zero out the parameters of a module and return it.

model/lib/stable_diffusion/ldm/modules/diffusionmodules/util.py:175

↓ 3 callersMethod__init__

(self, ddconfig, lossconfig, n_embed, embe

model/lib/latentdiff/ldm/models/autoencoder.py:15

↓ 3 callersMethod__init__

(self, unet_config, timesteps=1000, beta_schedule="linear",

model/lib/latentdiff/ldm/models/diffusion/ddpm.py:46

↓ 3 callersMethod__init__

(self, ddconfig, lossconfig, n_embed, embe

model/lib/stable_diffusion/ldm/models/autoencoder.py:15

↓ 3 callersMethod__init__

(self, unet_config, timesteps=1000, beta_schedule="linear",

model/lib/stable_diffusion/ldm/models/diffusion/ddpm.py:46

↓ 3 callersFunctionadd_blur

(img, sf=4)

model/lib/latentdiff/ldm/modules/image_degradation/bsrgan_light.py:325

↓ 3 callersFunctionadd_blur

(img, sf=4)

model/lib/stable_diffusion/ldm/modules/image_degradation/bsrgan_light.py:325

↓ 3 callersFunctioncheckpoint

Evaluate a function without caching intermediate activations, allowing for reduced memory at the expense of extra compute in the backward pas

model/lib/latentdiff/ldm/modules/diffusionmodules/util.py:103

↓ 3 callersFunctioncheckpoint

Evaluate a function without caching intermediate activations, allowing for reduced memory at the expense of extra compute in the backward pas

model/lib/stable_diffusion/ldm/modules/diffusionmodules/util.py:103

↓ 3 callersFunctioncustom_to_pil

(x)

model/lib/latentdiff/sample_diffusion.py:15

↓ 3 callersFunctiondefault

(val, d)

model/lib/latentdiff/ldm/modules/attention.py:19

↓ 3 callersFunctiondefault

(val, d)

model/lib/stable_diffusion/ldm/modules/attention.py:19

↓ 3 callersFunctionexists

(val)

model/lib/latentdiff/ldm/modules/attention.py:11

↓ 3 callersFunctionexists

(val)

model/lib/stable_diffusion/ldm/modules/attention.py:11

↓ 3 callersFunctionfspecial

python code from: https://github.com/ronaldosena/imagens-medicas-2/blob/40171a6c259edec7827a6693a93955de2bd39e76/Aulas/aula_2_-_uniform_filte

model/lib/latentdiff/ldm/modules/image_degradation/bsrgan_light.py:210

↓ 3 callersFunctionfspecial

python code from: https://github.com/ronaldosena/imagens-medicas-2/blob/40171a6c259edec7827a6693a93955de2bd39e76/Aulas/aula_2_-_uniform_filte

model/lib/latentdiff/ldm/modules/image_degradation/bsrgan.py:210

↓ 3 callersFunctionfspecial

python code from: https://github.com/ronaldosena/imagens-medicas-2/blob/40171a6c259edec7827a6693a93955de2bd39e76/Aulas/aula_2_-_uniform_filte

model/lib/stable_diffusion/ldm/modules/image_degradation/bsrgan_light.py:210

↓ 3 callersFunctionfspecial

python code from: https://github.com/ronaldosena/imagens-medicas-2/blob/40171a6c259edec7827a6693a93955de2bd39e76/Aulas/aula_2_-_uniform_filte

model/lib/stable_diffusion/ldm/modules/image_degradation/bsrgan.py:210

↓ 3 callersFunctionget_config

(cfg_name)

utils/config_utils.py:65

↓ 3 callersFunctionget_gan_wrapper

(args, target=False)

model/gan_wrapper/get_gan_wrapper.py:3

↓ 3 callersMethodget_input

(self, batch, k)

model/lib/latentdiff/ldm/models/autoencoder.py:344

↓ 3 callersMethodget_input

(self, batch, k)

model/lib/stable_diffusion/ldm/models/autoencoder.py:344

↓ 3 callersMethodget_loss

(self, pred, target, mean=True)

model/lib/latentdiff/ldm/models/diffusion/ddpm.py:275

↓ 3 callersMethodget_loss

(self, pred, target, mean=True)

model/lib/stable_diffusion/ldm/models/diffusion/ddpm.py:276

↓ 3 callersMethodget_weighting

(self, h, w, Ly, Lx, device)

model/lib/latentdiff/ldm/models/diffusion/ddpm.py:579

↓ 3 callersMethodget_weighting

(self, h, w, Ly, Lx, device)

model/lib/stable_diffusion/ldm/models/diffusion/ddpm.py:579

↓ 3 callersFunctionismap

(x)

model/lib/latentdiff/ldm/util.py:34

↓ 3 callersFunctionismap

(x)

model/lib/stable_diffusion/ldm/util.py:41

↓ 3 callersFunctionlinear

Create a linear module.

model/lib/ddpm_ddim/models/improved_ddpm/nn.py:35

↓ 3 callersFunctionlist_image_files_recursively

(data_dir)

utils/file_utils.py:17

↓ 3 callersFunctionload_model

(config, ckpt, gpu, eval_mode)

model/lib/latentdiff/sample_diffusion.py:225

↓ 3 callersFunctionload_model_from_config

(config, ckpt, verbose=False)

model/lib/latentdiff/txt2img.py:17

↓ 3 callersMethodlog_metrics

Log metrics in a specially formatted way Under distributed environment this is done only for a process with rank 0. Args:

trainer/trainer.py:603

↓ 3 callersFunctionlog_txt_as_img

(wh, xc, size=10)

model/lib/latentdiff/ldm/util.py:10

↓ 3 callersFunctionlog_txt_as_img

(wh, xc, size=10)

model/lib/stable_diffusion/ldm/util.py:17

↓ 3 callersMethodmode

(self)

model/lib/latentdiff/ldm/modules/distributions/distributions.py:20

↓ 3 callersMethodmode

(self)

model/lib/stable_diffusion/ldm/modules/distributions/distributions.py:20

↓ 3 callersFunctionnested_concat

Concat the `new_tensors` to `tensors` on the first dim and pad them on the second if needed. Works for tensors or nested list/tuples/dict of

trainer/trainer.py:64

↓ 3 callersMethodp_sample_ddim

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

model/lib/stable_diffusion/ldm/models/diffusion/ddim.py:503

↓ 3 callersMethodsample

(self, S, batch_size, shape, conditioning=None,

model/lib/stable_diffusion/ldm/models/diffusion/ddim.py:57

↓ 3 callersFunctionsave_image

(image_path, image)

evaluation/utils.py:7

↓ 3 callersMethodsave_metrics

Save metrics into a json file for that split, e.g. ``train_results.json``. Under distributed environment this is done only for a pro

trainer/trainer.py:691

↓ 3 callersMethodshared_step

(self, batch)

model/lib/latentdiff/ldm/models/diffusion/ddpm.py:333

↓ 3 callersMethodshared_step

(self, batch)

model/lib/stable_diffusion/ldm/models/diffusion/ddpm.py:334

↓ 3 callersFunctionspeed_metrics

Measure and return speed performance metrics. This function requires a time snapshot `start_time` before the operation to be measured starts

trainer/trainer.py:118

↓ 3 callersFunctionssim

(img1, img2)

model/lib/latentdiff/ldm/modules/image_degradation/utils_image.py:669

↓ 3 callersFunctionssim

(img1, img2)

model/lib/stable_diffusion/ldm/modules/image_degradation/utils_image.py:669

↓ 3 callersFunctionssim

(img1, img2)

evaluation/utils.py:35

↓ 3 callersMethodto_rgb

(self, x)

model/lib/latentdiff/ldm/models/diffusion/ddpm.py:1370

↓ 3 callersMethodto_rgb

(self, x)

model/lib/stable_diffusion/ldm/models/diffusion/ddpm.py:1370

↓ 3 callersFunctionzero_module

Zero out the parameters of a module and return it.

model/lib/ddpm_ddim/models/improved_ddpm/nn.py:68

↓ 2 callersFunctionNormalize

(in_channels)

model/lib/latentdiff/ldm/modules/attention.py:76

↓ 2 callersFunctionNormalize

(in_channels)

model/lib/stable_diffusion/ldm/modules/attention.py:76

↓ 2 callersMethod__init__

(self, batch_frequency, max_images, clamp=True, increase_log_steps=True, rescale=True, disabl

model/lib/latentdiff/main.py:290

↓ 2 callersMethod__init__

(self, batch_frequency, max_images, clamp=True, increase_log_steps=True, rescale=True, disabl

model/lib/stable_diffusion/main.py:290

↓ 2 callersMethod_compute_norms

(self, grad_scale=1.0)

model/lib/ddpm_ddim/models/improved_ddpm/fp16_util.py:216

↓ 2 callersMethod_get_denoise_row_from_list

(self, samples, desc='', force_no_decoder_quantization=False)

model/lib/latentdiff/ldm/models/diffusion/ddpm.py:523

↓ 2 callersMethod_get_denoise_row_from_list

(self, samples, desc='', force_no_decoder_quantization=False)

model/lib/stable_diffusion/ldm/models/diffusion/ddpm.py:524

↓ 2 callersMethod_get_rows_from_list

(self, samples)

model/lib/latentdiff/ldm/models/diffusion/ddpm.py:365

↓ 2 callersMethod_get_rows_from_list

(self, samples)

model/lib/stable_diffusion/ldm/models/diffusion/ddpm.py:366

↓ 2 callersMethod_prepare_inputs

(self, inputs)

trainer/trainer.py:728

↓ 2 callersMethod_truncate

(self, s)

model/lib/ddpm_ddim/models/improved_ddpm/logger.py:79

↓ 2 callersMethod_validation_step

(self, batch, batch_idx, suffix="")

model/lib/latentdiff/ldm/models/autoencoder.py:170

↓ 2 callersMethod_validation_step

(self, batch, batch_idx, suffix="")

model/lib/stable_diffusion/ldm/models/autoencoder.py:170

↓ 2 callersFunctionadd_Gaussian_noise

(img, noise_level1=2, noise_level2=25)

model/lib/latentdiff/ldm/modules/image_degradation/bsrgan_light.py:373

↓ 2 callersFunctionadd_Gaussian_noise

(img, noise_level1=2, noise_level2=25)

model/lib/stable_diffusion/ldm/modules/image_degradation/bsrgan_light.py:373

↓ 2 callersFunctionadd_Poisson_noise

(img)

model/lib/latentdiff/ldm/modules/image_degradation/bsrgan.py:404

↓ 2 callersFunctionadd_Poisson_noise

(img)

model/lib/stable_diffusion/ldm/modules/image_degradation/bsrgan.py:404

↓ 2 callersFunctionadd_resize

(img, sf=4)

model/lib/latentdiff/ldm/modules/image_degradation/bsrgan.py:339

↓ 2 callersFunctionadd_resize

(img, sf=4)

model/lib/stable_diffusion/ldm/modules/image_degradation/bsrgan.py:339

↓ 2 callersFunctionadd_speckle_noise

(img, noise_level1=2, noise_level2=25)

model/lib/latentdiff/ldm/modules/image_degradation/bsrgan.py:386

↓ 2 callersFunctionadd_speckle_noise

(img, noise_level1=2, noise_level2=25)

model/lib/stable_diffusion/ldm/modules/image_degradation/bsrgan.py:386

↓ 2 callersFunctionalways

(val)

model/lib/latentdiff/ldm/modules/x_transformer.py:64

↓ 2 callersFunctionalways

(val)

model/lib/stable_diffusion/ldm/modules/x_transformer.py:64

↓ 2 callersFunctionbicubic_degradation

Args: x: HxWxC image, [0, 1] sf: down-scale factor Return: bicubicly downsampled LR image

model/lib/latentdiff/ldm/modules/image_degradation/bsrgan_light.py:228

↓ 2 callersFunctionbicubic_degradation

Args: x: HxWxC image, [0, 1] sf: down-scale factor Return: bicubicly downsampled LR image

model/lib/latentdiff/ldm/modules/image_degradation/bsrgan.py:228

← previous next →101–200 of 1,414, ranked by callers