Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/cschenxiang/DRSformer
/ functions
Functions
336 in github.com/cschenxiang/DRSformer
⨍
Functions
336
◇
Types & classes
69
Method
get_text
(self, filepath)
basicsr/utils/file_client.py:146
Method
get_text
(self, filepath)
basicsr/utils/file_client.py:185
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:203
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
Method
init_dlib
Initialize the dlib detectors and predictors.
basicsr/utils/face_util.py:41
Function
load_gray_img
(filepath)
test.py:30
Function
load_gray_img
(filepath)
utils.py:82
Function
load_img
(filepath)
utils.py:76
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:31
Method
next
(self)
basicsr/data/prefetch_dataloader.py:74
Method
optimize_parameters
(self, current_iter)
basicsr/models/image_restoration_model.py:149
Method
pad_test
(self, window_size)
basicsr/models/image_restoration_model.py:175
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:307
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
Function
read_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
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
Method
run
(self)
basicsr/data/prefetch_dataloader.py:25
Method
save
(self, epoch, current_iter)
basicsr/models/image_restoration_model.py:319
Function
save_gray_img
(filepath, img)
test.py:33
Function
save_gray_img
(filepath, img)
utils.py:85
Function
save_img
(filepath, img)
utils.py:79
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:95
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:122
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
← previous
301–336 of 336, ranked by callers