MCPcopy Create free account

hub / github.com/caiyuanhao1998/Retinexformer / functions

Functions364 in github.com/caiyuanhao1998/Retinexformer

Method__init__
(self, opt)
basicsr/data/reds_dataset.py:51
Method__init__
(self, opt)
basicsr/data/SID_image_dataset.py:15
Method__init__
(self, opt)
basicsr/models/base_model.py:17
Method__init__
(self, opt)
basicsr/models/image_restoration_model.py:64
Method__init__
(self, optimizer, milestones, gamma=0.1, r
basicsr/models/lr_scheduler.py:20
Method__init__
(self, optimizer, total_iter, last_epoch=-1)
basicsr/models/lr_scheduler.py:82
Method__init__
(self, optimizer, periods, restart_weights=(1, ),
basicsr/models/lr_scheduler.py:156
Method__init__
(self, optimizer, periods, restart_weights=(1, ),
basicsr/models/lr_scheduler.py:203
Method__init__
(self, num_feat=64, res_scale=1, pytorch_init=False)
basicsr/models/archs/arch_util.py:80
Method__init__
(self, in_features, hidden_features=None, out_features=None, act_layer=nn.GELU, drop=0.)
basicsr/models/archs/layers.py:8
Method__init__
(self, dim, fn)
basicsr/models/archs/RetinexFormer_arch.py:61
Method__init__
( self, n_fea_middle, n_fea_in=4, n_fea_out=3)
basicsr/models/archs/RetinexFormer_arch.py:96
Method__init__
( self, dim, dim_head=64, heads=8, )
basicsr/models/archs/RetinexFormer_arch.py:126
Method__init__
(self, dim, mult=4)
basicsr/models/archs/RetinexFormer_arch.py:183
Method__init__
(self, in_dim=3, out_dim=3, dim=31, level=2, num_blocks=[2, 4, 4])
basicsr/models/archs/RetinexFormer_arch.py:234
Method__init__
(self, in_channels=3, out_channels=3, n_feat=31, level=2, num_blocks=[1, 1, 1])
basicsr/models/archs/RetinexFormer_arch.py:325
Method__init__
(self, in_channels=3, out_channels=3, n_feat=31, stage=3, num_blocks=[1,1,1])
basicsr/models/archs/RetinexFormer_arch.py:344
Method__init__
(self, dim, fn)
basicsr/models/archs/MST_Plus_Plus_arch.py:36
Method__init__
( self, dim, dim_head, heads, )
basicsr/models/archs/MST_Plus_Plus_arch.py:67
Method__init__
(self, dim, mult=4)
basicsr/models/archs/MST_Plus_Plus_arch.py:120
Method__init__
( self, dim, dim_head, heads, num_blocks, )
basicsr/models/archs/MST_Plus_Plus_arch.py:139
Method__init__
(self, in_channels=3, out_channels=3, n_feat=30, stage=1)
basicsr/models/archs/MST_Plus_Plus_arch.py:249
Method__init__
(self, loss_weight=1.0, reduction='mean')
basicsr/models/losses/losses.py:64
Method__init__
(self, loss_weight=1.0, reduction='mean', toY=False)
basicsr/models/losses/losses.py:86
Method__init__
(self, loss_weight=1.0, reduction='mean', eps=1e-3)
basicsr/models/losses/losses.py:114
Method__iter__
(self)
basicsr/data/data_sampler.py:30
Method__iter__
(self)
basicsr/data/prefetch_dataloader.py:59
Method__len__
(self)
basicsr/data/data_sampler.py:45
Method__len__
(self)
basicsr/data/paired_image_dataset.py:133
Method__len__
(self)
basicsr/data/paired_image_dataset.py:285
Method__len__
(self)
basicsr/data/paired_image_dataset.py:408
Method__len__
(self)
basicsr/data/paired_image_dataset.py:551
Method__len__
(self)
basicsr/data/paired_image_dataset.py:641
Method__len__
(self)
basicsr/data/video_test_dataset.py:146
Method__len__
(self)
basicsr/data/video_test_dataset.py:227
Method__len__
(self)
basicsr/data/video_test_dataset.py:324
Method__len__
(self)
basicsr/data/vimeo90k_dataset.py:129
Method__len__
(self)
basicsr/data/ffhq_dataset.py:64
Method__len__
(self)
basicsr/data/single_image_dataset.py:66
Method__len__
(self)
basicsr/data/SMID_image_dataset.py:128
Method__len__
(self)
basicsr/data/SDSD_image_dataset.py:127
Method__len__
(self)
basicsr/data/reds_dataset.py:236
Method__len__
(self)
basicsr/data/SID_image_dataset.py:124
Method__next__
(self)
basicsr/data/prefetch_dataloader.py:30
Method_init_weights
(self, m)
basicsr/models/archs/RetinexFormer_arch.py:278
Method_init_weights
(self, m)
basicsr/models/archs/MST_Plus_Plus_arch.py:209
Function_ssim
Calculate SSIM (structural similarity) for one channel images. It is called by func:`calculate_ssim`. Args: img1 (ndarray): Images w
basicsr/metrics/psnr_ssim.py:66
Methodadd_restored_face
(self, face)
basicsr/utils/face_util.py:170
Functionaugment
horizontal flip OR rotate (0, 90, 180, 270 degrees)
basicsr/data/util.py:231
Functionaugment_flow
horizontal flip OR rotate (0, 90, 180, 270 degrees) with flows
basicsr/data/util.py:270
Functionaugment_torch
horizontal flip OR rotate (0, 90, 180, 270 degrees)
basicsr/data/util.py:251
Functionbundle_submissions_raw
Bundles submission data for raw denoising submission_folder Folder where denoised images reside Output is written to <submission_folder
basicsr/utils/bundle_submissions.py:13
Functionbundle_submissions_srgb
Bundles submission data for sRGB denoising submission_folder Folder where denoised images reside Output is written to <submission_f
basicsr/utils/bundle_submissions.py:46
Functionbundle_submissions_srgb_v1
Bundles submission data for sRGB denoising submission_folder Folder where denoised images reside Output is written to <submission_f
basicsr/utils/bundle_submissions.py:79
Functioncalculate_fid
Numpy implementation of the Frechet Distance. The Frechet distance between two multivariate Gaussians X_1 ~ N(mu_1, C_1) and X_2 ~ N(mu_2, C_
basicsr/metrics/fid.py:57
Functioncalculate_niqe
Calculate NIQE (Natural Image Quality Evaluator) metric. Ref: Making a "Completely Blind" Image Quality Analyzer. This implementation could p
basicsr/metrics/niqe.py:158
Functioncalculate_psnr
(img1, img2, border=0)
Enhancement/utils.py:14
Functioncalculate_psnr
Calculate PSNR (Peak Signal-to-Noise Ratio). Ref: https://en.wikipedia.org/wiki/Peak_signal-to-noise_ratio Args: img1 (ndarray/tenso
basicsr/metrics/psnr_ssim.py:9
Functioncalculate_ssim
calculate SSIM the same outputs as MATLAB's img1, img2: [0, 255]
Enhancement/utils.py:42
Functioncalculate_ssim
Calculate SSIM (structural similarity). Ref: Image quality assessment: From error visibility to structural similarity The results are th
basicsr/metrics/psnr_ssim.py:225
Functioncallback
get the image data and update pbar.
basicsr/utils/lmdb_util.py:79
Functionchannel_convert
conversion among BGR, gray and y
basicsr/data/util.py:303
Methodclean_all
(self)
basicsr/utils/face_util.py:211
Functionconv
(in_channels, out_channels, kernel_size, bias=False, padding = 1, stride = 1)
basicsr/models/archs/MST_Plus_Plus_arch.py:50
Functioncreate_lmdb_for_SIDD
()
basicsr/utils/create_lmdb.py:84
Functioncreate_lmdb_for_gopro
()
basicsr/utils/create_lmdb.py:46
Functioncreate_lmdb_for_rain13k
()
basicsr/utils/create_lmdb.py:71
Functioncreate_lmdb_for_reds
()
basicsr/utils/create_lmdb.py:24
Functioncrop_border
Crop borders of images. Args: imgs (list[ndarray] | ndarray): Images with shape (h, w, c). crop_border (int): Crop border for eac
basicsr/utils/img_util.py:200
Methoddetect_faces
Args: img_path (str): Image path. upsample_num_times (int): Upsamples the image before running the fa
basicsr/utils/face_util.py:56
Functiondict_constructor
(loader, node)
basicsr/utils/options.py:23
Functiondict_representer
(dumper, data)
basicsr/utils/options.py:20
Functiondownload_file_from_google_drive
Download files from google drive. Ref: https://stackoverflow.com/questions/25010369/wget-curl-large-file-from-google-drive # noqa E501
basicsr/utils/download_util.py:8
Functionextract_inception_features
Extract inception features. Args: data_generator (generator): A data generator. inception (nn.Module): Inception model. l
basicsr/metrics/fid.py:23
Methodfeed_data
(self, data)
basicsr/models/base_model.py:24
Functionflow_warp
Warp an image or feature map with optical flow. Args: x (Tensor): Tensor with size (n, c, h, w). flow (Tensor): Tensor with size
basicsr/models/archs/arch_util.py:119
Functionflowread
Read an optical flow map. Args: flow_path (ndarray or str): Flow path. quantize (bool): whether to read quantized pair, if set to
basicsr/utils/flow_util.py:7
Functionflowwrite
Write optical flow to file. If the flow is not quantized, it will be saved as a .flo file losslessly, otherwise a jpeg image which is lossy b
basicsr/utils/flow_util.py:47
Methodforward
(self, x)
basicsr/models/archs/arch_util.py:90
Methodforward
(self, x)
basicsr/models/archs/layers.py:17
Methodforward
(self, x, *args, **kwargs)
basicsr/models/archs/RetinexFormer_arch.py:66
Methodforward
(self, x)
basicsr/models/archs/RetinexFormer_arch.py:72
Methodforward
(self, img)
basicsr/models/archs/RetinexFormer_arch.py:107
Methodforward
x_in: [b,h,w,c] # input_feature illu_fea: [b,h,w,c] # mask shift? 为什么是 b, h, w, c? return out: [b,h,w,c]
basicsr/models/archs/RetinexFormer_arch.py:147
Methodforward
x: [b,h,w,c] return out: [b,h,w,c]
basicsr/models/archs/RetinexFormer_arch.py:194
Methodforward
x: [b,c,h,w] illu_fea: [b,c,h,w] return out: [b,c,h,w]
basicsr/models/archs/RetinexFormer_arch.py:219
Methodforward
x: [b,c,h,w] x是feature, 不是image illu_fea: [b,c,h,w] return out: [b,c,h,w]
basicsr/models/archs/RetinexFormer_arch.py:287
Methodforward
(self, img)
basicsr/models/archs/RetinexFormer_arch.py:330
Methodforward
x: [b,c,h,w] return out:[b,c,h,w]
basicsr/models/archs/RetinexFormer_arch.py:353
Methodforward
(self, x, *args, **kwargs)
basicsr/models/archs/MST_Plus_Plus_arch.py:41
Methodforward
(self, x)
basicsr/models/archs/MST_Plus_Plus_arch.py:47
Methodforward
x_in: [b,h,w,c] return out: [b,h,w,c]
basicsr/models/archs/MST_Plus_Plus_arch.py:88
Methodforward
x: [b,h,w,c] return out: [b,h,w,c]
basicsr/models/archs/MST_Plus_Plus_arch.py:130
Methodforward
x: [b,c,h,w] return out: [b,c,h,w]
basicsr/models/archs/MST_Plus_Plus_arch.py:154
Methodforward
x: [b,c,h,w] return out:[b,c,h,w]
basicsr/models/archs/MST_Plus_Plus_arch.py:218
Methodforward
x: [b,c,h,w] return out:[b,c,h,w]
basicsr/models/archs/MST_Plus_Plus_arch.py:257
Methodforward
Args: pred (Tensor): of shape (N, C, H, W). Predicted tensor. target (Tensor): of shape (N, C, H, W). Ground truth te
basicsr/models/losses/losses.py:44
Methodforward
Args: pred (Tensor): of shape (N, C, H, W). Predicted tensor. target (Tensor): of shape (N, C, H, W). Ground truth te
basicsr/models/losses/losses.py:73
Methodforward
(self, pred, target)
basicsr/models/losses/losses.py:95
Methodforward
(self, x, y)
basicsr/models/losses/losses.py:118
← previousnext →201–300 of 364, ranked by callers