MCPcopy Create free account

hub / github.com/HPLQAQ/SAM-Deblur / functions

Functions372 in github.com/HPLQAQ/SAM-Deblur

⨍ Functions 372◇ Types & classes 60
↓ 1 callersMethodupdate_learning_rate
Update learning rate. Args: current_iter (int): Current iteration. warmup_iter (int)： Warmup iter numbers. -1 for no
basicsr/models/base_model.py:164
↓ 1 callersFunctionweight_reduce_loss
Apply element-wise weight and reduce loss. Args: loss (Tensor): Element-wise loss. weight (Tensor): Element-wise weights. Default
basicsr/models/losses/loss_util.py:31
↓ 1 callersFunctionwrite_version_py
()
setup.py:59
Method__call__
Format logging message. Args: log_vars (dict): It contains the following keys: epoch (int): Epoch number.
basicsr/utils/logger.py:38
Method__call__
(self, target, input_)
basicsr/models/image_clean_model.py:41
Method__getitem__
(self, index)
basicsr/data/paired_image_dataset.py:78
Method__getitem__
(self, index)
basicsr/data/paired_image_mask_dataset.py:79
Method__getitem__
(self, index)
basicsr/data/paired_image_group_mask_dataset.py:79
Method__init__
(self, opt, start_iter=1, tb_logger=None)
basicsr/utils/logger.py:27
Method__init__
(self, server_list_cfg, client_cfg, sys_path=None)
basicsr/utils/file_client.py:38
Method__init__
(self, db_paths, client_keys='default', readonly=True,
basicsr/utils/file_client.py:102
Method__init__
(self, backend='disk', **kwargs)
basicsr/utils/file_client.py:175
Method__init__
(self, lmdb_path, map_size=1024**4, batch=5000,
basicsr/utils/lmdb_util.py:181
Method__init__
(self, upscale_factor, face_size=512)
basicsr/utils/face_util.py:25
Method__init__
(self, dataset, num_replicas, rank, ratio=1)
basicsr/data/data_sampler.py:28
Method__init__
(self, opt)
basicsr/data/paired_image_dataset.py:48
Method__init__
(self, generator, num_prefetch_queue)
basicsr/data/prefetch_dataloader.py:24
Method__init__
(self, num_prefetch_queue, **kwargs)
basicsr/data/prefetch_dataloader.py:61
Method__init__
(self, loader, opt)
basicsr/data/prefetch_dataloader.py:103
Method__init__
(self, opt)
basicsr/data/paired_image_mask_dataset.py:49
Method__init__
(self, opt)
basicsr/data/paired_image_group_mask_dataset.py:49
Method__init__
(self, opt)
basicsr/models/image_clean_model.py:54
Method__init__
(self, opt)
basicsr/models/base_model.py:23
Method__init__
(self, opt)
basicsr/models/image_restoration_model.py:26
Method__init__
(self, optimizer, milestones, gamma=0.1, r
basicsr/models/lr_scheduler.py:25
Method__init__
(self, optimizer, total_iter, last_epoch=-1)
basicsr/models/lr_scheduler.py:87
Method__init__
(self, optimizer, periods, restart_weights=(1, ),
basicsr/models/lr_scheduler.py:161
Method__init__
(self, num_feat=64, res_scale=1, pytorch_init=False)
basicsr/models/archs/arch_util.py:86
Method__init__
(self, channels, eps=1e-6)
basicsr/models/archs/arch_util.py:293
Method__init__
(self, c, DW_Expand=2, FFN_Expand=2, drop_out_rate=0.)
basicsr/models/archs/NAFNet_arch.py:28
Method__init__
(self, *args, mask=False, train_size=(1, 3, 256, 256), fast_imp=False, **kwargs)
basicsr/models/archs/NAFNet_arch.py:166
Method__init__
(self, kernel_size=None, base_size=None, auto_pad=True, fast_imp=False, train_size=None)
basicsr/models/archs/seg_local_arch.py:11
Method__init__
(self, drop_rate, module)
basicsr/models/archs/NAFSSR_arch.py:63
Method__init__
(self, c, fusion=False, drop_out_rate=0.)
basicsr/models/archs/NAFSSR_arch.py:83
Method__init__
(self, up_scale=4, width=48, num_blks=16, img_channel=3, drop_path_rate=0., drop_out_rate=0., fusion_from=-1,
basicsr/models/archs/NAFSSR_arch.py:98
Method__init__
(self, *args, train_size=(1, 6, 30, 90), fast_imp=False, fusion_from=-1, fusion_to=1000, **kwargs)
basicsr/models/archs/NAFSSR_arch.py:132
Method__init__
(self, normalized_shape)
basicsr/models/archs/restormer_arch.py:26
Method__init__
(self, normalized_shape)
basicsr/models/archs/restormer_arch.py:42
Method__init__
(self, dim, LayerNorm_type)
basicsr/models/archs/restormer_arch.py:61
Method__init__
(self, dim, ffn_expansion_factor, bias)
basicsr/models/archs/restormer_arch.py:77
Method__init__
(self, dim, num_heads, bias)
basicsr/models/archs/restormer_arch.py:100
Method__init__
(self, dim, num_heads, ffn_expansion_factor, bias, LayerNorm_type)
basicsr/models/archs/restormer_arch.py:138
Method__init__
(self, in_c=3, embed_dim=48, bias=False)
basicsr/models/archs/restormer_arch.py:157
Method__init__
(self, n_feat)
basicsr/models/archs/restormer_arch.py:172
Method__init__
(self, inp_channels=3, out_channels=3, dim = 48, num_blocks = [4,6,6,8],
basicsr/models/archs/restormer_arch.py:194
Method__init__
(self, c, DW_Expand=1, FFN_Expand=2, drop_out_rate=0.)
basicsr/models/archs/Baseline_arch.py:23
Method__init__
(self, *args, train_size=(1, 3, 256, 256), fast_imp=False, **kwargs)
basicsr/models/archs/Baseline_arch.py:164
Method__init__
(self, kernel_size=None, base_size=None, auto_pad=True, fast_imp=False, train_size=None)
basicsr/models/archs/local_arch.py:11
Method__init__
(self, *args, train_size=(1, 3, 256, 256), fast_imp=False, **kwargs)
basicsr/models/archs/SegNAFNet_arch.py:179
Method__init__
(self, loss_weight=1.0, reduction='mean')
basicsr/models/losses/losses.py:70
Method__init__
(self, loss_weight=1.0, reduction='mean', toY=False)
basicsr/models/losses/losses.py:92
Method__iter__
(self)
basicsr/data/data_sampler.py:37
Method__iter__
(self)
basicsr/data/prefetch_dataloader.py:65
Method__len__
(self)
basicsr/data/data_sampler.py:52
Method__len__
(self)
basicsr/data/paired_image_dataset.py:134
Method__len__
(self)
basicsr/data/paired_image_mask_dataset.py:145
Method__len__
(self)
basicsr/data/paired_image_group_mask_dataset.py:156
Method__next__
(self)
basicsr/data/prefetch_dataloader.py:36
Methodadd_restored_face
(self, face)
basicsr/utils/face_util.py:176
Functionaugment
(imgs, hflip=True, rotation=True, flows=None, return_status=False, vflip=False)
basicsr/data/paired_image_group_mask_dataset.py:401
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:63
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:164
Functioncalculate_psnr_left
(img1, img2, crop_border, input_order='HWC',
basicsr/metrics/psnr_ssim.py:80
Functioncalculate_skimage_ssim_left
(img1, img2)
basicsr/metrics/psnr_ssim.py:355
Functioncalculate_ssim_left
(img1, img2, crop_border, input_order='HWC',
basicsr/metrics/psnr_ssim.py:339
Functioncallback
get the image data and update pbar.
basicsr/utils/lmdb_util.py:85
Methodclean_all
(self)
basicsr/utils/face_util.py:217
Methodconvert
(self, *args, train_size, **kwargs)
basicsr/models/archs/seg_local_arch.py:100
Functioncreate_lmdb_for_SIDD
()
basicsr/utils/create_lmdb.py:102
Functioncreate_lmdb_for_gopro
(folder_path)
scripts/data_preparation/lmdb_create_images.py:34
Functioncreate_lmdb_for_gopro
(folder_path)
scripts/data_preparation/lmdb_create_grouped_masks.py:42
Functioncreate_lmdb_for_gopro
(folder_path)
scripts/data_preparation/crop_masks.py:51
Functioncreate_lmdb_for_rain13k
()
basicsr/utils/create_lmdb.py:89
Functioncreate_lmdb_for_reds
()
basicsr/utils/create_lmdb.py:30
Functioncreate_lmdb_for_reloblur
(stage: str)
basicsr/utils/create_lmdb.py:51
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:166
Methoddetect_faces
Args: img_path (str): Image path. upsample_num_times (int): Upsamples the image before running the fa
basicsr/utils/face_util.py:62
Functiondict_constructor
(loader, node)
basicsr/utils/options.py:29
Functiondict_representer
(dumper, data)
basicsr/utils/options.py:26
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:14
Functionduf_downsample
Downsamping with Gaussian kernel used in the DUF official code. Args: x (Tensor): Frames to be downsampled, with shape (b, t, c, h, w).
basicsr/data/data_util.py:308
Methodextra_repr
(self)
basicsr/models/archs/seg_local_arch.py:24
Methodextra_repr
(self)
basicsr/models/archs/local_arch.py:24
Functionextract_inception_features
Extract inception features. Args: data_generator (generator): A data generator. inception (nn.Module): Inception model. l
basicsr/metrics/fid.py:29
Methodfeed_train_data
(self, data)
basicsr/models/image_clean_model.py:136
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:125
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:13
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:53
Methodforward
(self, x)
basicsr/models/archs/arch_util.py:96
Methodforward
(ctx, x, weight, bias, eps)
basicsr/models/archs/arch_util.py:267
Methodforward
(self, x)
basicsr/models/archs/arch_util.py:299
Methodforward
(self, *inputs)
basicsr/models/archs/arch_util.py:304
Methodforward
(self, x)
basicsr/models/archs/NAFNet_arch.py:23
Methodforward
(self, inp)
basicsr/models/archs/NAFNet_arch.py:59
Methodforward
(self, inp)
basicsr/models/archs/NAFNet_arch.py:133
Methodforward
(self, x_l, x_r)
basicsr/models/archs/NAFSSR_arch.py:44
Methodforward
(self, *feats)
basicsr/models/archs/NAFSSR_arch.py:68
Methodforward
(self, *feats)
basicsr/models/archs/NAFSSR_arch.py:88
Methodforward
(self, inp)
basicsr/models/archs/NAFSSR_arch.py:119
Methodforward
(self, x)
basicsr/models/archs/restormer_arch.py:37
← previousnext →201–300 of 372, ranked by callers