SYMBOL INDEX (673 symbols across 94 files)

FILE: basicsr/archs/BSRN_arch.py
  class DepthWiseConv (line 17) | class DepthWiseConv(nn.Module):
    method __init__ (line 18) | def __init__(self, in_ch, out_ch, kernel_size=3, stride=1, padding=1,
    method forward (line 45) | def forward(self, input):
  class BSConvU (line 51) | class BSConvU(torch.nn.Module):
    method __init__ (line 52) | def __init__(self, in_channels, out_channels, kernel_size=3, stride=1,...
    method forward (line 85) | def forward(self, fea):
  class BSConvS (line 91) | class BSConvS(torch.nn.Module):
    method __init__ (line 92) | def __init__(self, in_channels, out_channels, kernel_size=3, stride=1,...
    method forward (line 139) | def forward(self, x):
    method _reg_loss (line 145) | def _reg_loss(self):
  function stdv_channels (line 152) | def stdv_channels(F):
  function mean_channels (line 159) | def mean_channels(F):
  class CCALayer (line 165) | class CCALayer(nn.Module):
    method __init__ (line 166) | def __init__(self, channel, reduction=16):
    method forward (line 178) | def forward(self, x):
  class ChannelAttention (line 184) | class ChannelAttention(nn.Module):
    method __init__ (line 192) | def __init__(self, num_feat, squeeze_factor=16):
    method forward (line 198) | def forward(self, x):
  class ESA (line 203) | class ESA(nn.Module):
    method __init__ (line 204) | def __init__(self, num_feat=50, conv=nn.Conv2d, p=0.25):
    method forward (line 221) | def forward(self, input):
  class ESDB (line 236) | class ESDB(nn.Module):
    method __init__ (line 237) | def __init__(self, in_channels, out_channels, conv=nn.Conv2d, p=0.25):
    method forward (line 260) | def forward(self, input):
  function make_layer (line 283) | def make_layer(block, n_layers):
  class BSRN (line 291) | class BSRN(nn.Module):
    method __init__ (line 292) | def __init__(self, num_in_ch=3, num_feat=64, num_block=8, num_out_ch=3...
    method forward (line 335) | def forward(self, input):

FILE: basicsr/archs/Blocks.py
  class DepthWiseConv (line 18) | class DepthWiseConv(nn.Module):
    method __init__ (line 19) | def __init__(self, in_ch, out_ch, kernel_size=3, stride=1, padding=1,
    method forward (line 37) | def forward(self, input):
  class BSConvU (line 45) | class BSConvU(torch.nn.Module):
    method __init__ (line 46) | def __init__(self, in_channels, out_channels, kernel_size=3, stride=1,...
    method forward (line 83) | def forward(self, fea):
  class BSConvS (line 92) | class BSConvS(torch.nn.Module):
    method __init__ (line 93) | def __init__(self, in_channels, out_channels, kernel_size=3, stride=1,...
    method forward (line 150) | def forward(self, x):
    method _reg_loss (line 161) | def _reg_loss(self):
  class ChannelReplicate (line 169) | class ChannelReplicate(nn.Module):
    method __init__ (line 170) | def __init__(self, factor=3):
    method forward (line 174) | def forward(self, input):
  class ShuffleV1Block (line 183) | class ShuffleV1Block(nn.Module):
    method __init__ (line 184) | def __init__(self, inp, oup, *, group, first_group, mid_channels, ksiz...
    method forward (line 221) | def forward(self, old_x):
  class ShuffleV2Block (line 249) | class ShuffleV2Block(nn.Module):
    method __init__ (line 250) | def __init__(self, inp, oup, mid_channels, *, ksize, stride):
    method forward (line 292) | def forward(self, old_x):
    method channel_shuffle_v2 (line 303) | def channel_shuffle_v2(self, x):
  class FeatureMap (line 321) | class FeatureMap(nn.Module):
    method __init__ (line 323) | def __init__(self, query_dims):
    method new_feature_map (line 327) | def new_feature_map(self, device):
    method forward_queries (line 332) | def forward_queries(self, x):
    method forward_keys (line 336) | def forward_keys(self, x):
    method forward (line 340) | def forward(self, x):
    method factory (line 347) | def factory(cls, *args, **kwargs):
  class ActivationFunctionFeatureMap (line 358) | class ActivationFunctionFeatureMap(FeatureMap):
    method __init__ (line 361) | def __init__(self, query_dims, activation_function):
    method new_feature_map (line 365) | def new_feature_map(self, device):
    method forward (line 368) | def forward(self, x):
  class LinearAttention (line 372) | class LinearAttention(nn.Module):
    method __init__ (line 393) | def __init__(self, query_dimensions, feature_map=None, eps=1e-6,
    method forward (line 406) | def forward(self, queries, keys, values, attn_mask, query_lengths,

FILE: basicsr/archs/RFDN_arch.py
  function conv_layer (line 8) | def conv_layer(in_channels, out_channels, kernel_size, stride=1, dilatio...
  function norm (line 14) | def norm(norm_type, nc):
  function pad (line 25) | def pad(pad_type, padding):
  function get_valid_padding (line 38) | def get_valid_padding(kernel_size, dilation):
  function conv_block (line 44) | def conv_block(in_nc, out_nc, kernel_size, stride=1, dilation=1, groups=...
  function activation (line 57) | def activation(act_type, inplace=True, neg_slope=0.05, n_prelu=1):
  class ShortcutBlock (line 70) | class ShortcutBlock(nn.Module):
    method __init__ (line 71) | def __init__(self, submodule):
    method forward (line 75) | def forward(self, x):
  function mean_channels (line 80) | def mean_channels(F):
  function stdv_channels (line 86) | def stdv_channels(F):
  function sequential (line 93) | def sequential(*args):
  class ESA (line 108) | class ESA(nn.Module):
    method __init__ (line 109) | def __init__(self, n_feats, conv):
    method forward (line 122) | def forward(self, x):
  class RFDB (line 137) | class RFDB(nn.Module):
    method __init__ (line 138) | def __init__(self, in_channels, distillation_rate=0.25):
    method forward (line 153) | def forward(self, input):
  function pixelshuffle_block (line 174) | def pixelshuffle_block(in_channels, out_channels, upscale_factor=2, kern...
  function make_model (line 180) | def make_model(args, parent=False):
  class RFDN (line 186) | class RFDN(nn.Module):
    method __init__ (line 187) | def __init__(self, in_nc=3, nf=50, num_modules=4, out_nc=3, upscale=4):
    method forward (line 205) | def forward(self, input):
    method set_scale (line 220) | def set_scale(self, scale_idx):

FILE: basicsr/archs/Upsamplers.py
  class PA (line 14) | class PA(nn.Module):
    method __init__ (line 17) | def __init__(self, nf):
    method forward (line 22) | def forward(self, x):
  class PA_UP (line 30) | class PA_UP(nn.Module):
    method __init__ (line 31) | def __init__(self, nf, unf, out_nc, scale=4, conv=nn.Conv2d):
    method forward (line 45) | def forward(self, fea):
  class PA_UP_Dropout (line 55) | class PA_UP_Dropout(nn.Module):
    method __init__ (line 56) | def __init__(self, nf, unf, out_nc, scale=4, conv=nn.Conv2d, P=0.7):
    method forward (line 71) | def forward(self, fea):
  class UpsampleOneStep (line 82) | class UpsampleOneStep(nn.Sequential):
    method __init__ (line 90) | def __init__(self, scale, num_feat, num_out_ch, input_resolution=None):
    method flops (line 98) | def flops(self):
  class PixelShuffleDirect (line 104) | class PixelShuffleDirect(nn.Module):
    method __init__ (line 105) | def __init__(self, scale, num_feat, num_out_ch, input_resolution=None):
    method forward (line 109) | def forward(self, x):
  class PixelShuffleBlcok (line 113) | class PixelShuffleBlcok(nn.Module):
    method __init__ (line 114) | def __init__(self, in_feat, num_feat, num_out_ch):
    method forward (line 126) | def forward(self, x):
  class NearestConv (line 132) | class NearestConv(nn.Module):
    method __init__ (line 133) | def __init__(self, in_ch, num_feat, num_out_ch, conv=nn.Conv2d):
    method forward (line 143) | def forward(self, x):
  class NearestConvDropout (line 151) | class NearestConvDropout(nn.Module):
    method __init__ (line 152) | def __init__(self, in_ch, num_feat, num_out_ch, P=0.7):
    method forward (line 163) | def forward(self, x):

FILE: basicsr/archs/__init__.py
  function build_network (line 19) | def build_network(opt):

FILE: basicsr/archs/arch_util.py
  function default_init_weights (line 18) | def default_init_weights(module_list, scale=1, bias_fill=0, **kwargs):
  function make_layer (line 48) | def make_layer(basic_block, num_basic_block, **kwarg):
  class ResidualBlockNoBN (line 64) | class ResidualBlockNoBN(nn.Module):
    method __init__ (line 79) | def __init__(self, num_feat=64, res_scale=1, pytorch_init=False):
    method forward (line 89) | def forward(self, x):
  class Upsample (line 95) | class Upsample(nn.Sequential):
    method __init__ (line 103) | def __init__(self, scale, num_feat):
  function flow_warp (line 117) | def flow_warp(x, flow, interp_mode='bilinear', padding_mode='zeros', ali...
  function resize_flow (line 151) | def resize_flow(flow, size_type, sizes, interp_mode='bilinear', align_co...
  function pixel_unshuffle (line 190) | def pixel_unshuffle(x, scale):
  class DCNv2Pack (line 209) | class DCNv2Pack(ModulatedDeformConvPack):
    method forward (line 220) | def forward(self, x, feat):
  function _no_grad_trunc_normal_ (line 239) | def _no_grad_trunc_normal_(tensor, mean, std, a, b):
  function trunc_normal_ (line 277) | def trunc_normal_(tensor, mean=0., std=1., a=-2., b=2.):
  function _ntuple (line 304) | def _ntuple(n):

FILE: basicsr/archs/rfdnfinalB5_arch.py
  class ESA (line 18) | class ESA(nn.Module):
    method __init__ (line 19) | def __init__(self, num_feat=50, conv=nn.Conv2d, p=0.25):
    method forward (line 36) | def forward(self, input):
  class RFDB (line 52) | class RFDB(nn.Module):
    method __init__ (line 53) | def __init__(self, in_channels, out_channels, distillation_rate=0.25, ...
    method forward (line 77) | def forward(self, input):
  function make_layer (line 102) | def make_layer(block, n_layers):
  class RFDNFINALB5 (line 110) | class RFDNFINALB5(nn.Module):
    method __init__ (line 111) | def __init__(self, num_in_ch=3, num_feat=50, num_block=4, num_out_ch=3...
    method forward (line 155) | def forward(self, input):

FILE: basicsr/archs/vgg_arch.py
  function insert_bn (line 36) | def insert_bn(names):
  class VGGFeatureExtractor (line 55) | class VGGFeatureExtractor(nn.Module):
    method __init__ (line 78) | def __init__(self,
    method forward (line 141) | def forward(self, x):

FILE: basicsr/data/__init__.py
  function build_dataset (line 25) | def build_dataset(dataset_opt):
  function build_dataloader (line 40) | def build_dataloader(dataset, dataset_opt, num_gpu=1, dist=False, sample...
  function worker_init_fn (line 97) | def worker_init_fn(worker_id, num_workers, rank, seed):

FILE: basicsr/data/data_sampler.py
  class EnlargedSampler (line 6) | class EnlargedSampler(Sampler):
    method __init__ (line 21) | def __init__(self, dataset, num_replicas, rank, ratio=1):
    method __iter__ (line 29) | def __iter__(self):
    method __len__ (line 44) | def __len__(self):
    method set_epoch (line 47) | def set_epoch(self, epoch):

FILE: basicsr/data/data_util.py
  function read_img_seq (line 11) | def read_img_seq(path, require_mod_crop=False, scale=1, return_imgname=F...
  function generate_frame_indices (line 43) | def generate_frame_indices(crt_idx, max_frame_num, num_frames, padding='...
  function paired_paths_from_lmdb (line 95) | def paired_paths_from_lmdb(folders, keys):
  function paired_paths_from_meta_info_file (line 154) | def paired_paths_from_meta_info_file(folders, keys, meta_info_file, file...
  function paired_paths_from_folder (line 198) | def paired_paths_from_folder(folders, keys, filename_tmpl):
  function paths_from_folder (line 234) | def paths_from_folder(folder):
  function paths_from_lmdb (line 249) | def paths_from_lmdb(folder):
  function generate_gaussian_kernel (line 265) | def generate_gaussian_kernel(kernel_size=13, sigma=1.6):
  function duf_downsample (line 283) | def duf_downsample(x, kernel_size=13, scale=4):

FILE: basicsr/data/degradations.py
  function sigma_matrix2 (line 16) | def sigma_matrix2(sig_x, sig_y, theta):
  function mesh_grid (line 32) | def mesh_grid(kernel_size):
  function pdf2 (line 50) | def pdf2(sigma_matrix, grid):
  function cdf2 (line 66) | def cdf2(d_matrix, grid):
  function bivariate_Gaussian (line 84) | def bivariate_Gaussian(kernel_size, sig_x, sig_y, theta, grid=None, isot...
  function bivariate_generalized_Gaussian (line 112) | def bivariate_generalized_Gaussian(kernel_size, sig_x, sig_y, theta, bet...
  function bivariate_plateau (line 147) | def bivariate_plateau(kernel_size, sig_x, sig_y, theta, beta, grid=None,...
  function random_bivariate_Gaussian (line 179) | def random_bivariate_Gaussian(kernel_size,
  function random_bivariate_generalized_Gaussian (line 223) | def random_bivariate_generalized_Gaussian(kernel_size,
  function random_bivariate_plateau (line 275) | def random_bivariate_plateau(kernel_size,
  function random_mixed_kernels (line 327) | def random_mixed_kernels(kernel_list,
  function circular_lowpass_kernel (line 392) | def circular_lowpass_kernel(cutoff, kernel_size, pad_to=0):
  function generate_gaussian_noise (line 420) | def generate_gaussian_noise(img, sigma=10, gray_noise=False):
  function add_gaussian_noise (line 439) | def add_gaussian_noise(img, sigma=10, clip=True, rounds=False, gray_nois...
  function generate_gaussian_noise_pt (line 461) | def generate_gaussian_noise_pt(img, sigma=10, gray_noise=0):
  function add_gaussian_noise_pt (line 493) | def add_gaussian_noise_pt(img, sigma=10, gray_noise=0, clip=True, rounds...
  function random_generate_gaussian_noise (line 516) | def random_generate_gaussian_noise(img, sigma_range=(0, 10), gray_prob=0):
  function random_add_gaussian_noise (line 525) | def random_add_gaussian_noise(img, sigma_range=(0, 1.0), gray_prob=0, cl...
  function random_generate_gaussian_noise_pt (line 537) | def random_generate_gaussian_noise_pt(img, sigma_range=(0, 10), gray_pro...
  function random_add_gaussian_noise_pt (line 545) | def random_add_gaussian_noise_pt(img, sigma_range=(0, 1.0), gray_prob=0,...
  function generate_poisson_noise (line 560) | def generate_poisson_noise(img, scale=1.0, gray_noise=False):
  function add_poisson_noise (line 587) | def add_poisson_noise(img, scale=1.0, clip=True, rounds=False, gray_nois...
  function generate_poisson_noise_pt (line 610) | def generate_poisson_noise_pt(img, scale=1.0, gray_noise=0):
  function add_poisson_noise_pt (line 658) | def add_poisson_noise_pt(img, scale=1.0, clip=True, rounds=False, gray_n...
  function random_generate_poisson_noise (line 686) | def random_generate_poisson_noise(img, scale_range=(0, 1.0), gray_prob=0):
  function random_add_poisson_noise (line 695) | def random_add_poisson_noise(img, scale_range=(0, 1.0), gray_prob=0, cli...
  function random_generate_poisson_noise_pt (line 707) | def random_generate_poisson_noise_pt(img, scale_range=(0, 1.0), gray_pro...
  function random_add_poisson_noise_pt (line 715) | def random_add_poisson_noise_pt(img, scale_range=(0, 1.0), gray_prob=0, ...
  function add_jpg_compression (line 732) | def add_jpg_compression(img, quality=90):
  function random_add_jpg_compression (line 751) | def random_add_jpg_compression(img, quality_range=(90, 100)):

FILE: basicsr/data/ffhq_dataset.py
  class FFHQDataset (line 13) | class FFHQDataset(data.Dataset):
    method __init__ (line 26) | def __init__(self, opt):
    method __getitem__ (line 47) | def __getitem__(self, index):
    method __len__ (line 79) | def __len__(self):

FILE: basicsr/data/paired_image_dataset.py
  class PairedImageDataset (line 12) | class PairedImageDataset(data.Dataset):
    method __init__ (line 41) | def __init__(self, opt):
    method __getitem__ (line 66) | def __getitem__(self, index):
    method __len__ (line 108) | def __len__(self):

FILE: basicsr/data/prefetch_dataloader.py
  class PrefetchGenerator (line 7) | class PrefetchGenerator(threading.Thread):
    method __init__ (line 18) | def __init__(self, generator, num_prefetch_queue):
    method run (line 25) | def run(self):
    method __next__ (line 30) | def __next__(self):
    method __iter__ (line 36) | def __iter__(self):
  class PrefetchDataLoader (line 40) | class PrefetchDataLoader(DataLoader):
    method __init__ (line 55) | def __init__(self, num_prefetch_queue, **kwargs):
    method __iter__ (line 59) | def __iter__(self):
  class CPUPrefetcher (line 63) | class CPUPrefetcher():
    method __init__ (line 70) | def __init__(self, loader):
    method next (line 74) | def next(self):
    method reset (line 80) | def reset(self):
  class CUDAPrefetcher (line 84) | class CUDAPrefetcher():
    method __init__ (line 97) | def __init__(self, loader, opt):
    method preload (line 105) | def preload(self):
    method next (line 117) | def next(self):
    method reset (line 123) | def reset(self):

FILE: basicsr/data/reds_dataset.py
  class REDSDataset (line 14) | class REDSDataset(data.Dataset):
    method __init__ (line 53) | def __init__(self, opt):
    method __getitem__ (line 99) | def __getitem__(self, index):
    method __len__ (line 208) | def __len__(self):
  class REDSRecurrentDataset (line 213) | class REDSRecurrentDataset(data.Dataset):
    method __init__ (line 252) | def __init__(self, opt):
    method __getitem__ (line 298) | def __getitem__(self, index):
    method __len__ (line 359) | def __len__(self):

FILE: basicsr/data/single_image_dataset.py
  class SingleImageDataset (line 12) | class SingleImageDataset(data.Dataset):
    method __init__ (line 28) | def __init__(self, opt):
    method __getitem__ (line 48) | def __getitem__(self, index):
    method __len__ (line 68) | def __len__(self):

FILE: basicsr/data/transforms.py
  function mod_crop (line 6) | def mod_crop(img, scale):
  function paired_random_crop (line 26) | def paired_random_crop(img_gts, img_lqs, gt_patch_size, scale, gt_path=N...
  function augment (line 94) | def augment(imgs, hflip=True, rotation=True, flows=None, return_status=F...
  function img_rotate (line 161) | def img_rotate(img, angle, center=None, scale=1.0):

FILE: basicsr/data/video_test_dataset.py
  class VideoTestDataset (line 12) | class VideoTestDataset(data.Dataset):
    method __init__ (line 45) | def __init__(self, opt):
    method __getitem__ (line 101) | def __getitem__(self, index):
    method __len__ (line 128) | def __len__(self):
  class VideoTestVimeo90KDataset (line 133) | class VideoTestVimeo90KDataset(data.Dataset):
    method __init__ (line 153) | def __init__(self, opt):
    method __getitem__ (line 181) | def __getitem__(self, index):
    method __len__ (line 197) | def __len__(self):
  class VideoTestDUFDataset (line 202) | class VideoTestDUFDataset(VideoTestDataset):
    method __getitem__ (line 215) | def __getitem__(self, index):
  class VideoRecurrentTestDataset (line 255) | class VideoRecurrentTestDataset(VideoTestDataset):
    method __init__ (line 266) | def __init__(self, opt):
    method __getitem__ (line 271) | def __getitem__(self, index):
    method __len__ (line 286) | def __len__(self):

FILE: basicsr/data/vimeo90k_dataset.py
  class Vimeo90KDataset (line 12) | class Vimeo90KDataset(data.Dataset):
    method __init__ (line 52) | def __init__(self, opt):
    method __getitem__ (line 77) | def __getitem__(self, index):
    method __len__ (line 125) | def __len__(self):
  class Vimeo90KRecurrentDataset (line 130) | class Vimeo90KRecurrentDataset(Vimeo90KDataset):
    method __init__ (line 132) | def __init__(self, opt):
    method __getitem__ (line 138) | def __getitem__(self, index):
    method __len__ (line 191) | def __len__(self):

FILE: basicsr/losses/__init__.py
  function build_loss (line 14) | def build_loss(opt):

FILE: basicsr/losses/loss_util.py
  function reduce_loss (line 5) | def reduce_loss(loss, reduction):
  function weight_reduce_loss (line 25) | def weight_reduce_loss(loss, weight=None, reduction='mean'):
  function weighted_loss (line 57) | def weighted_loss(loss_func):

FILE: basicsr/losses/losses.py
  function l1_loss (line 15) | def l1_loss(pred, target):
  function mse_loss (line 20) | def mse_loss(pred, target):
  function charbonnier_loss (line 25) | def charbonnier_loss(pred, target, eps=1e-12):
  class L1Loss (line 30) | class L1Loss(nn.Module):
    method __init__ (line 39) | def __init__(self, loss_weight=1.0, reduction='mean'):
    method forward (line 47) | def forward(self, pred, target, weight=None, **kwargs):
  class MSELoss (line 58) | class MSELoss(nn.Module):
    method __init__ (line 67) | def __init__(self, loss_weight=1.0, reduction='mean'):
    method forward (line 75) | def forward(self, pred, target, weight=None, **kwargs):
  class CharbonnierLoss (line 86) | class CharbonnierLoss(nn.Module):
    method __init__ (line 100) | def __init__(self, loss_weight=1.0, reduction='mean', eps=1e-12):
    method forward (line 109) | def forward(self, pred, target, weight=None, **kwargs):
  class WeightedTVLoss (line 120) | class WeightedTVLoss(L1Loss):
    method __init__ (line 127) | def __init__(self, loss_weight=1.0, reduction='mean'):
    method forward (line 132) | def forward(self, pred, weight=None):
  class PerceptualLoss (line 149) | class PerceptualLoss(nn.Module):
    method __init__ (line 172) | def __init__(self,
    method forward (line 200) | def forward(self, x, gt):
    method _gram_mat (line 242) | def _gram_mat(self, x):
  class GANLoss (line 259) | class GANLoss(nn.Module):
    method __init__ (line 271) | def __init__(self, gan_type, real_label_val=1.0, fake_label_val=0.0, l...
    method _wgan_loss (line 291) | def _wgan_loss(self, input, target):
    method _wgan_softplus_loss (line 303) | def _wgan_softplus_loss(self, input, target):
    method get_target_label (line 320) | def get_target_label(self, input, target_is_real):
    method forward (line 337) | def forward(self, input, target_is_real, is_disc=False):
  class MultiScaleGANLoss (line 364) | class MultiScaleGANLoss(GANLoss):
    method __init__ (line 369) | def __init__(self, gan_type, real_label_val=1.0, fake_label_val=0.0, l...
    method forward (line 372) | def forward(self, input, target_is_real, is_disc=False):
  function r1_penalty (line 391) | def r1_penalty(real_pred, real_img):
  function g_path_regularize (line 408) | def g_path_regularize(fake_img, latents, mean_path_length, decay=0.01):
  function gradient_penalty_loss (line 420) | def gradient_penalty_loss(discriminator, real_data, fake_data, weight=No...
  class GANFeatLoss (line 460) | class GANFeatLoss(nn.Module):
    method __init__ (line 470) | def __init__(self, criterion='l1', loss_weight=1.0, reduction='mean'):
    method forward (line 483) | def forward(self, pred_fake, pred_real):

FILE: basicsr/metrics/__init__.py
  function calculate_metric (line 10) | def calculate_metric(data, opt):

FILE: basicsr/metrics/fid.py
  function load_patched_inception_v3 (line 10) | def load_patched_inception_v3(device='cuda', resize_input=True, normaliz...
  function extract_inception_features (line 19) | def extract_inception_features(data_generator, inception, len_generator=...
  function calculate_fid (line 50) | def calculate_fid(mu1, sigma1, mu2, sigma2, eps=1e-6):

FILE: basicsr/metrics/metric_util.py
  function reorder_image (line 6) | def reorder_image(img, input_order='HWC'):
  function to_y_channel (line 32) | def to_y_channel(img):

FILE: basicsr/metrics/niqe.py
  function estimate_aggd_param (line 13) | def estimate_aggd_param(block):
  function compute_feature (line 41) | def compute_feature(block):
  function niqe (line 68) | def niqe(img, mu_pris_param, cov_pris_param, gaussian_window, block_size...
  function calculate_niqe (line 144) | def calculate_niqe(img, crop_border, input_order='HWC', convert_to='y', ...

FILE: basicsr/metrics/psnr_ssim.py
  function calculate_psnr (line 9) | def calculate_psnr(img, img2, crop_border, input_order='HWC', test_y_cha...
  function _ssim (line 49) | def _ssim(img, img2):
  function calculate_ssim (line 84) | def calculate_ssim(img, img2, crop_border, input_order='HWC', test_y_cha...

FILE: basicsr/models/__init__.py
  function build_model (line 19) | def build_model(opt):

FILE: basicsr/models/base_model.py
  class BaseModel (line 13) | class BaseModel():
    method __init__ (line 16) | def __init__(self, opt):
    method feed_data (line 23) | def feed_data(self, data):
    method optimize_parameters (line 26) | def optimize_parameters(self):
    method get_current_visuals (line 29) | def get_current_visuals(self):
    method save (line 32) | def save(self, epoch, current_iter):
    method validation (line 36) | def validation(self, dataloader, current_iter, tb_logger, save_img=Fal...
    method _initialize_best_metric_results (line 50) | def _initialize_best_metric_results(self, dataset_name):
    method _update_best_metric_result (line 65) | def _update_best_metric_result(self, dataset_name, metric, val, curren...
    method model_ema (line 75) | def model_ema(self, decay=0.999):
    method get_current_log (line 84) | def get_current_log(self):
    method model_to_device (line 87) | def model_to_device(self, net):
    method get_optimizer (line 103) | def get_optimizer(self, optim_type, params, lr, **kwargs):
    method setup_schedulers (line 112) | def setup_schedulers(self):
    method get_bare_model (line 125) | def get_bare_model(self, net):
    method print_network (line 134) | def print_network(self, net):
    method _set_lr (line 153) | def _set_lr(self, lr_groups_l):
    method _get_init_lr (line 163) | def _get_init_lr(self):
    method update_learning_rate (line 171) | def update_learning_rate(self, current_iter, warmup_iter=-1):
    method get_current_learning_rate (line 194) | def get_current_learning_rate(self):
    method save_network (line 198) | def save_network(self, net, net_label, current_iter, param_key='params'):
    method _print_different_keys_loading (line 244) | def _print_different_keys_loading(self, crt_net, load_net, strict=True):
    method load_network (line 279) | def load_network(self, net, load_path, strict=True, param_key='params'):
    method save_training_state (line 308) | def save_training_state(self, epoch, current_iter):
    method resume_training (line 342) | def resume_training(self, resume_state):
    method reduce_loss_dict (line 357) | def reduce_loss_dict(self, loss_dict):

FILE: basicsr/models/edvr_model.py
  class EDVRModel (line 7) | class EDVRModel(VideoBaseModel):
    method __init__ (line 13) | def __init__(self, opt):
    method setup_optimizers (line 18) | def setup_optimizers(self):
    method optimize_parameters (line 48) | def optimize_parameters(self, current_iter):

FILE: basicsr/models/esrgan_model.py
  class ESRGANModel (line 9) | class ESRGANModel(SRGANModel):
    method optimize_parameters (line 12) | def optimize_parameters(self, current_iter):

FILE: basicsr/models/hifacegan_model.py
  class HiFaceGANModel (line 15) | class HiFaceGANModel(SRModel):
    method init_training_settings (line 21) | def init_training_settings(self):
    method setup_optimizers (line 63) | def setup_optimizers(self):
    method discriminate (line 74) | def discriminate(self, input_lq, output, ground_truth):
    method _divide_pred (line 98) | def _divide_pred(pred):
    method optimize_parameters (line 116) | def optimize_parameters(self, current_iter):
    method validation (line 194) | def validation(self, dataloader, current_iter, tb_logger, save_img=Fal...
    method nondist_validation (line 216) | def nondist_validation(self, dataloader, current_iter, tb_logger, save...
    method save (line 282) | def save(self, epoch, current_iter):

FILE: basicsr/models/lr_scheduler.py
  class MultiStepRestartLR (line 6) | class MultiStepRestartLR(_LRScheduler):
    method __init__ (line 19) | def __init__(self, optimizer, milestones, gamma=0.1, restarts=(0, ), r...
    method get_lr (line 27) | def get_lr(self):
  function get_position_from_periods (line 36) | def get_position_from_periods(iteration, cumulative_period):
  class CosineAnnealingRestartLR (line 57) | class CosineAnnealingRestartLR(_LRScheduler):
    method __init__ (line 77) | def __init__(self, optimizer, periods, restart_weights=(1, ), eta_min=...
    method get_lr (line 86) | def get_lr(self):

FILE: basicsr/models/sr_model.py
  class SRModel (line 15) | class SRModel(BaseModel):
    method __init__ (line 18) | def __init__(self, opt):
    method init_training_settings (line 35) | def init_training_settings(self):
    method setup_optimizers (line 73) | def setup_optimizers(self):
    method feed_data (line 87) | def feed_data(self, data):
    method optimize_parameters (line 92) | def optimize_parameters(self, current_iter):
    method test (line 121) | def test(self):
    method dist_validation (line 132) | def dist_validation(self, dataloader, current_iter, tb_logger, save_img):
    method nondist_validation (line 136) | def nondist_validation(self, dataloader, current_iter, tb_logger, save...
    method _log_validation_metric_values (line 203) | def _log_validation_metric_values(self, current_iter, dataset_name, tb...
    method get_current_visuals (line 218) | def get_current_visuals(self):
    method save (line 226) | def save(self, epoch, current_iter):

FILE: basicsr/models/srgan_model.py
  class SRGANModel (line 12) | class SRGANModel(SRModel):
    method init_training_settings (line 15) | def init_training_settings(self):
    method setup_optimizers (line 69) | def setup_optimizers(self):
    method optimize_parameters (line 80) | def optimize_parameters(self, current_iter):
    method save (line 138) | def save(self, epoch, current_iter):

FILE: basicsr/models/stylegan2_model.py
  class StyleGAN2Model (line 18) | class StyleGAN2Model(BaseModel):
    method __init__ (line 21) | def __init__(self, opt):
    method init_training_settings (line 42) | def init_training_settings(self):
    method setup_optimizers (line 88) | def setup_optimizers(self):
    method feed_data (line 169) | def feed_data(self, data):
    method make_noise (line 172) | def make_noise(self, batch, num_noise):
    method mixing_noise (line 179) | def mixing_noise(self, batch, prob):
    method optimize_parameters (line 185) | def optimize_parameters(self, current_iter):
    method test (line 256) | def test(self):
    method dist_validation (line 261) | def dist_validation(self, dataloader, current_iter, tb_logger, save_img):
    method nondist_validation (line 265) | def nondist_validation(self, dataloader, current_iter, tb_logger, save...
    method save (line 280) | def save(self, epoch, current_iter):

FILE: basicsr/models/swinir_model.py
  class SwinIRModel (line 9) | class SwinIRModel(SRModel):
    method test (line 11) | def test(self):

FILE: basicsr/models/video_base_model.py
  class VideoBaseModel (line 15) | class VideoBaseModel(SRModel):
    method dist_validation (line 18) | def dist_validation(self, dataloader, current_iter, tb_logger, save_img):
    method nondist_validation (line 113) | def nondist_validation(self, dataloader, current_iter, tb_logger, save...
    method _log_validation_metric_values (line 118) | def _log_validation_metric_values(self, current_iter, dataset_name, tb...

FILE: basicsr/models/video_gan_model.py
  class VideoGANModel (line 7) | class VideoGANModel(SRGANModel, VideoBaseModel):

FILE: basicsr/models/video_recurrent_gan_model.py
  class VideoRecurrentGANModel (line 12) | class VideoRecurrentGANModel(VideoRecurrentModel):
    method init_training_settings (line 14) | def init_training_settings(self):
    method setup_optimizers (line 68) | def setup_optimizers(self):
    method optimize_parameters (line 101) | def optimize_parameters(self, current_iter):
    method save (line 174) | def save(self, epoch, current_iter):

FILE: basicsr/models/video_recurrent_model.py
  class VideoRecurrentModel (line 15) | class VideoRecurrentModel(VideoBaseModel):
    method __init__ (line 17) | def __init__(self, opt):
    method setup_optimizers (line 22) | def setup_optimizers(self):
    method optimize_parameters (line 52) | def optimize_parameters(self, current_iter):
    method dist_validation (line 66) | def dist_validation(self, dataloader, current_iter, tb_logger, save_img):
    method test (line 176) | def test(self):

FILE: basicsr/ops/dcn/deform_conv.py
  class DeformConvFunction (line 33) | class DeformConvFunction(Function):
    method forward (line 36) | def forward(ctx,
    method backward (line 75) | def backward(ctx, grad_output):
    method _output_size (line 107) | def _output_size(input, weight, padding, dilation, stride):
  class ModulatedDeformConvFunction (line 121) | class ModulatedDeformConvFunction(Function):
    method forward (line 124) | def forward(ctx,
    method backward (line 157) | def backward(ctx, grad_output):
    method _infer_shape (line 177) | def _infer_shape(ctx, input, weight):
  class DeformConv (line 191) | class DeformConv(nn.Module):
    method __init__ (line 193) | def __init__(self,
    method reset_parameters (line 225) | def reset_parameters(self):
    method forward (line 232) | def forward(self, x, offset):
  class DeformConvPack (line 248) | class DeformConvPack(DeformConv):
    method __init__ (line 266) | def __init__(self, *args, **kwargs):
    method init_offset (line 279) | def init_offset(self):
    method forward (line 283) | def forward(self, x):
  class ModulatedDeformConv (line 289) | class ModulatedDeformConv(nn.Module):
    method __init__ (line 291) | def __init__(self,
    method init_weights (line 322) | def init_weights(self):
    method forward (line 331) | def forward(self, x, offset, mask):
  class ModulatedDeformConvPack (line 336) | class ModulatedDeformConvPack(ModulatedDeformConv):
    method __init__ (line 354) | def __init__(self, *args, **kwargs):
    method init_weights (line 367) | def init_weights(self):
    method forward (line 373) | def forward(self, x):

FILE: basicsr/ops/dcn/src/deform_conv_cuda.cpp
  function shape_check (line 62) | void shape_check(at::Tensor input, at::Tensor offset, at::Tensor *gradOu...
  function deform_conv_forward_cuda (line 152) | int deform_conv_forward_cuda(at::Tensor input, at::Tensor weight,
  function deform_conv_backward_input_cuda (line 262) | int deform_conv_backward_input_cuda(at::Tensor input, at::Tensor offset,
  function deform_conv_backward_parameters_cuda (line 376) | int deform_conv_backward_parameters_cuda(
  function modulated_deform_conv_cuda_forward (line 490) | void modulated_deform_conv_cuda_forward(
  function modulated_deform_conv_cuda_backward (line 571) | void modulated_deform_conv_cuda_backward(

FILE: basicsr/ops/dcn/src/deform_conv_ext.cpp
  function deform_conv_forward (line 52) | int deform_conv_forward(at::Tensor input, at::Tensor weight,
  function deform_conv_backward_input (line 70) | int deform_conv_backward_input(at::Tensor input, at::Tensor offset,
  function deform_conv_backward_parameters (line 89) | int deform_conv_backward_parameters(
  function modulated_deform_conv_forward (line 107) | void modulated_deform_conv_forward(
  function modulated_deform_conv_backward (line 127) | void modulated_deform_conv_backward(
  function PYBIND11_MODULE (line 150) | PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {

FILE: basicsr/ops/fused_act/fused_act.py
  class FusedLeakyReLUFunctionBackward (line 30) | class FusedLeakyReLUFunctionBackward(Function):
    method forward (line 33) | def forward(ctx, grad_output, out, negative_slope, scale):
    method backward (line 52) | def backward(ctx, gradgrad_input, gradgrad_bias):
  class FusedLeakyReLUFunction (line 60) | class FusedLeakyReLUFunction(Function):
    method forward (line 63) | def forward(ctx, input, bias, negative_slope, scale):
    method backward (line 73) | def backward(ctx, grad_output):
  class FusedLeakyReLU (line 81) | class FusedLeakyReLU(nn.Module):
    method __init__ (line 83) | def __init__(self, channel, negative_slope=0.2, scale=2**0.5):
    method forward (line 90) | def forward(self, input):
  function fused_leaky_relu (line 94) | def fused_leaky_relu(input, bias, negative_slope=0.2, scale=2**0.5):

FILE: basicsr/ops/fused_act/src/fused_bias_act.cpp
  function fused_bias_act (line 14) | torch::Tensor fused_bias_act(const torch::Tensor& input,
  function PYBIND11_MODULE (line 24) | PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {

FILE: basicsr/ops/upfirdn2d/src/upfirdn2d.cpp
  function upfirdn2d (line 13) | torch::Tensor upfirdn2d(const torch::Tensor& input, const torch::Tensor&...
  function PYBIND11_MODULE (line 22) | PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {

FILE: basicsr/ops/upfirdn2d/upfirdn2d.py
  class UpFirDn2dBackward (line 30) | class UpFirDn2dBackward(Function):
    method forward (line 33) | def forward(ctx, grad_output, kernel, grad_kernel, up, down, pad, g_pa...
    method backward (line 73) | def backward(ctx, gradgrad_input):
  class UpFirDn2d (line 97) | class UpFirDn2d(Function):
    method forward (line 100) | def forward(ctx, input, kernel, up, down, pad):
    method backward (line 135) | def backward(ctx, grad_output):
  function upfirdn2d (line 153) | def upfirdn2d(input, kernel, up=1, down=1, pad=(0, 0)):
  function upfirdn2d_native (line 162) | def upfirdn2d_native(input, kernel, up_x, up_y, down_x, down_y, pad_x0, ...

FILE: basicsr/test.py
  function test_pipeline (line 11) | def test_pipeline(root_path):

FILE: basicsr/train.py
  function init_tb_loggers (line 17) | def init_tb_loggers(opt):
  function create_train_val_dataloader (line 29) | def create_train_val_dataloader(opt, logger):
  function load_resume_state (line 68) | def load_resume_state(opt):
  function train_pipeline (line 91) | def train_pipeline(root_path):

FILE: basicsr/utils/diffjpeg.py
  function diff_round (line 26) | def diff_round(x):
  function quality_to_factor (line 32) | def quality_to_factor(quality):
  class RGB2YCbCrJpeg (line 49) | class RGB2YCbCrJpeg(nn.Module):
    method __init__ (line 53) | def __init__(self):
    method forward (line 60) | def forward(self, image):
  class ChromaSubsampling (line 73) | class ChromaSubsampling(nn.Module):
    method __init__ (line 77) | def __init__(self):
    method forward (line 80) | def forward(self, image):
  class BlockSplitting (line 98) | class BlockSplitting(nn.Module):
    method __init__ (line 102) | def __init__(self):
    method forward (line 106) | def forward(self, image):
  class DCT8x8 (line 121) | class DCT8x8(nn.Module):
    method __init__ (line 125) | def __init__(self):
    method forward (line 134) | def forward(self, image):
  class YQuantize (line 148) | class YQuantize(nn.Module):
    method __init__ (line 155) | def __init__(self, rounding):
    method forward (line 160) | def forward(self, image, factor=1):
  class CQuantize (line 178) | class CQuantize(nn.Module):
    method __init__ (line 185) | def __init__(self, rounding):
    method forward (line 190) | def forward(self, image, factor=1):
  class CompressJpeg (line 208) | class CompressJpeg(nn.Module):
    method __init__ (line 215) | def __init__(self, rounding=torch.round):
    method forward (line 222) | def forward(self, image, factor=1):
  class YDequantize (line 247) | class YDequantize(nn.Module):
    method __init__ (line 251) | def __init__(self):
    method forward (line 255) | def forward(self, image, factor=1):
  class CDequantize (line 272) | class CDequantize(nn.Module):
    method __init__ (line 276) | def __init__(self):
    method forward (line 280) | def forward(self, image, factor=1):
  class iDCT8x8 (line 297) | class iDCT8x8(nn.Module):
    method __init__ (line 301) | def __init__(self):
    method forward (line 310) | def forward(self, image):
  class BlockMerging (line 324) | class BlockMerging(nn.Module):
    method __init__ (line 328) | def __init__(self):
    method forward (line 331) | def forward(self, patches, height, width):
  class ChromaUpsampling (line 348) | class ChromaUpsampling(nn.Module):
    method __init__ (line 352) | def __init__(self):
    method forward (line 355) | def forward(self, y, cb, cr):
  class YCbCr2RGBJpeg (line 378) | class YCbCr2RGBJpeg(nn.Module):
    method __init__ (line 382) | def __init__(self):
    method forward (line 389) | def forward(self, image):
  class DeCompressJpeg (line 401) | class DeCompressJpeg(nn.Module):
    method __init__ (line 408) | def __init__(self, rounding=torch.round):
    method forward (line 417) | def forward(self, y, cb, cr, imgh, imgw, factor=1):
  class DiffJPEG (line 449) | class DiffJPEG(nn.Module):
    method __init__ (line 457) | def __init__(self, differentiable=True):
    method forward (line 467) | def forward(self, x, quality):

FILE: basicsr/utils/dist_util.py
  function init_dist (line 10) | def init_dist(launcher, backend='nccl', **kwargs):
  function _init_dist_pytorch (line 21) | def _init_dist_pytorch(backend, **kwargs):
  function _init_dist_slurm (line 28) | def _init_dist_slurm(backend, port=None):
  function get_dist_info (line 60) | def get_dist_info():
  function master_only (line 74) | def master_only(func):

FILE: basicsr/utils/download_util.py
  function download_file_from_google_drive (line 11) | def download_file_from_google_drive(file_id, save_path):
  function get_confirm_token (line 42) | def get_confirm_token(response):
  function save_response_content (line 49) | def save_response_content(response, destination, file_size=None, chunk_s...
  function load_file_from_url (line 70) | def load_file_from_url(url, model_dir=None, progress=True, file_name=None):

FILE: basicsr/utils/face_util.py
  class FaceRestorationHelper (line 15) | class FaceRestorationHelper(object):
    method __init__ (line 18) | def __init__(self, upscale_factor, face_size=512):
    method init_dlib (line 38) | def init_dlib(self, detection_path, landmark5_path, landmark68_path):
    method free_dlib_gpu_memory (line 44) | def free_dlib_gpu_memory(self):
    method read_input_image (line 49) | def read_input_image(self, img_path):
    method detect_faces (line 53) | def detect_faces(self, img_path, upsample_num_times=1, only_keep_large...
    method get_face_landmarks_5 (line 81) | def get_face_landmarks_5(self):
    method get_face_landmarks_68 (line 88) | def get_face_landmarks_68(self):
    method warp_crop_faces (line 121) | def warp_crop_faces(self, save_cropped_path=None, save_inverse_affine_...
    method add_restored_face (line 154) | def add_restored_face(self, face):
    method paste_faces_to_input_image (line 157) | def paste_faces_to_input_image(self, save_path):
    method clean_all (line 186) | def clean_all(self):

FILE: basicsr/utils/file_client.py
  class BaseStorageBackend (line 5) | class BaseStorageBackend(metaclass=ABCMeta):
    method get (line 14) | def get(self, filepath):
    method get_text (line 18) | def get_text(self, filepath):
  class MemcachedBackend (line 22) | class MemcachedBackend(BaseStorageBackend):
    method __init__ (line 32) | def __init__(self, server_list_cfg, client_cfg, sys_path=None):
    method get (line 47) | def get(self, filepath):
    method get_text (line 54) | def get_text(self, filepath):
  class HardDiskBackend (line 58) | class HardDiskBackend(BaseStorageBackend):
    method get (line 61) | def get(self, filepath):
    method get_text (line 67) | def get_text(self, filepath):
  class LmdbBackend (line 74) | class LmdbBackend(BaseStorageBackend):
    method __init__ (line 94) | def __init__(self, db_paths, client_keys='default', readonly=True, loc...
    method get (line 114) | def get(self, filepath, client_key):
    method get_text (line 128) | def get_text(self, filepath):
  class FileClient (line 132) | class FileClient(object):
    method __init__ (line 151) | def __init__(self, backend='disk', **kwargs):
    method get (line 158) | def get(self, filepath, client_key='default'):
    method get_text (line 166) | def get_text(self, filepath):

FILE: basicsr/utils/flow_util.py
  function flowread (line 7) | def flowread(flow_path, quantize=False, concat_axis=0, *args, **kwargs):
  function flowwrite (line 45) | def flowwrite(flow, filename, quantize=False, concat_axis=0, *args, **kw...
  function quantize_flow (line 76) | def quantize_flow(flow, max_val=0.02, norm=True):
  function dequantize_flow (line 102) | def dequantize_flow(dx, dy, max_val=0.02, denorm=True):
  function quantize (line 126) | def quantize(arr, min_val, max_val, levels, dtype=np.int64):
  function dequantize (line 150) | def dequantize(arr, min_val, max_val, levels, dtype=np.float64):

FILE: basicsr/utils/img_process_util.py
  function filter2D (line 7) | def filter2D(img, kernel):
  function usm_sharp (line 34) | def usm_sharp(img, weight=0.5, radius=50, threshold=10):
  class USMSharp (line 63) | class USMSharp(torch.nn.Module):
    method __init__ (line 65) | def __init__(self, radius=50, sigma=0):
    method forward (line 74) | def forward(self, img, weight=0.5, threshold=10):

FILE: basicsr/utils/img_util.py
  function img2tensor (line 9) | def img2tensor(imgs, bgr2rgb=True, float32=True):
  function tensor2img (line 38) | def tensor2img(tensor, rgb2bgr=True, out_type=np.uint8, min_max=(0, 1)):
  function tensor2img_fast (line 97) | def tensor2img_fast(tensor, rgb2bgr=True, min_max=(0, 1)):
  function imfrombytes (line 114) | def imfrombytes(content, flag='color', float32=False):
  function imwrite (line 135) | def imwrite(img, file_path, params=None, auto_mkdir=True):
  function crop_border (line 156) | def crop_border(imgs, crop_border):

FILE: basicsr/utils/lmdb_util.py
  function make_lmdb_from_imgs (line 9) | def make_lmdb_from_imgs(data_path,
  function read_img_worker (line 132) | def read_img_worker(path, key, compress_level):
  class LmdbMaker (line 156) | class LmdbMaker():
    method __init__ (line 167) | def __init__(self, lmdb_path, map_size=1024**4, batch=5000, compress_l...
    method put (line 182) | def put(self, img_byte, key, img_shape):
    method close (line 193) | def close(self):

FILE: basicsr/utils/logger.py
  class AvgTimer (line 10) | class AvgTimer():
    method __init__ (line 12) | def __init__(self, window=200):
    method start (line 20) | def start(self):
    method record (line 23) | def record(self):
    method get_current_time (line 38) | def get_current_time(self):
    method get_avg_time (line 41) | def get_avg_time(self):
  class MessageLogger (line 45) | class MessageLogger():
    method __init__ (line 58) | def __init__(self, opt, start_iter=1, tb_logger=None):
    method reset_start_time (line 68) | def reset_start_time(self):
    method __call__ (line 72) | def __call__(self, log_vars):
  function init_tb_logger (line 119) | def init_tb_logger(log_dir):
  function init_wandb_logger (line 126) | def init_wandb_logger(opt):
  function get_root_logger (line 149) | def get_root_logger(logger_name='basicsr', log_level=logging.INFO, log_f...
  function get_env_info (line 191) | def get_env_info():

FILE: basicsr/utils/matlab_functions.py
  function cubic (line 6) | def cubic(x):
  function calculate_weights_indices (line 16) | def calculate_weights_indices(in_length, out_length, scale, kernel, kern...
  function imresize (line 86) | def imresize(img, scale, antialiasing=True):
  function rgb2ycbcr (line 181) | def rgb2ycbcr(img, y_only=False):
  function bgr2ycbcr (line 214) | def bgr2ycbcr(img, y_only=False):
  function ycbcr2rgb (line 247) | def ycbcr2rgb(img):
  function ycbcr2bgr (line 276) | def ycbcr2bgr(img):
  function _convert_input_type_range (line 305) | def _convert_input_type_range(img):
  function _convert_output_type_range (line 332) | def _convert_output_type_range(img, dst_type):

FILE: basicsr/utils/misc.py
  function set_random_seed (line 11) | def set_random_seed(seed):
  function get_time_str (line 20) | def get_time_str():
  function mkdir_and_rename (line 24) | def mkdir_and_rename(path):
  function make_exp_dirs (line 38) | def make_exp_dirs(opt):
  function scandir (line 52) | def scandir(dir_path, suffix=None, recursive=False, full_path=False):
  function check_resume (line 94) | def check_resume(opt, resume_iter):
  function sizeof_fmt (line 127) | def sizeof_fmt(size, suffix='B'):

FILE: basicsr/utils/options.py
  function ordered_yaml (line 12) | def ordered_yaml():
  function dict2str (line 37) | def dict2str(opt, indent_level=1):
  function _postprocess_yml_value (line 58) | def _postprocess_yml_value(value):
  function parse_options (line 82) | def parse_options(root_path, is_train=True):
  function copy_opt_file (line 181) | def copy_opt_file(opt_file, experiments_root):

FILE: basicsr/utils/registry.py
  class Registry (line 4) | class Registry():
    method __init__ (line 30) | def __init__(self, name):
    method _do_register (line 38) | def _do_register(self, name, obj):
    method register (line 43) | def register(self, obj=None):
    method get (line 62) | def get(self, name):
    method __contains__ (line 68) | def __contains__(self, name):
    method __iter__ (line 71) | def __iter__(self):
    method keys (line 74) | def keys(self):

FILE: scripts/data_preparation/create_lmdb.py
  function create_lmdb_for_div2k (line 8) | def create_lmdb_for_div2k():
  function prepare_keys_div2k (line 45) | def prepare_keys_div2k(folder_path):
  function create_lmdb_for_reds (line 62) | def create_lmdb_for_reds():
  function prepare_keys_reds (line 85) | def prepare_keys_reds(folder_path):
  function create_lmdb_for_vimeo90k (line 102) | def create_lmdb_for_vimeo90k():
  function prepare_keys_vimeo90k (line 123) | def prepare_keys_vimeo90k(folder_path, train_list_path, mode):

FILE: scripts/data_preparation/download_datasets.py
  function download_dataset (line 9) | def download_dataset(dataset, file_ids):

FILE: scripts/data_preparation/extract_images_from_tfrecords.py
  function convert_celeba_tfrecords (line 10) | def convert_celeba_tfrecords(tf_file, log_resolution, save_root, save_ty...
  function convert_ffhq_tfrecords (line 69) | def convert_ffhq_tfrecords(tf_file, log_resolution, save_root, save_type...
  function make_ffhq_lmdb_from_imgs (line 119) | def make_ffhq_lmdb_from_imgs(folder_path, log_resolution, save_root, sav...

FILE: scripts/data_preparation/extract_subimages.py
  function main (line 12) | def main():
  function extract_subimages (line 79) | def extract_subimages(opt):
  function worker (line 109) | def worker(path, opt):

FILE: scripts/data_preparation/generate_meta_info.py
  function generate_meta_info_div2k (line 7) | def generate_meta_info_div2k():

FILE: scripts/data_preparation/prepare_hifacegan_dataset.py
  class Mosaic16x (line 6) | class Mosaic16x:
    method augment_image (line 13) | def augment_image(self, x):
  class DegradationSimulator (line 25) | class DegradationSimulator:
    method __init__ (line 35) | def __init__(self, ):
    method create_training_dataset (line 80) | def create_training_dataset(self, deg, gt_folder, lq_folder=None):

FILE: scripts/data_preparation/regroup_reds_dataset.py
  function regroup_reds_dataset (line 5) | def regroup_reds_dataset(train_path, val_path):

FILE: scripts/download_pretrained_models.py
  function download_pretrained_models (line 8) | def download_pretrained_models(method, file_ids):

FILE: scripts/metrics/calculate_fid_folder.py
  function calculate_fid_folder (line 11) | def calculate_fid_folder():

FILE: scripts/metrics/calculate_fid_stats_from_datasets.py
  function calculate_stats_from_dataset (line 11) | def calculate_stats_from_dataset():

FILE: scripts/metrics/calculate_lpips.py
  function main (line 15) | def main():

FILE: scripts/metrics/calculate_niqe.py
  function main (line 10) | def main(args):

FILE: scripts/metrics/calculate_psnr_ssim.py
  function main (line 11) | def main(args):

FILE: scripts/metrics/calculate_stylegan2_fid.py
  function calculate_stylegan2_fid (line 11) | def calculate_stylegan2_fid():

FILE: scripts/model_conversion/convert_dfdnet.py
  function convert_net (line 7) | def convert_net(ori_net, crt_net):

FILE: scripts/model_conversion/convert_models.py
  function convert_edvr (line 4) | def convert_edvr():
  function convert_edsr (line 102) | def convert_edsr(ori_net_path, crt_net_path, save_path, num_block=32):
  function convert_rcan_model (line 138) | def convert_rcan_model():
  function convert_esrgan_model (line 174) | def convert_esrgan_model():
  function convert_duf_model (line 202) | def convert_duf_model():

FILE: scripts/model_conversion/convert_stylegan.py
  function convert_net_g (line 6) | def convert_net_g(ori_net, crt_net):
  function convert_net_d (line 49) | def convert_net_d(ori_net, crt_net):

FILE: scripts/publish_models.py
  function update_sha (line 8) | def update_sha(paths):
  function convert_to_backward_compatible_models (line 39) | def convert_to_backward_compatible_models(paths):

FILE: setup.py
  function readme (line 14) | def readme():
  function get_git_hash (line 20) | def get_git_hash():
  function get_hash (line 45) | def get_hash():
  function write_version_py (line 61) | def write_version_py():
  function get_version (line 78) | def get_version():
  function make_cuda_ext (line 84) | def make_cuda_ext(name, module, sources, sources_cuda=None):
  function get_requirements (line 110) | def get_requirements(filename='requirements.txt'):

FILE: tests/test_archs/test_basicvsr_arch.py
  function test_basicvsr (line 6) | def test_basicvsr():
  function test_convresidualblocks (line 16) | def test_convresidualblocks():
  function test_iconvsr (line 26) | def test_iconvsr():

FILE: tests/test_archs/test_discriminator_arch.py
  function test_vggstylediscriminator (line 7) | def test_vggstylediscriminator():

FILE: tests/test_archs/test_duf_arch.py
  function test_duf (line 7) | def test_duf():
  function test_dynamicupsamplingfilter (line 31) | def test_dynamicupsamplingfilter():

FILE: tests/test_archs/test_ecbsr_arch.py
  function test_ecbsr (line 7) | def test_ecbsr():
  function test_seqconv3x3 (line 23) | def test_seqconv3x3():
  function test_ecb (line 57) | def test_ecb():

FILE: tests/test_archs/test_srresnet_arch.py
  function test_msrresnet (line 6) | def test_msrresnet():

FILE: tests/test_data/test_paired_image_dataset.py
  function test_pairedimagedataset (line 6) | def test_pairedimagedataset():

FILE: tests/test_data/test_single_image_dataset.py
  function test_singleimagedataset (line 6) | def test_singleimagedataset():

FILE: tests/test_losses/test_losses.py
  function test_pixellosses (line 8) | def test_pixellosses(loss_class):
  function test_weightedtvloss (line 41) | def test_weightedtvloss():

FILE: tests/test_metrics/test_psnr_ssim.py
  function test_calculate_psnr (line 7) | def test_calculate_psnr():
  function test_calculate_ssim (line 26) | def test_calculate_ssim():

FILE: tests/test_models/test_sr_model.py
  function test_srmodel (line 11) | def test_srmodel():