MCPcopy Create free account

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

Functions540 in github.com/VisionXLab/OF-Diff

Functionresize_image
Resize image and make it fit for network. Args: img (array): image Returns: tensor: data ready for network
ldm/modules/midas/utils.py:116
Functionrgb2ycbcr
same as matlab rgb2ycbcr only_y: only return Y channel Input: uint8, [0, 255] float, [0, 1]
ldm/modules/image_degradation/utils_image.py:529
Methodsample
(self)
ldm/modules/distributions/distributions.py:6
Methodsample
(self)
ldm/modules/distributions/distributions.py:35
Methodsample
(self, S, batch_size, shape, conditioning=None,
ldm/models/diffusion/plms.py:59
Methodsample
(self, S, batch_size, shape, conditioning=None,
ldm/models/diffusion/dpm_solver/sampler.py:27
Functionscale_module
Scale the parameters of a module and return it.
ldm/modules/diffusionmodules/util.py:186
Methodshared_step
(self, batch, **kwargs)
ldm/models/diffusion/ddpm.py:842
Functionshave
(img_in, border=0)
ldm/modules/image_degradation/utils_image.py:510
Functionsingle2tensor3
(img)
ldm/modules/image_degradation/utils_image.py:302
Functionsingle2tensor4
(img)
ldm/modules/image_degradation/utils_image.py:307
Functionsingle2tensor5
(img)
ldm/modules/image_degradation/utils_image.py:329
Functionsingle2uint
(img)
ldm/modules/image_degradation/utils_image.py:254
Functionsingle2uint16
(img)
ldm/modules/image_degradation/utils_image.py:264
Functionsingle32tensor5
(img)
ldm/modules/image_degradation/utils_image.py:333
Functionsingle42tensor4
(img)
ldm/modules/image_degradation/utils_image.py:337
Functionspatial_norm_thresholding
(x0, value)
ldm/models/diffusion/sampling_util.py:19
Functionsplit_imageset
split the large images from original_dataroot into small overlapped images with size (p_size)x(p_size), and save them into taget_dataroot; on
ldm/modules/image_degradation/utils_image.py:125
Functionsrmd_degradation
blur + bicubic downsampling Args: x: HxWxC image, [0, 1] k: hxw, double sf: down-scale factor Return: downsam
ldm/modules/image_degradation/bsrgan_light.py:239
Functionsrmd_degradation
blur + bicubic downsampling Args: x: HxWxC image, [0, 1] k: hxw, double sf: down-scale factor Return: downsam
ldm/modules/image_degradation/bsrgan.py:240
Methodstep
Performs a single optimization step. Args: closure (callable, optional): A closure that reevaluates the model and
ldm/util.py:119
Methodstochastic_encode
(self, x0, t, use_original_steps=False, noise=None)
ldm/models/diffusion/ddim.py:301
Methodstochastic_encode
(self, x0, t, use_original_steps=False, noise=None)
cldm/ddim_hacked.py:282
Functionsurf
(Z, cmap='rainbow', figsize=None)
ldm/modules/image_degradation/utils_image.py:47
Functiontensor2img
Converts a torch Tensor into an image Numpy array of BGR channel order Input: 4D(B,(3/1),H,W), 3D(C,H,W), or 2D(H,W), any range, RGB channel
ldm/modules/image_degradation/utils_image.py:342
Functiontensor2single
(img)
ldm/modules/image_degradation/utils_image.py:312
Functiontensor2single3
(img)
ldm/modules/image_degradation/utils_image.py:320
Functiontensor2uint
(img)
ldm/modules/image_degradation/utils_image.py:289
Methodtraining_step
(self, batch, batch_idx, optimizer_idx)
ldm/models/autoencoder.py:117
Methodtraining_step
(self, batch, batch_idx)
ldm/models/diffusion/ddpm.py:439
Functionuint162single
(img)
ldm/modules/image_degradation/utils_image.py:259
Functionuint2single
(img)
ldm/modules/image_degradation/utils_image.py:249
Functionuint2tensor3
(img)
ldm/modules/image_degradation/utils_image.py:282
Functionuint2tensor4
(img)
ldm/modules/image_degradation/utils_image.py:275
Functionuniform_on_device
(r1, r2, shape, device)
ldm/models/diffusion/ddpm.py:49
Functionuniq
(arr)
ldm/modules/attention.py:27
Methodvalidation_step
(self, batch, batch_idx)
ldm/models/autoencoder.py:138
Methodvalidation_step
(self, batch, batch_idx)
ldm/models/diffusion/ddpm.py:464
Functionwrite_depth
Write depth map to pfm and png file. Args: path (str): filepath without extension depth (array): depth
ldm/modules/midas/utils.py:170
Functionycbcr2rgb
same as matlab ycbcr2rgb Input: uint8, [0, 255] float, [0, 1]
ldm/modules/image_degradation/utils_image.py:553
← previous501–540 of 540, ranked by callers