MCPcopy Create free account

hub / github.com/cschenxiang/DRSformer / functions

Functions336 in github.com/cschenxiang/DRSformer

Methodget_text
(self, filepath)
basicsr/utils/file_client.py:146
Methodget_text
(self, filepath)
basicsr/utils/file_client.py:185
Functionimg_rotate
Rotate image. Args: img (ndarray): Image to be rotated. angle (float): Rotation angle in degrees. Positive values mean
basicsr/data/transforms.py:203
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
Methodinit_dlib
Initialize the dlib detectors and predictors.
basicsr/utils/face_util.py:41
Functionload_gray_img
(filepath)
test.py:30
Functionload_gray_img
(filepath)
utils.py:82
Functionload_img
(filepath)
utils.py:76
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:31
Methodnext
(self)
basicsr/data/prefetch_dataloader.py:74
Methodoptimize_parameters
(self, current_iter)
basicsr/models/image_restoration_model.py:149
Methodpad_test
(self, window_size)
basicsr/models/image_restoration_model.py:175
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:307
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
Functionread_img_seq
Read a sequence of images from a given folder path. Args: path (list[str] | str): List of image paths or image folder path. requi
basicsr/data/data_util.py:12
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
Methodrun
(self)
basicsr/data/prefetch_dataloader.py:25
Methodsave
(self, epoch, current_iter)
basicsr/models/image_restoration_model.py:319
Functionsave_gray_img
(filepath, img)
test.py:33
Functionsave_gray_img
(filepath, img)
utils.py:85
Functionsave_img
(filepath, img)
utils.py:79
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:95
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:122
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
← previous301–336 of 336, ranked by callers