Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/caiyuanhao1998/Retinexformer
/ functions
Functions
364 in github.com/caiyuanhao1998/Retinexformer
⨍
Functions
364
◇
Types & classes
60
Method
free_dlib_gpu_memory
(self)
basicsr/utils/face_util.py:47
Method
get
(self, filepath)
basicsr/utils/file_client.py:14
Method
get
(self, filepath)
basicsr/utils/file_client.py:49
Method
get
(self, filepath)
basicsr/utils/file_client.py:63
Method
get
Get values according to the filepath from one lmdb named client_key. Args: filepath (str | obj:`Path`): Here, filepath is the lmd
basicsr/utils/file_client.py:131
Method
get_current_visuals
(self)
basicsr/models/base_model.py:30
Method
get_face_landmarks_5
(self)
basicsr/utils/face_util.py:89
Method
get_face_landmarks_68
Get 68 densemarks for cropped images. Should only have one face at most in the cropped image.
basicsr/utils/face_util.py:96
Function
get_image_paths
get image path list support lmdb or image files
basicsr/data/util.py:52
Method
get_lr
(self)
basicsr/models/lr_scheduler.py:35
Method
get_lr
(self)
basicsr/models/lr_scheduler.py:66
Method
get_lr
(self)
basicsr/models/lr_scheduler.py:89
Method
get_lr
(self)
basicsr/models/lr_scheduler.py:172
Method
get_lr
(self)
basicsr/models/lr_scheduler.py:219
Method
get_text
(self, filepath)
basicsr/utils/file_client.py:18
Method
get_text
(self, filepath)
basicsr/utils/file_client.py:56
Method
get_text
(self, filepath)
basicsr/utils/file_client.py:69
Method
get_text
(self, filepath)
basicsr/utils/file_client.py:146
Method
get_text
(self, filepath)
basicsr/utils/file_client.py:185
Function
glob_file_list
(root)
basicsr/data/util.py:66
Function
img_rotate
Rotate image. Args: img (ndarray): Image to be rotated. angle (float): Rotation angle in degrees. Positive values mean
basicsr/data/transforms.py:207
Function
imresize
imresize function same as MATLAB. It now only supports bicubic. The same scale applies for both height and width. Args: img (Ten
basicsr/utils/matlab_functions.py:88
Function
index_generation
Generate an index list for reading N frames from a sequence of images Args: crt_i (int): current center index max_n (int): max num
basicsr/data/util.py:176
Method
init_dlib
Initialize the dlib detectors and predictors.
basicsr/utils/face_util.py:41
Function
init_wandb_logger
We now only use wandb to sync tensorboard log.
basicsr/utils/logger.py:89
Function
lecun_normal_
(tensor)
basicsr/models/archs/RetinexFormer_arch.py:56
Function
load_gray_img
(filepath)
Enhancement/utils.py:100
Function
load_img
(filepath)
Enhancement/utils.py:92
Function
load_patched_inception_v3
(device='cuda', resize_input=True, normalize_input
basicsr/metrics/fid.py:10
Function
make_layer
Make layers by stacking the same blocks. Args: basic_block (nn.module): nn.module class for basic block. num_basic_block (int): n
basicsr/models/archs/arch_util.py:49
Function
master_only
(func)
basicsr/utils/dist_util.py:75
Method
mixup
(self, target, input_)
basicsr/models/image_restoration_model.py:40
Function
modcrop
img_in: Numpy, HWC or HW
basicsr/data/util.py:385
Function
my_summary
(test_model, H = 256, W = 256, C = 3, N = 1)
Enhancement/utils.py:126
Method
next
(self)
basicsr/data/prefetch_dataloader.py:74
Method
optimize_parameters
(self, current_iter)
basicsr/models/image_restoration_model.py:171
Method
pad_test
(self, window_size)
basicsr/models/image_restoration_model.py:204
Method
paste_faces_to_input_image
(self, save_path)
basicsr/utils/face_util.py:173
Function
paths_from_folder
Generate paths from folder. Args: folder (str): Folder path. Returns: list[str]: Returned path list.
basicsr/data/data_util.py:309
Function
pixel_unshuffle
Pixel unshuffle. Args: x (Tensor): Input feature with shape (b, c, hh, hw). scale (int): Downsample ratio. Returns:
basicsr/models/archs/arch_util.py:211
Function
prepare_for_ssim
(img, k)
basicsr/metrics/psnr_ssim.py:101
Function
prepare_for_ssim_rgb
(img, k)
basicsr/metrics/psnr_ssim.py:115
Method
print_network
Print the str and parameter number of a network. Args: net (nn.Module)
basicsr/models/base_model.py:144
Function
read_img_seq
Read a sequence of images from a given folder path Args: path (list/str): list of image paths/image folder path Returns: imgs
basicsr/data/util.py:124
Function
read_img_seq2
Read a sequence of images from a given folder path Args: path (list/str): list of image paths/image folder path Returns: imgs
basicsr/data/util.py:149
Method
reset
(self)
basicsr/data/prefetch_dataloader.py:80
Function
resize_flow
Resize a flow according to ratio or shape. Args: flow (Tensor): Precomputed flow. shape [N, 2, H, W]. size_type (str): 'ratio' or
basicsr/models/archs/arch_util.py:164
Function
rgb2ycbcr
Convert a RGB image to YCbCr image. This function produces the same results as Matlab's `rgb2ycbcr` function. It implements the ITU-R BT.601
basicsr/utils/matlab_functions.py:173
Function
rgb2ycbcr
same as matlab rgb2ycbcr only_y: only return Y channel Input: uint8, [0, 255] float, [0, 1]
basicsr/data/util.py:317
Method
run
(self)
basicsr/data/prefetch_dataloader.py:25
Function
save_gray_img
(filepath, img)
Enhancement/utils.py:104
Function
save_img
(filepath, img)
Enhancement/utils.py:96
Function
scandir_SIDD
Scan a directory to find the interested files. Args: dir_path (str): Path of the directory. keywords (str | tuple(str), optional)
basicsr/utils/misc.py:119
Function
shift_back
(inputs, step=2)
basicsr/models/archs/RetinexFormer_arch.py:83
Function
shift_back
(inputs,step=2)
basicsr/models/archs/MST_Plus_Plus_arch.py:56
Function
visualization
:param feature: [C,H,W] :param save_path: saving path :param type: 'mean' or 'max' :param colormap: the type of the pseudocolor map
Enhancement/utils.py:108
Method
warp_crop_faces
Get affine matrix, warp and cropped faces. Also get inverse affine matrix for post-processing.
basicsr/utils/face_util.py:132
Function
weighted_loss
Create a weighted version of a given loss function. To use this decorator, the loss function must have the signature like `loss_func(pred, ta
basicsr/models/losses/loss_util.py:57
Function
worker_init_fn
(worker_id, num_workers, rank, seed)
basicsr/data/__init__.py:125
Function
wrapper
(*args, **kwargs)
basicsr/utils/dist_util.py:78
Function
wrapper
(pred, target, weight=None, reduction='mean', **kwargs)
basicsr/models/losses/loss_util.py:89
Function
ycbcr2bgr
Convert a YCbCr image to BGR image. The bgr version of ycbcr2rgb. It implements the ITU-R BT.601 conversion for standard-definition telev
basicsr/utils/matlab_functions.py:273
Function
ycbcr2rgb
Convert a YCbCr image to RGB image. This function produces the same results as Matlab's ycbcr2rgb function. It implements the ITU-R BT.601 co
basicsr/utils/matlab_functions.py:241
Function
ycbcr2rgb
same as matlab ycbcr2rgb Input: uint8, [0, 255] float, [0, 1]
basicsr/data/util.py:365
← previous
301–364 of 364, ranked by callers