Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/VisionXLab/OF-Diff
/ functions
Functions
540 in github.com/VisionXLab/OF-Diff
⨍
Functions
540
◇
Types & classes
95
Function
resize_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
Function
rgb2ycbcr
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
Method
sample
(self)
ldm/modules/distributions/distributions.py:6
Method
sample
(self)
ldm/modules/distributions/distributions.py:35
Method
sample
(self, S, batch_size, shape, conditioning=None,
ldm/models/diffusion/plms.py:59
Method
sample
(self, S, batch_size, shape, conditioning=None,
ldm/models/diffusion/dpm_solver/sampler.py:27
Function
scale_module
Scale the parameters of a module and return it.
ldm/modules/diffusionmodules/util.py:186
Method
shared_step
(self, batch, **kwargs)
ldm/models/diffusion/ddpm.py:842
Function
shave
(img_in, border=0)
ldm/modules/image_degradation/utils_image.py:510
Function
single2tensor3
(img)
ldm/modules/image_degradation/utils_image.py:302
Function
single2tensor4
(img)
ldm/modules/image_degradation/utils_image.py:307
Function
single2tensor5
(img)
ldm/modules/image_degradation/utils_image.py:329
Function
single2uint
(img)
ldm/modules/image_degradation/utils_image.py:254
Function
single2uint16
(img)
ldm/modules/image_degradation/utils_image.py:264
Function
single32tensor5
(img)
ldm/modules/image_degradation/utils_image.py:333
Function
single42tensor4
(img)
ldm/modules/image_degradation/utils_image.py:337
Function
spatial_norm_thresholding
(x0, value)
ldm/models/diffusion/sampling_util.py:19
Function
split_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
Function
srmd_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
Function
srmd_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
Method
step
Performs a single optimization step. Args: closure (callable, optional): A closure that reevaluates the model and
ldm/util.py:119
Method
stochastic_encode
(self, x0, t, use_original_steps=False, noise=None)
ldm/models/diffusion/ddim.py:301
Method
stochastic_encode
(self, x0, t, use_original_steps=False, noise=None)
cldm/ddim_hacked.py:282
Function
surf
(Z, cmap='rainbow', figsize=None)
ldm/modules/image_degradation/utils_image.py:47
Function
tensor2img
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
Function
tensor2single
(img)
ldm/modules/image_degradation/utils_image.py:312
Function
tensor2single3
(img)
ldm/modules/image_degradation/utils_image.py:320
Function
tensor2uint
(img)
ldm/modules/image_degradation/utils_image.py:289
Method
training_step
(self, batch, batch_idx, optimizer_idx)
ldm/models/autoencoder.py:117
Method
training_step
(self, batch, batch_idx)
ldm/models/diffusion/ddpm.py:439
Function
uint162single
(img)
ldm/modules/image_degradation/utils_image.py:259
Function
uint2single
(img)
ldm/modules/image_degradation/utils_image.py:249
Function
uint2tensor3
(img)
ldm/modules/image_degradation/utils_image.py:282
Function
uint2tensor4
(img)
ldm/modules/image_degradation/utils_image.py:275
Function
uniform_on_device
(r1, r2, shape, device)
ldm/models/diffusion/ddpm.py:49
Function
uniq
(arr)
ldm/modules/attention.py:27
Method
validation_step
(self, batch, batch_idx)
ldm/models/autoencoder.py:138
Method
validation_step
(self, batch, batch_idx)
ldm/models/diffusion/ddpm.py:464
Function
write_depth
Write depth map to pfm and png file. Args: path (str): filepath without extension depth (array): depth
ldm/modules/midas/utils.py:170
Function
ycbcr2rgb
same as matlab ycbcr2rgb Input: uint8, [0, 255] float, [0, 1]
ldm/modules/image_degradation/utils_image.py:553
← previous
501–540 of 540, ranked by callers