MCPcopy Create free account

hub / github.com/caiyuanhao1998/Retinexformer / functions

Functions364 in github.com/caiyuanhao1998/Retinexformer

Methodfree_dlib_gpu_memory
(self)
basicsr/utils/face_util.py:47
Methodget
(self, filepath)
basicsr/utils/file_client.py:14
Methodget
(self, filepath)
basicsr/utils/file_client.py:49
Methodget
(self, filepath)
basicsr/utils/file_client.py:63
Methodget
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
Methodget_current_visuals
(self)
basicsr/models/base_model.py:30
Methodget_face_landmarks_5
(self)
basicsr/utils/face_util.py:89
Methodget_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
Functionget_image_paths
get image path list support lmdb or image files
basicsr/data/util.py:52
Methodget_lr
(self)
basicsr/models/lr_scheduler.py:35
Methodget_lr
(self)
basicsr/models/lr_scheduler.py:66
Methodget_lr
(self)
basicsr/models/lr_scheduler.py:89
Methodget_lr
(self)
basicsr/models/lr_scheduler.py:172
Methodget_lr
(self)
basicsr/models/lr_scheduler.py:219
Methodget_text
(self, filepath)
basicsr/utils/file_client.py:18
Methodget_text
(self, filepath)
basicsr/utils/file_client.py:56
Methodget_text
(self, filepath)
basicsr/utils/file_client.py:69
Methodget_text
(self, filepath)
basicsr/utils/file_client.py:146
Methodget_text
(self, filepath)
basicsr/utils/file_client.py:185
Functionglob_file_list
(root)
basicsr/data/util.py:66
Functionimg_rotate
Rotate image. Args: img (ndarray): Image to be rotated. angle (float): Rotation angle in degrees. Positive values mean
basicsr/data/transforms.py:207
Functionimresize
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
Functionindex_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
Methodinit_dlib
Initialize the dlib detectors and predictors.
basicsr/utils/face_util.py:41
Functioninit_wandb_logger
We now only use wandb to sync tensorboard log.
basicsr/utils/logger.py:89
Functionlecun_normal_
(tensor)
basicsr/models/archs/RetinexFormer_arch.py:56
Functionload_gray_img
(filepath)
Enhancement/utils.py:100
Functionload_img
(filepath)
Enhancement/utils.py:92
Functionload_patched_inception_v3
(device='cuda', resize_input=True, normalize_input
basicsr/metrics/fid.py:10
Functionmake_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
Functionmaster_only
(func)
basicsr/utils/dist_util.py:75
Methodmixup
(self, target, input_)
basicsr/models/image_restoration_model.py:40
Functionmodcrop
img_in: Numpy, HWC or HW
basicsr/data/util.py:385
Functionmy_summary
(test_model, H = 256, W = 256, C = 3, N = 1)
Enhancement/utils.py:126
Methodnext
(self)
basicsr/data/prefetch_dataloader.py:74
Methodoptimize_parameters
(self, current_iter)
basicsr/models/image_restoration_model.py:171
Methodpad_test
(self, window_size)
basicsr/models/image_restoration_model.py:204
Methodpaste_faces_to_input_image
(self, save_path)
basicsr/utils/face_util.py:173
Functionpaths_from_folder
Generate paths from folder. Args: folder (str): Folder path. Returns: list[str]: Returned path list.
basicsr/data/data_util.py:309
Functionpixel_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
Functionprepare_for_ssim
(img, k)
basicsr/metrics/psnr_ssim.py:101
Functionprepare_for_ssim_rgb
(img, k)
basicsr/metrics/psnr_ssim.py:115
Methodprint_network
Print the str and parameter number of a network. Args: net (nn.Module)
basicsr/models/base_model.py:144
Functionread_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
Functionread_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
Methodreset
(self)
basicsr/data/prefetch_dataloader.py:80
Functionresize_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
Functionrgb2ycbcr
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
Functionrgb2ycbcr
same as matlab rgb2ycbcr only_y: only return Y channel Input: uint8, [0, 255] float, [0, 1]
basicsr/data/util.py:317
Methodrun
(self)
basicsr/data/prefetch_dataloader.py:25
Functionsave_gray_img
(filepath, img)
Enhancement/utils.py:104
Functionsave_img
(filepath, img)
Enhancement/utils.py:96
Functionscandir_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
Functionshift_back
(inputs, step=2)
basicsr/models/archs/RetinexFormer_arch.py:83
Functionshift_back
(inputs,step=2)
basicsr/models/archs/MST_Plus_Plus_arch.py:56
Functionvisualization
: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
Methodwarp_crop_faces
Get affine matrix, warp and cropped faces. Also get inverse affine matrix for post-processing.
basicsr/utils/face_util.py:132
Functionweighted_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
Functionworker_init_fn
(worker_id, num_workers, rank, seed)
basicsr/data/__init__.py:125
Functionwrapper
(*args, **kwargs)
basicsr/utils/dist_util.py:78
Functionwrapper
(pred, target, weight=None, reduction='mean', **kwargs)
basicsr/models/losses/loss_util.py:89
Functionycbcr2bgr
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
Functionycbcr2rgb
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
Functionycbcr2rgb
same as matlab ycbcr2rgb Input: uint8, [0, 255] float, [0, 1]
basicsr/data/util.py:365
← previous301–364 of 364, ranked by callers