MCPcopy Create free account

hub / github.com/VisionXLab/OF-Diff / functions

Functions540 in github.com/VisionXLab/OF-Diff

⨍ Functions 540◇ Types & classes 95
Methodforward
Apply the model to an input batch. :param x: an [N x C x ...] Tensor of inputs. :param timesteps: a 1-D batch of timesteps.
ldm/modules/diffusionmodules/openaimodel.py:756
Methodforward
(self, x)
ldm/modules/diffusionmodules/upscaling.py:49
Methodforward
(self, x)
ldm/modules/diffusionmodules/upscaling.py:62
Methodforward
(self, x, noise_level=None)
ldm/modules/diffusionmodules/upscaling.py:72
Methodforward
(ctx, run_function, length, *args)
ldm/modules/diffusionmodules/util.py:121
Methodforward
(self, x)
ldm/modules/diffusionmodules/util.py:218
Methodforward
(self, c_concat, c_crossattn)
ldm/modules/diffusionmodules/util.py:261
Methodforward
(self, x)
ldm/modules/diffusionmodules/model.py:68
Methodforward
(self, x)
ldm/modules/diffusionmodules/model.py:87
Methodforward
(self, x, temb)
ldm/modules/diffusionmodules/model.py:136
Methodforward
(self, x)
ldm/modules/diffusionmodules/model.py:186
Methodforward
(self, x)
ldm/modules/diffusionmodules/model.py:246
Methodforward
(self, x, context=None, mask=None)
ldm/modules/diffusionmodules/model.py:279
Methodforward
(self, x)
ldm/modules/diffusionmodules/model.py:527
Methodforward
(self, z)
ldm/modules/diffusionmodules/model.py:628
Methodforward
(self, x)
ldm/modules/diffusionmodules/model.py:687
Methodforward
(self, x)
ldm/modules/diffusionmodules/model.py:734
Methodforward
(self, x)
ldm/modules/diffusionmodules/model.py:773
Methodforward
(self, x)
ldm/modules/diffusionmodules/model.py:798
Methodforward
(self, x)
ldm/modules/diffusionmodules/model.py:815
Methodforward
(self, x)
ldm/modules/diffusionmodules/model.py:834
Methodforward
(self, x, scale_factor=1.0)
ldm/modules/diffusionmodules/model.py:856
Methodforward
(self, batch, key=None, disable_dropout=False)
ldm/modules/encoders/modules.py:33
Methodforward
(self, text)
ldm/modules/encoders/modules.py:75
Methodforward
(self, text)
ldm/modules/encoders/modules.py:117
Methodforward
(self, text)
ldm/modules/encoders/modules.py:170
Methodforward
(self, text)
ldm/modules/encoders/modules.py:210
Methodforward
(self, x)
ldm/modules/midas/api.py:157
Methodforward
(self, x)
ldm/modules/midas/midas/vit.py:23
Methodforward
(self, x)
ldm/modules/midas/midas/vit.py:38
Methodforward
(self, x)
ldm/modules/midas/midas/vit.py:51
Methodforward
Forward pass. Args: x (tensor): input data (image) Returns: tensor: depth
ldm/modules/midas/midas/midas_net_custom.py:73
Methodforward
Forward pass. Args: x (tensor): input Returns: tensor: interpolated data
ldm/modules/midas/midas/blocks.py:138
Methodforward
Forward pass. Args: x (tensor): input Returns: tensor: output
ldm/modules/midas/midas/blocks.py:177
Methodforward
Forward pass. Returns: tensor: output
ldm/modules/midas/midas/blocks.py:209
Methodforward
Forward pass. Args: x (tensor): input Returns: tensor: output
ldm/modules/midas/midas/blocks.py:263
Methodforward
Forward pass. Returns: tensor: output
ldm/modules/midas/midas/blocks.py:320
Methodforward
Forward pass. Args: x (tensor): input data (image) Returns: tensor: depth
ldm/modules/midas/midas/midas_net.py:49
Methodforward
(self, x)
ldm/modules/midas/midas/dpt_depth.py:107
Methodforward
(self, input, sample_posterior=True)
ldm/models/autoencoder.py:101
Methodforward
(self, x, *args, **kwargs)
ldm/models/autoencoder.py:225
Methodforward
(self, x, *args, **kwargs)
ldm/models/diffusion/ddpm.py:420
Methodforward
(self, x, c, *args, **kwargs)
ldm/models/diffusion/ddpm.py:847
Methodforward
(self, x, t, c_concat: list = None, c_crossattn: list = None, c_adm=None)
ldm/models/diffusion/ddpm.py:1328
Methodforward
(self, x)
cldm/fet.py:26
Methodforward
(self, x)
cldm/fet.py:43
Methodforward
(self, x)
cldm/fet.py:77
Methodforward
(self, x, timesteps=None, context=None, control=None, only_mid_control=False, **kwargs)
cldm/cldm.py:27
Methodforward
(self, x, hint, timesteps, context, **kwargs)
cldm/cldm.py:293
Functionforward_flex
(self, x)
ldm/modules/midas/midas/vit.py:117
Functionfuse_model
(m)
ldm/modules/midas/midas/midas_net_custom.py:109
Functiongen_kernel
# modified version of https://github.com/assafshocher/BlindSR_dataset_generator # Kai Zhang # min_var = 0.175 * sf # variance of the gau
ldm/modules/image_degradation/bsrgan_light.py:144
Functiongen_kernel
# modified version of https://github.com/assafshocher/BlindSR_dataset_generator # Kai Zhang # min_var = 0.175 * sf # variance of the gau
ldm/modules/image_degradation/bsrgan.py:145
Methodget_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])
ldm/models/diffusion/ddpm.py:722
Methodget_input
(self, batch, k, cond_key=None, bs=None, return_first_stage_outputs=False)
ldm/models/diffusion/ddpm.py:1659
Methodget_input
(self, batch, k, cond_key=None, bs=None, return_first_stage_outputs=False)
ldm/models/diffusion/ddpm.py:1703
Methodget_input
(self, batch, k, cond_key=None, bs=None, return_first_stage_outputs=False)
ldm/models/diffusion/ddpm.py:1768
Methodget_last_layer
(self)
ldm/modules/diffusionmodules/model.py:455
Methodget_model_output
(x, t)
ldm/models/diffusion/plms.py:184
Methodget_unconditional_conditioning
(self, bs, device="cuda")
ldm/modules/encoders/modules.py:45
Methodget_x_prev_and_pred_x0
(e_t, index)
ldm/models/diffusion/plms.py:205
Functionhack_everything
(clip_skip=0)
cldm/hack.py:23
Functionhook
(model, input, output)
ldm/modules/midas/midas/vit.py:160
Functionimresize
(img, scale, antialiasing=True)
ldm/modules/image_degradation/utils_image.py:766
Functionimresize_np
(img, scale, antialiasing=True)
ldm/modules/image_degradation/utils_image.py:839
Functionimsave
(img, img_path)
ldm/modules/image_degradation/utils_image.py:203
Functionimshow
(x, title=None, cbar=False, figsize=None)
ldm/modules/image_degradation/utils_image.py:37
Functionimwrite
(img, img_path)
ldm/modules/image_degradation/utils_image.py:209
Functioninit_
(tensor)
ldm/modules/attention.py:41
Methodkl
(self, other=None)
ldm/modules/distributions/distributions.py:39
Methodlog_images
(self, batch, only_inputs=False, log_ema=False, **kwargs)
ldm/models/autoencoder.py:175
Methodlog_images
(self, batch, N=8, n_row=4, sample=True, ddim_steps=50, ddim_eta=0., return_keys=None, quan
ldm/models/diffusion/ddpm.py:1157
Methodlog_images
(self, batch, N=8, n_row=4, sample=True, ddim_steps=200, ddim_eta=1., return_keys=None, plo
ldm/models/diffusion/ddpm.py:1401
Methodlog_images
(self, batch, N=8, n_row=4, sample=True, ddim_steps=200, ddim_eta=1., return_keys=None, qua
ldm/models/diffusion/ddpm.py:1561
Methodlog_images
(self, *args, **kwargs)
ldm/models/diffusion/ddpm.py:1685
Methodlog_images
(self, *args, **kwargs)
ldm/models/diffusion/ddpm.py:1736
Methodlog_images
(self, *args, **kwargs)
ldm/models/diffusion/ddpm.py:1802
Methodlow_vram_shift
(self, is_diffusing)
cldm/cldm.py:467
Functionmax_neg_value
(t)
ldm/modules/attention.py:37
Functionmean_flat
https://github.com/openai/guided-diffusion/blob/27c20a8fab9cb472df5d6bdd6c8d11c8f430b924/guided_diffusion/nn.py#L86 Take the mean over all no
ldm/util.py:57
Functionmkdir_and_rename
(path)
ldm/modules/image_degradation/utils_image.py:166
Functionmkdirs
(paths)
ldm/modules/image_degradation/utils_image.py:158
Functionmodcrop
(img_in, scale)
ldm/modules/image_degradation/utils_image.py:494
Functionmodcrop_np
Args: img: numpy image, WxH or WxHxC sf: scale factor Return: cropped image
ldm/modules/image_degradation/bsrgan_light.py:28
Functionmodcrop_np
Args: img: numpy image, WxH or WxHxC sf: scale factor Return: cropped image
ldm/modules/image_degradation/bsrgan.py:29
Methodmode
(self)
ldm/modules/distributions/distributions.py:9
Methodmode
(self)
ldm/modules/distributions/distributions.py:61
Functionmodel_fn
The noise predicition model function that is used for DPM-Solver.
ldm/models/diffusion/dpm_solver/dpm_solver.py:289
Methodnll
(self, sample, dims=[1,2,3])
ldm/modules/distributions/distributions.py:53
Methodnp2pt
(self, x)
ldm/data/util.py:15
Methodon_train_batch_end
(self, *args, **kwargs)
ldm/models/autoencoder.py:85
Methodon_train_batch_end
(self, *args, **kwargs)
ldm/models/diffusion/ddpm.py:472
Methodon_train_batch_end
(self, trainer, pl_module, outputs, batch, batch_idx)
cldm/logger.py:87
Methodon_train_batch_start
(self, batch, batch_idx)
ldm/models/diffusion/ddpm.py:599
Methodp_losses
(self, x_start, cond, t, noise=None)
cldm/cldm.py:479
Functionread_image
Read image and output RGB image (0-1). Args: path (str): path to file Returns: array: RGB image (0-1)
ldm/modules/midas/utils.py:97
Functionread_img
(path)
ldm/modules/image_degradation/utils_image.py:220
Functionread_pfm
Read pfm file. Args: path (str): path to file Returns: tuple: (data, scale)
ldm/modules/midas/utils.py:9
Methodregister_schedule
(self, given_betas=None, beta_schedule="linear", timesteps=1000,
ldm/models/diffusion/ddpm.py:614
Functionresize_depth
Resize depth map and bring to CPU (numpy). Args: depth (tensor): depth width (int): image width height (int): image heigh
ldm/modules/midas/utils.py:151
← previousnext →401–500 of 540, ranked by callers