SYMBOL INDEX (952 symbols across 134 files)

FILE: basicsr/archs/__init__.py
  function build_network (line 18) | 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 75) | def __init__(self, num_feat=64, res_scale=1, pytorch_init=False):
    method forward (line 85) | def forward(self, x):
  class Upsample (line 91) | class Upsample(nn.Sequential):
    method __init__ (line 99) | def __init__(self, scale, num_feat):
  function flow_warp (line 113) | def flow_warp(x, flow, interp_mode='bilinear', padding_mode='zeros', ali...
  function resize_flow (line 147) | def resize_flow(flow, size_type, sizes, interp_mode='bilinear', align_co...
  function pixel_unshuffle (line 186) | def pixel_unshuffle(x, scale):
  class DCNv2Pack (line 205) | class DCNv2Pack(ModulatedDeformConvPack):
    method forward (line 215) | def forward(self, x, feat):
  function _no_grad_trunc_normal_ (line 234) | def _no_grad_trunc_normal_(tensor, mean, std, a, b):
  function trunc_normal_ (line 272) | def trunc_normal_(tensor, mean=0., std=1., a=-2., b=2.):
  function _ntuple (line 299) | def _ntuple(n):

FILE: basicsr/archs/basicvsr_arch.py
  class BasicVSR (line 12) | class BasicVSR(nn.Module):
    method __init__ (line 21) | def __init__(self, num_feat=64, num_block=15, spynet_path=None):
    method get_flow (line 44) | def get_flow(self, x):
    method forward (line 55) | def forward(self, x):
  class ConvResidualBlocks (line 101) | class ConvResidualBlocks(nn.Module):
    method __init__ (line 110) | def __init__(self, num_in_ch=3, num_out_ch=64, num_block=15):
    method forward (line 116) | def forward(self, fea):
  class IconVSR (line 121) | class IconVSR(nn.Module):
    method __init__ (line 133) | def __init__(self,
    method pad_spatial (line 169) | def pad_spatial(self, x):
    method get_flow (line 192) | def get_flow(self, x):
    method get_keyframe_feature (line 203) | def get_keyframe_feature(self, x, keyframe_idx):
    method forward (line 216) | def forward(self, x):
  class EDVRFeatureExtractor (line 271) | class EDVRFeatureExtractor(nn.Module):
    method __init__ (line 280) | def __init__(self, num_input_frame, num_feat, load_path):
    method forward (line 304) | def forward(self, x):

FILE: basicsr/archs/basicvsrpp_arch.py
  class BasicVSRPlusPlus (line 15) | class BasicVSRPlusPlus(nn.Module):
    method __init__ (line 43) | def __init__(self,
    method check_if_mirror_extended (line 109) | def check_if_mirror_extended(self, lqs):
    method compute_flow (line 123) | def compute_flow(self, lqs):
    method propagate (line 156) | def propagate(self, feats, flows, module_name):
    method upsample (line 237) | def upsample(self, lqs, feats):
    method forward (line 280) | def forward(self, lqs):
  class SecondOrderDeformableAlignment (line 347) | class SecondOrderDeformableAlignment(ModulatedDeformConvPack):
    method __init__ (line 365) | def __init__(self, *args, **kwargs):
    method init_offset (line 382) | def init_offset(self):
    method forward (line 392) | def forward(self, x, extra_feat, flow_1, flow_2):

FILE: basicsr/archs/dfdnet_arch.py
  class SFTUpBlock (line 12) | class SFTUpBlock(nn.Module):
    method __init__ (line 22) | def __init__(self, in_channel, out_channel, kernel_size=3, padding=1):
    method forward (line 45) | def forward(self, x, updated_feat):
  class DFDNet (line 57) | class DFDNet(nn.Module):
    method __init__ (line 67) | def __init__(self, num_feat, dict_path):
    method swap_feat (line 105) | def swap_feat(self, vgg_feat, updated_feat, dict_feat, location, part_...
    method put_dict_to_device (line 126) | def put_dict_to_device(self, x):
    method forward (line 133) | def forward(self, x, part_locations):

FILE: basicsr/archs/dfdnet_util.py
  class BlurFunctionBackward (line 8) | class BlurFunctionBackward(Function):
    method forward (line 11) | def forward(ctx, grad_output, kernel, kernel_flip):
    method backward (line 17) | def backward(ctx, gradgrad_output):
  class BlurFunction (line 23) | class BlurFunction(Function):
    method forward (line 26) | def forward(ctx, x, kernel, kernel_flip):
    method backward (line 32) | def backward(ctx, grad_output):
  class Blur (line 41) | class Blur(nn.Module):
    method __init__ (line 43) | def __init__(self, channel):
    method forward (line 53) | def forward(self, x):
  function calc_mean_std (line 57) | def calc_mean_std(feat, eps=1e-5):
  function adaptive_instance_normalization (line 74) | def adaptive_instance_normalization(content_feat, style_feat):
  function AttentionBlock (line 91) | def AttentionBlock(in_channel):
  function conv_block (line 97) | def conv_block(in_channels, out_channels, kernel_size=3, stride=1, dilat...
  class MSDilationBlock (line 123) | class MSDilationBlock(nn.Module):
    method __init__ (line 126) | def __init__(self, in_channels, kernel_size=3, dilation=(1, 1, 1, 1), ...
    method forward (line 141) | def forward(self, x):
  class UpResBlock (line 150) | class UpResBlock(nn.Module):
    method __init__ (line 152) | def __init__(self, in_channel):
    method forward (line 160) | def forward(self, x):

FILE: basicsr/archs/discriminator_arch.py
  class VGGStyleDiscriminator (line 9) | class VGGStyleDiscriminator(nn.Module):
    method __init__ (line 19) | def __init__(self, num_in_ch, num_feat, input_size=128):
    method forward (line 61) | def forward(self, x):
  class UNetDiscriminatorSN (line 91) | class UNetDiscriminatorSN(nn.Module):
    method __init__ (line 102) | def __init__(self, num_in_ch, num_feat=64, skip_connection=True):
    method forward (line 121) | def forward(self, x):

FILE: basicsr/archs/duf_arch.py
  class DenseBlocksTemporalReduce (line 9) | class DenseBlocksTemporalReduce(nn.Module):
    method __init__ (line 21) | def __init__(self, num_feat=64, num_grow_ch=32, adapt_official_weights...
    method forward (line 58) | def forward(self, x):
  class DenseBlocks (line 78) | class DenseBlocks(nn.Module):
    method __init__ (line 92) | def __init__(self, num_block, num_feat=64, num_grow_ch=16, adapt_offic...
    method forward (line 120) | def forward(self, x):
  class DynamicUpsamplingFilter (line 134) | class DynamicUpsamplingFilter(nn.Module):
    method __init__ (line 145) | def __init__(self, filter_size=(5, 5)):
    method forward (line 157) | def forward(self, x, filters):
  class DUF (line 182) | class DUF(nn.Module):
    method __init__ (line 204) | def __init__(self, scale=4, num_layer=52, adapt_official_weights=False):
    method forward (line 245) | def forward(self, x):

FILE: basicsr/archs/ecbsr_arch.py
  class SeqConv3x3 (line 8) | class SeqConv3x3(nn.Module):
    method __init__ (line 22) | def __init__(self, seq_type, in_channels, out_channels, depth_multipli...
    method forward (line 105) | def forward(self, x):
    method rep_params (line 131) | def rep_params(self):
  class ECB (line 156) | class ECB(nn.Module):
    method __init__ (line 170) | def __init__(self, in_channels, out_channels, depth_multiplier, act_ty...
    method forward (line 202) | def forward(self, x):
    method rep_params (line 214) | def rep_params(self):
  class ECBSR (line 236) | class ECBSR(nn.Module):
    method __init__ (line 252) | def __init__(self, num_in_ch, num_out_ch, num_block, num_channel, with...
    method forward (line 268) | def forward(self, x):

FILE: basicsr/archs/edsr_arch.py
  class EDSR (line 9) | class EDSR(nn.Module):
    method __init__ (line 30) | def __init__(self,
    method forward (line 50) | def forward(self, x):

FILE: basicsr/archs/edvr_arch.py
  class PCDAlignment (line 9) | class PCDAlignment(nn.Module):
    method __init__ (line 20) | def __init__(self, num_feat=64, deformable_groups=8):
    method forward (line 55) | def forward(self, nbr_feat_l, ref_feat_l):
  class TSAFusion (line 100) | class TSAFusion(nn.Module):
    method __init__ (line 115) | def __init__(self, num_feat=64, num_frame=5, center_frame_idx=2):
    method forward (line 140) | def forward(self, aligned_feat):
  class PredeblurModule (line 192) | class PredeblurModule(nn.Module):
    method __init__ (line 201) | def __init__(self, num_in_ch=3, num_feat=64, hr_in=False):
    method forward (line 223) | def forward(self, x):
  class EDVR (line 246) | class EDVR(nn.Module):
    method __init__ (line 271) | def __init__(self,
    method forward (line 325) | def forward(self, x):

FILE: basicsr/archs/hifacegan_arch.py
  class SPADEGenerator (line 10) | class SPADEGenerator(BaseNetwork):
    method __init__ (line 13) | def __init__(self,
    method encode (line 61) | def encode(self, input_tensor):
    method forward (line 71) | def forward(self, x):
    method mixed_guidance_forward (line 96) | def mixed_guidance_forward(self, input_x, seg=None, n=0, mode='progres...
  class HiFaceGAN (line 145) | class HiFaceGAN(SPADEGenerator):
    method __init__ (line 151) | def __init__(self,
    method encode (line 163) | def encode(self, input_tensor):
  class HiFaceGANDiscriminator (line 168) | class HiFaceGANDiscriminator(BaseNetwork):
    method __init__ (line 188) | def __init__(self,
    method downsample (line 208) | def downsample(self, x):
    method forward (line 213) | def forward(self, x):
  class NLayerDiscriminator (line 223) | class NLayerDiscriminator(BaseNetwork):
    method __init__ (line 226) | def __init__(self, input_nc, n_layers_d, num_feat, norm_d, keep_featur...
    method forward (line 251) | def forward(self, x):

FILE: basicsr/archs/hifacegan_util.py
  class SPADE (line 12) | class SPADE(nn.Module):
    method __init__ (line 14) | def __init__(self, config_text, norm_nc, label_nc):
    method forward (line 40) | def forward(self, x, segmap):
  class SPADEResnetBlock (line 57) | class SPADEResnetBlock(nn.Module):
    method __init__ (line 67) | def __init__(self, fin, fout, norm_g='spectralspadesyncbatch3x3', sema...
    method forward (line 95) | def forward(self, x, seg):
    method shortcut (line 102) | def shortcut(self, x, seg):
    method act (line 109) | def act(self, x):
  class BaseNetwork (line 113) | class BaseNetwork(nn.Module):
    method init_weights (line 116) | def init_weights(self, init_type='normal', gain=0.02):
    method forward (line 150) | def forward(self, x):
  function lip2d (line 154) | def lip2d(x, logit, kernel=3, stride=2, padding=1):
  class SoftGate (line 159) | class SoftGate(nn.Module):
    method forward (line 162) | def forward(self, x):
  class SimplifiedLIP (line 166) | class SimplifiedLIP(nn.Module):
    method __init__ (line 168) | def __init__(self, channels):
    method init_layer (line 174) | def init_layer(self):
    method forward (line 177) | def forward(self, x):
  class LIPEncoder (line 182) | class LIPEncoder(BaseNetwork):
    method __init__ (line 185) | def __init__(self, input_nc, ngf, sw, sh, n_2xdown, norm_layer=nn.Inst...
    method forward (line 213) | def forward(self, x):
  function get_nonspade_norm_layer (line 217) | def get_nonspade_norm_layer(norm_type='instance'):

FILE: basicsr/archs/inception.py
  class InceptionV3 (line 17) | class InceptionV3(nn.Module):
    method __init__ (line 32) | def __init__(self,
    method forward (line 124) | def forward(self, x):
  function fid_inception_v3 (line 155) | def fid_inception_v3():
  class FIDInceptionA (line 189) | class FIDInceptionA(models.inception.InceptionA):
    method __init__ (line 192) | def __init__(self, in_channels, pool_features):
    method forward (line 195) | def forward(self, x):
  class FIDInceptionC (line 214) | class FIDInceptionC(models.inception.InceptionC):
    method __init__ (line 217) | def __init__(self, in_channels, channels_7x7):
    method forward (line 220) | def forward(self, x):
  class FIDInceptionE_1 (line 242) | class FIDInceptionE_1(models.inception.InceptionE):
    method __init__ (line 245) | def __init__(self, in_channels):
    method forward (line 248) | def forward(self, x):
  class FIDInceptionE_2 (line 275) | class FIDInceptionE_2(models.inception.InceptionE):
    method __init__ (line 278) | def __init__(self, in_channels):
    method forward (line 281) | def forward(self, x):

FILE: basicsr/archs/rcan_arch.py
  class ChannelAttention (line 8) | class ChannelAttention(nn.Module):
    method __init__ (line 16) | def __init__(self, num_feat, squeeze_factor=16):
    method forward (line 22) | def forward(self, x):
  class RCAB (line 27) | class RCAB(nn.Module):
    method __init__ (line 36) | def __init__(self, num_feat, squeeze_factor=16, res_scale=1):
    method forward (line 44) | def forward(self, x):
  class ResidualGroup (line 49) | class ResidualGroup(nn.Module):
    method __init__ (line 59) | def __init__(self, num_feat, num_block, squeeze_factor=16, res_scale=1):
    method forward (line 66) | def forward(self, x):
  class RCAN (line 72) | class RCAN(nn.Module):
    method __init__ (line 96) | def __init__(self,
    method forward (line 124) | def forward(self, x):

FILE: basicsr/archs/ridnet_arch.py
  class MeanShift (line 8) | class MeanShift(nn.Conv2d):
    method __init__ (line 21) | def __init__(self, rgb_range, rgb_mean, rgb_std, sign=-1, requires_gra...
  class EResidualBlockNoBN (line 31) | class EResidualBlockNoBN(nn.Module):
    method __init__ (line 37) | def __init__(self, in_channels, out_channels):
    method forward (line 49) | def forward(self, x):
  class MergeRun (line 55) | class MergeRun(nn.Module):
    method __init__ (line 65) | def __init__(self, in_channels, out_channels, kernel_size=3, stride=1,...
    method forward (line 78) | def forward(self, x):
  class ChannelAttention (line 87) | class ChannelAttention(nn.Module):
    method __init__ (line 95) | def __init__(self, mid_channels, squeeze_factor=16):
    method forward (line 101) | def forward(self, x):
  class EAM (line 106) | class EAM(nn.Module):
    method __init__ (line 119) | def __init__(self, in_channels, mid_channels, out_channels):
    method forward (line 129) | def forward(self, x):
  class RIDNet (line 138) | class RIDNet(nn.Module):
    method __init__ (line 154) | def __init__(self,
    method forward (line 174) | def forward(self, x):

FILE: basicsr/archs/rrdbnet_arch.py
  class ResidualDenseBlock (line 9) | class ResidualDenseBlock(nn.Module):
    method __init__ (line 19) | def __init__(self, num_feat=64, num_grow_ch=32):
    method forward (line 32) | def forward(self, x):
  class RRDB (line 42) | class RRDB(nn.Module):
    method __init__ (line 52) | def __init__(self, num_feat, num_grow_ch=32):
    method forward (line 58) | def forward(self, x):
  class RRDBNet (line 67) | class RRDBNet(nn.Module):
    method __init__ (line 87) | def __init__(self, num_in_ch, num_out_ch, scale=4, num_feat=64, num_bl...
    method forward (line 105) | def forward(self, x):

FILE: basicsr/archs/spynet_arch.py
  class BasicModule (line 10) | class BasicModule(nn.Module):
    method __init__ (line 14) | def __init__(self):
    method forward (line 24) | def forward(self, tensor_input):
  class SpyNet (line 29) | class SpyNet(nn.Module):
    method __init__ (line 36) | def __init__(self, load_path=None):
    method preprocess (line 45) | def preprocess(self, tensor_input):
    method process (line 49) | def process(self, ref, supp):
    method forward (line 81) | def forward(self, ref, supp):

FILE: basicsr/archs/srresnet_arch.py
  class MSRResNet (line 9) | class MSRResNet(nn.Module):
    method __init__ (line 25) | def __init__(self, num_in_ch=3, num_out_ch=3, num_feat=64, num_block=1...
    method forward (line 52) | def forward(self, x):

FILE: basicsr/archs/srvgg_arch.py
  class SRVGGNetCompact (line 8) | class SRVGGNetCompact(nn.Module):
    method __init__ (line 23) | def __init__(self, num_in_ch=3, num_out_ch=3, num_feat=64, num_conv=16...
    method forward (line 61) | def forward(self, x):

FILE: basicsr/archs/stylegan2_arch.py
  class NormStyleCode (line 12) | class NormStyleCode(nn.Module):
    method forward (line 14) | def forward(self, x):
  function make_resample_kernel (line 26) | def make_resample_kernel(k):
  class UpFirDnUpsample (line 43) | class UpFirDnUpsample(nn.Module):
    method __init__ (line 56) | def __init__(self, resample_kernel, factor=2):
    method forward (line 64) | def forward(self, x):
    method __repr__ (line 68) | def __repr__(self):
  class UpFirDnDownsample (line 72) | class UpFirDnDownsample(nn.Module):
    method __init__ (line 81) | def __init__(self, resample_kernel, factor=2):
    method forward (line 89) | def forward(self, x):
    method __repr__ (line 93) | def __repr__(self):
  class UpFirDnSmooth (line 97) | class UpFirDnSmooth(nn.Module):
    method __init__ (line 108) | def __init__(self, resample_kernel, upsample_factor=1, downsample_fact...
    method forward (line 125) | def forward(self, x):
    method __repr__ (line 129) | def __repr__(self):
  class EqualLinear (line 134) | class EqualLinear(nn.Module):
    method __init__ (line 148) | def __init__(self, in_channels, out_channels, bias=True, bias_init_val...
    method forward (line 165) | def forward(self, x):
    method __repr__ (line 177) | def __repr__(self):
  class ModulatedConv2d (line 182) | class ModulatedConv2d(nn.Module):
    method __init__ (line 202) | def __init__(self,
    method forward (line 239) | def forward(self, x, style):
    method __repr__ (line 281) | def __repr__(self):
  class StyleConv (line 288) | class StyleConv(nn.Module):
    method __init__ (line 303) | def __init__(self,
    method forward (line 323) | def forward(self, x, style, noise=None):
  class ToRGB (line 336) | class ToRGB(nn.Module):
    method __init__ (line 347) | def __init__(self, in_channels, num_style_feat, upsample=True, resampl...
    method forward (line 357) | def forward(self, x, style, skip=None):
  class ConstantInput (line 377) | class ConstantInput(nn.Module):
    method __init__ (line 385) | def __init__(self, num_channel, size):
    method forward (line 389) | def forward(self, batch):
  class StyleGAN2Generator (line 395) | class StyleGAN2Generator(nn.Module):
    method __init__ (line 411) | def __init__(self,
    method make_noise (line 493) | def make_noise(self):
    method get_latent (line 504) | def get_latent(self, x):
    method mean_latent (line 507) | def mean_latent(self, num_latent):
    method forward (line 512) | def forward(self,
  class ScaledLeakyReLU (line 589) | class ScaledLeakyReLU(nn.Module):
    method __init__ (line 596) | def __init__(self, negative_slope=0.2):
    method forward (line 600) | def forward(self, x):
  class EqualConv2d (line 605) | class EqualConv2d(nn.Module):
    method __init__ (line 620) | def __init__(self, in_channels, out_channels, kernel_size, stride=1, p...
    method forward (line 635) | def forward(self, x):
    method __repr__ (line 646) | def __repr__(self):
  class ConvLayer (line 654) | class ConvLayer(nn.Sequential):
    method __init__ (line 671) | def __init__(self,
  class ResBlock (line 704) | class ResBlock(nn.Module):
    method __init__ (line 716) | def __init__(self, in_channels, out_channels, resample_kernel=(1, 3, 3...
    method forward (line 725) | def forward(self, x):
  class StyleGAN2Discriminator (line 734) | class StyleGAN2Discriminator(nn.Module):
    method __init__ (line 748) | def __init__(self, out_size, channel_multiplier=2, resample_kernel=(1,...
    method forward (line 783) | def forward(self, x):

FILE: basicsr/archs/stylegan2_bilinear_arch.py
  class NormStyleCode (line 11) | class NormStyleCode(nn.Module):
    method forward (line 13) | def forward(self, x):
  class EqualLinear (line 25) | class EqualLinear(nn.Module):
    method __init__ (line 39) | def __init__(self, in_channels, out_channels, bias=True, bias_init_val...
    method forward (line 56) | def forward(self, x):
    method __repr__ (line 68) | def __repr__(self):
  class ModulatedConv2d (line 73) | class ModulatedConv2d(nn.Module):
    method __init__ (line 91) | def __init__(self,
    method forward (line 121) | def forward(self, x, style):
    method __repr__ (line 156) | def __repr__(self):
  class StyleConv (line 163) | class StyleConv(nn.Module):
    method __init__ (line 176) | def __init__(self,
    method forward (line 196) | def forward(self, x, style, noise=None):
  class ToRGB (line 209) | class ToRGB(nn.Module):
    method __init__ (line 218) | def __init__(self, in_channels, num_style_feat, upsample=True, interpo...
    method forward (line 236) | def forward(self, x, style, skip=None):
  class ConstantInput (line 257) | class ConstantInput(nn.Module):
    method __init__ (line 265) | def __init__(self, num_channel, size):
    method forward (line 269) | def forward(self, batch):
  class StyleGAN2GeneratorBilinear (line 275) | class StyleGAN2GeneratorBilinear(nn.Module):
    method __init__ (line 288) | def __init__(self,
    method make_noise (line 370) | def make_noise(self):
    method get_latent (line 381) | def get_latent(self, x):
    method mean_latent (line 384) | def mean_latent(self, num_latent):
    method forward (line 389) | def forward(self,
  class ScaledLeakyReLU (line 466) | class ScaledLeakyReLU(nn.Module):
    method __init__ (line 473) | def __init__(self, negative_slope=0.2):
    method forward (line 477) | def forward(self, x):
  class EqualConv2d (line 482) | class EqualConv2d(nn.Module):
    method __init__ (line 497) | def __init__(self, in_channels, out_channels, kernel_size, stride=1, p...
    method forward (line 512) | def forward(self, x):
    method __repr__ (line 523) | def __repr__(self):
  class ConvLayer (line 531) | class ConvLayer(nn.Sequential):
    method __init__ (line 544) | def __init__(self,
  class ResBlock (line 580) | class ResBlock(nn.Module):
    method __init__ (line 588) | def __init__(self, in_channels, out_channels, interpolation_mode='bili...
    method forward (line 609) | def forward(self, x):

FILE: basicsr/archs/swinir_arch.py
  function drop_path (line 14) | def drop_path(x, drop_prob: float = 0., training: bool = False):
  class DropPath (line 29) | class DropPath(nn.Module):
    method __init__ (line 35) | def __init__(self, drop_prob=None):
    method forward (line 39) | def forward(self, x):
  class Mlp (line 43) | class Mlp(nn.Module):
    method __init__ (line 45) | def __init__(self, in_features, hidden_features=None, out_features=Non...
    method forward (line 54) | def forward(self, x):
  function window_partition (line 63) | def window_partition(x, window_size):
  function window_reverse (line 78) | def window_reverse(windows, window_size, h, w):
  class WindowAttention (line 95) | class WindowAttention(nn.Module):
    method __init__ (line 109) | def __init__(self, dim, window_size, num_heads, qkv_bias=True, qk_scal...
    method forward (line 144) | def forward(self, x, mask=None):
    method extra_repr (line 177) | def extra_repr(self) -> str:
    method flops (line 180) | def flops(self, n):
  class SwinTransformerBlock (line 194) | class SwinTransformerBlock(nn.Module):
    method __init__ (line 213) | def __init__(self,
    method calculate_mask (line 262) | def calculate_mask(self, x_size):
    method forward (line 283) | def forward(self, x, x_size):
    method extra_repr (line 325) | def extra_repr(self) -> str:
    method flops (line 329) | def flops(self):
  class PatchMerging (line 344) | class PatchMerging(nn.Module):
    method __init__ (line 353) | def __init__(self, input_resolution, dim, norm_layer=nn.LayerNorm):
    method forward (line 360) | def forward(self, x):
    method extra_repr (line 383) | def extra_repr(self) -> str:
    method flops (line 386) | def flops(self):
  class BasicLayer (line 393) | class BasicLayer(nn.Module):
    method __init__ (line 413) | def __init__(self,
    method forward (line 458) | def forward(self, x, x_size):
    method extra_repr (line 468) | def extra_repr(self) -> str:
    method flops (line 471) | def flops(self):
  class RSTB (line 480) | class RSTB(nn.Module):
    method __init__ (line 503) | def __init__(self,
    method forward (line 557) | def forward(self, x, x_size):
    method flops (line 560) | def flops(self):
  class PatchEmbed (line 571) | class PatchEmbed(nn.Module):
    method __init__ (line 582) | def __init__(self, img_size=224, patch_size=4, in_chans=3, embed_dim=9...
    method forward (line 600) | def forward(self, x):
    method flops (line 606) | def flops(self):
  class PatchUnEmbed (line 614) | class PatchUnEmbed(nn.Module):
    method __init__ (line 625) | def __init__(self, img_size=224, patch_size=4, in_chans=3, embed_dim=9...
    method forward (line 638) | def forward(self, x, x_size):
    method flops (line 642) | def flops(self):
  class Upsample (line 647) | class Upsample(nn.Sequential):
    method __init__ (line 655) | def __init__(self, scale, num_feat):
  class UpsampleOneStep (line 669) | class UpsampleOneStep(nn.Sequential):
    method __init__ (line 679) | def __init__(self, scale, num_feat, num_out_ch, input_resolution=None):
    method flops (line 687) | def flops(self):
  class SwinIR (line 694) | class SwinIR(nn.Module):
    method __init__ (line 722) | def __init__(self,
    method _init_weights (line 859) | def _init_weights(self, m):
    method no_weight_decay (line 869) | def no_weight_decay(self):
    method no_weight_decay_keywords (line 873) | def no_weight_decay_keywords(self):
    method forward_features (line 876) | def forward_features(self, x):
    method forward (line 891) | def forward(self, x):
    method flops (line 924) | def flops(self):

FILE: basicsr/archs/tof_arch.py
  class BasicModule (line 9) | class BasicModule(nn.Module):
    method __init__ (line 16) | def __init__(self):
    method forward (line 29) | def forward(self, tensor_input):
  class SPyNetTOF (line 42) | class SPyNetTOF(nn.Module):
    method __init__ (line 59) | def __init__(self, load_path=None):
    method forward (line 66) | def forward(self, ref, supp):
  class TOFlow (line 94) | class TOFlow(nn.Module):
    method __init__ (line 111) | def __init__(self, adapt_official_weights=False):
    method normalize (line 131) | def normalize(self, img):
    method denormalize (line 134) | def denormalize(self, img):
    method forward (line 137) | def forward(self, lrs):

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 156) | def paired_paths_from_meta_info_file(folders, keys, meta_info_file, file...
  function paired_paths_from_folder (line 200) | def paired_paths_from_folder(folders, keys, filename_tmpl):
  function paths_from_folder (line 236) | def paths_from_folder(folder):
  function paths_from_lmdb (line 251) | def paths_from_lmdb(folder):
  function generate_gaussian_kernel (line 267) | def generate_gaussian_kernel(kernel_size=13, sigma=1.6):
  function duf_downsample (line 285) | 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 143) | def bivariate_plateau(kernel_size, sig_x, sig_y, theta, beta, grid=None,...
  function random_bivariate_Gaussian (line 176) | def random_bivariate_Gaussian(kernel_size,
  function random_bivariate_generalized_Gaussian (line 220) | def random_bivariate_generalized_Gaussian(kernel_size,
  function random_bivariate_plateau (line 272) | def random_bivariate_plateau(kernel_size,
  function random_mixed_kernels (line 324) | def random_mixed_kernels(kernel_list,
  function circular_lowpass_kernel (line 389) | def circular_lowpass_kernel(cutoff, kernel_size, pad_to=0):
  function generate_gaussian_noise (line 419) | def generate_gaussian_noise(img, sigma=10, gray_noise=False):
  function add_gaussian_noise (line 438) | def add_gaussian_noise(img, sigma=10, clip=True, rounds=False, gray_nois...
  function generate_gaussian_noise_pt (line 460) | def generate_gaussian_noise_pt(img, sigma=10, gray_noise=0):
  function add_gaussian_noise_pt (line 492) | def add_gaussian_noise_pt(img, sigma=10, gray_noise=0, clip=True, rounds...
  function random_generate_gaussian_noise (line 515) | def random_generate_gaussian_noise(img, sigma_range=(0, 10), gray_prob=0):
  function random_add_gaussian_noise (line 524) | def random_add_gaussian_noise(img, sigma_range=(0, 1.0), gray_prob=0, cl...
  function random_generate_gaussian_noise_pt (line 536) | def random_generate_gaussian_noise_pt(img, sigma_range=(0, 10), gray_pro...
  function random_add_gaussian_noise_pt (line 544) | def random_add_gaussian_noise_pt(img, sigma_range=(0, 1.0), gray_prob=0,...
  function generate_poisson_noise (line 559) | def generate_poisson_noise(img, scale=1.0, gray_noise=False):
  function add_poisson_noise (line 586) | def add_poisson_noise(img, scale=1.0, clip=True, rounds=False, gray_nois...
  function generate_poisson_noise_pt (line 609) | def generate_poisson_noise_pt(img, scale=1.0, gray_noise=0):
  function add_poisson_noise_pt (line 657) | def add_poisson_noise_pt(img, scale=1.0, clip=True, rounds=False, gray_n...
  function random_generate_poisson_noise (line 685) | def random_generate_poisson_noise(img, scale_range=(0, 1.0), gray_prob=0):
  function random_add_poisson_noise (line 694) | def random_add_poisson_noise(img, scale_range=(0, 1.0), gray_prob=0, cli...
  function random_generate_poisson_noise_pt (line 706) | def random_generate_poisson_noise_pt(img, scale_range=(0, 1.0), gray_pro...
  function random_add_poisson_noise_pt (line 714) | def random_add_poisson_noise_pt(img, scale_range=(0, 1.0), gray_prob=0, ...
  function add_jpg_compression (line 731) | def add_jpg_compression(img, quality=90):
  function random_add_jpg_compression (line 750) | 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 11) | class PairedImageDataset(data.Dataset):
    method __init__ (line 38) | def __init__(self, opt):
    method __getitem__ (line 63) | def __getitem__(self, index):
    method __len__ (line 105) | def __len__(self):

FILE: basicsr/data/prefetch_dataloader.py
  class PrefetchGenerator (line 7) | class PrefetchGenerator(threading.Thread):
    method __init__ (line 17) | def __init__(self, generator, num_prefetch_queue):
    method run (line 24) | def run(self):
    method __next__ (line 29) | def __next__(self):
    method __iter__ (line 35) | def __iter__(self):
  class PrefetchDataLoader (line 39) | class PrefetchDataLoader(DataLoader):
    method __init__ (line 53) | def __init__(self, num_prefetch_queue, **kwargs):
    method __iter__ (line 57) | def __iter__(self):
  class CPUPrefetcher (line 61) | class CPUPrefetcher():
    method __init__ (line 68) | def __init__(self, loader):
    method next (line 72) | def next(self):
    method reset (line 78) | def reset(self):
  class CUDAPrefetcher (line 82) | class CUDAPrefetcher():
    method __init__ (line 94) | def __init__(self, loader, opt):
    method preload (line 102) | def preload(self):
    method next (line 114) | def next(self):
    method reset (line 120) | def reset(self):

FILE: basicsr/data/realesrgan_dataset.py
  class RealESRGANDataset (line 18) | class RealESRGANDataset(data.Dataset):
    method __init__ (line 36) | def __init__(self, opt):
    method __getitem__ (line 84) | def __getitem__(self, index):
    method __len__ (line 192) | def __len__(self):

FILE: basicsr/data/realesrgan_paired_dataset.py
  class RealESRGANPairedDataset (line 12) | class RealESRGANPairedDataset(data.Dataset):
    method __init__ (line 39) | def __init__(self, opt):
    method __getitem__ (line 73) | def __getitem__(self, index):
    method __len__ (line 105) | def __len__(self):

FILE: basicsr/data/reds_dataset.py
  class REDSDataset (line 14) | class REDSDataset(data.Dataset):
    method __init__ (line 49) | def __init__(self, opt):
    method __getitem__ (line 95) | def __getitem__(self, index):
    method __len__ (line 204) | def __len__(self):
  class REDSRecurrentDataset (line 209) | class REDSRecurrentDataset(data.Dataset):
    method __init__ (line 244) | def __init__(self, opt):
    method __getitem__ (line 290) | def __getitem__(self, index):
    method __len__ (line 351) | def __len__(self):

FILE: basicsr/data/single_image_dataset.py
  class SingleImageDataset (line 11) | class SingleImageDataset(data.Dataset):
    method __init__ (line 27) | def __init__(self, opt):
    method __getitem__ (line 47) | def __getitem__(self, index):
    method __len__ (line 67) | 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 46) | def __init__(self, opt):
    method __getitem__ (line 102) | def __getitem__(self, index):
    method __len__ (line 129) | def __len__(self):
  class VideoTestVimeo90KDataset (line 134) | 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 212) | def __getitem__(self, index):
  class VideoRecurrentTestDataset (line 252) | class VideoRecurrentTestDataset(VideoTestDataset):
    method __init__ (line 262) | def __init__(self, opt):
    method __getitem__ (line 267) | def __getitem__(self, index):
    method __len__ (line 282) | def __len__(self):

FILE: basicsr/data/vimeo90k_dataset.py
  class Vimeo90KDataset (line 12) | class Vimeo90KDataset(data.Dataset):
    method __init__ (line 59) | def __init__(self, opt):
    method __getitem__ (line 84) | def __getitem__(self, index):
    method __len__ (line 132) | def __len__(self):
  class Vimeo90KRecurrentDataset (line 137) | class Vimeo90KRecurrentDataset(Vimeo90KDataset):
    method __init__ (line 139) | def __init__(self, opt):
    method __getitem__ (line 145) | def __getitem__(self, index):
    method __len__ (line 198) | def __len__(self):

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

FILE: basicsr/losses/basic_loss.py
  function l1_loss (line 13) | def l1_loss(pred, target):
  function mse_loss (line 18) | def mse_loss(pred, target):
  function charbonnier_loss (line 23) | def charbonnier_loss(pred, target, eps=1e-12):
  class L1Loss (line 28) | class L1Loss(nn.Module):
    method __init__ (line 37) | def __init__(self, loss_weight=1.0, reduction='mean'):
    method forward (line 45) | def forward(self, pred, target, weight=None, **kwargs):
  class MSELoss (line 56) | class MSELoss(nn.Module):
    method __init__ (line 65) | def __init__(self, loss_weight=1.0, reduction='mean'):
    method forward (line 73) | def forward(self, pred, target, weight=None, **kwargs):
  class CharbonnierLoss (line 84) | class CharbonnierLoss(nn.Module):
    method __init__ (line 98) | def __init__(self, loss_weight=1.0, reduction='mean', eps=1e-12):
    method forward (line 107) | def forward(self, pred, target, weight=None, **kwargs):
  class WeightedTVLoss (line 118) | class WeightedTVLoss(L1Loss):
    method __init__ (line 125) | def __init__(self, loss_weight=1.0, reduction='mean'):
    method forward (line 130) | def forward(self, pred, weight=None):
  class PerceptualLoss (line 147) | class PerceptualLoss(nn.Module):
    method __init__ (line 170) | def __init__(self,
    method forward (line 198) | def forward(self, x, gt):
    method _gram_mat (line 240) | def _gram_mat(self, x):

FILE: basicsr/losses/gan_loss.py
  class GANLoss (line 11) | class GANLoss(nn.Module):
    method __init__ (line 23) | def __init__(self, gan_type, real_label_val=1.0, fake_label_val=0.0, l...
    method _wgan_loss (line 43) | def _wgan_loss(self, input, target):
    method _wgan_softplus_loss (line 55) | def _wgan_softplus_loss(self, input, target):
    method get_target_label (line 72) | def get_target_label(self, input, target_is_real):
    method forward (line 89) | def forward(self, input, target_is_real, is_disc=False):
  class MultiScaleGANLoss (line 116) | class MultiScaleGANLoss(GANLoss):
    method __init__ (line 121) | def __init__(self, gan_type, real_label_val=1.0, fake_label_val=0.0, l...
    method forward (line 124) | def forward(self, input, target_is_real, is_disc=False):
  function r1_penalty (line 143) | def r1_penalty(real_pred, real_img):
  function g_path_regularize (line 159) | def g_path_regularize(fake_img, latents, mean_path_length, decay=0.01):
  function gradient_penalty_loss (line 171) | def gradient_penalty_loss(discriminator, real_data, fake_data, weight=No...

FILE: basicsr/losses/loss_util.py
  function reduce_loss (line 6) | def reduce_loss(loss, reduction):
  function weight_reduce_loss (line 26) | def weight_reduce_loss(loss, weight=None, reduction='mean'):
  function weighted_loss (line 58) | def weighted_loss(loss_func):
  function get_local_weights (line 99) | def get_local_weights(residual, ksize):
  function get_refined_artifact_map (line 121) | def get_refined_artifact_map(img_gt, img_output, img_ema, ksize):

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 145) | def calculate_niqe(img, crop_border, input_order='HWC', convert_to='y', ...

FILE: basicsr/metrics/psnr_ssim.py
  function calculate_psnr (line 12) | def calculate_psnr(img, img2, crop_border, input_order='HWC', test_y_cha...
  function calculate_psnr_pt (line 52) | def calculate_psnr_pt(img, img2, crop_border, test_y_channel=False, **kw...
  function calculate_ssim (line 85) | def calculate_ssim(img, img2, crop_border, input_order='HWC', test_y_cha...
  function calculate_ssim_pt (line 132) | def calculate_ssim_pt(img, img2, crop_border, test_y_channel=False, **kw...
  function _ssim (line 170) | def _ssim(img, img2):
  function _ssim_pth (line 201) | def _ssim_pth(img, img2):

FILE: basicsr/metrics/test_metrics/test_psnr_ssim.py
  function test (line 9) | def test(img_path, img_path2, crop_border, test_y_channel=False):

FILE: basicsr/models/__init__.py
  function build_model (line 18) | 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 122) | def setup_schedulers(self):
    method get_bare_model (line 135) | def get_bare_model(self, net):
    method print_network (line 144) | def print_network(self, net):
    method _set_lr (line 163) | def _set_lr(self, lr_groups_l):
    method _get_init_lr (line 173) | def _get_init_lr(self):
    method update_learning_rate (line 181) | def update_learning_rate(self, current_iter, warmup_iter=-1):
    method get_current_learning_rate (line 204) | def get_current_learning_rate(self):
    method save_network (line 208) | def save_network(self, net, net_label, current_iter, param_key='params'):
    method _print_different_keys_loading (line 254) | def _print_different_keys_loading(self, crt_net, load_net, strict=True):
    method load_network (line 289) | def load_network(self, net, load_path, strict=True, param_key='params'):
    method save_training_state (line 318) | def save_training_state(self, epoch, current_iter):
    method resume_training (line 352) | def resume_training(self, resume_state):
    method reduce_loss_dict (line 367) | 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/realesrgan_model.py
  class RealESRGANModel (line 17) | class RealESRGANModel(SRGANModel):
    method __init__ (line 25) | def __init__(self, opt):
    method _dequeue_and_enqueue (line 32) | def _dequeue_and_enqueue(self):
    method feed_data (line 69) | def feed_data(self, data):
    method nondist_validation (line 187) | def nondist_validation(self, dataloader, current_iter, tb_logger, save...
    method optimize_parameters (line 193) | def optimize_parameters(self, current_iter):

FILE: basicsr/models/realesrnet_model.py
  class RealESRNetModel (line 15) | class RealESRNetModel(SRModel):
    method __init__ (line 24) | def __init__(self, opt):
    method _dequeue_and_enqueue (line 31) | def _dequeue_and_enqueue(self):
    method feed_data (line 68) | def feed_data(self, data):
    method nondist_validation (line 185) | def nondist_validation(self, dataloader, current_iter, tb_logger, save...

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 test_selfensemble (line 132) | def test_selfensemble(self):
    method dist_validation (line 180) | def dist_validation(self, dataloader, current_iter, tb_logger, save_img):
    method nondist_validation (line 184) | def nondist_validation(self, dataloader, current_iter, tb_logger, save...
    method _log_validation_metric_values (line 251) | def _log_validation_metric_values(self, current_iter, dataset_name, tb...
    method get_current_visuals (line 266) | def get_current_visuals(self):
    method save (line 274) | 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 74) | def setup_optimizers(self):
    method optimize_parameters (line 85) | def optimize_parameters(self, current_iter):
    method save (line 143) | 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/color_util.py
  function rgb2ycbcr (line 5) | def rgb2ycbcr(img, y_only=False):
  function bgr2ycbcr (line 38) | def bgr2ycbcr(img, y_only=False):
  function ycbcr2rgb (line 71) | def ycbcr2rgb(img):
  function ycbcr2bgr (line 100) | def ycbcr2bgr(img):
  function _convert_input_type_range (line 129) | def _convert_input_type_range(img):
  function _convert_output_type_range (line 156) | def _convert_output_type_range(img, dst_type):
  function rgb2ycbcr_pt (line 186) | def rgb2ycbcr_pt(img, y_only=False):

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 41) | def get_confirm_token(response):
  function save_response_content (line 48) | def save_response_content(response, destination, file_size=None, chunk_s...
  function load_file_from_url (line 69) | def load_file_from_url(url, model_dir=None, progress=True, file_name=None):

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 135) | def read_img_worker(path, key, compress_level):
  class LmdbMaker (line 159) | class LmdbMaker():
    method __init__ (line 170) | def __init__(self, lmdb_path, map_size=1024**4, batch=5000, compress_l...
    method put (line 185) | def put(self, img_byte, key, img_shape):
    method close (line 196) | 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 146) | def get_root_logger(logger_name='basicsr', log_level=logging.INFO, log_f...
  function get_env_info (line 188) | 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):

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 13) | def ordered_yaml():
  function yaml_load (line 38) | def yaml_load(f):
  function dict2str (line 54) | def dict2str(opt, indent_level=1):
  function _postprocess_yml_value (line 75) | def _postprocess_yml_value(value):
  function parse_options (line 99) | def parse_options(root_path, is_train=True):
  function copy_opt_file (line 205) | def copy_opt_file(opt_file, experiments_root):

FILE: basicsr/utils/plot_util.py
  function read_data_from_tensorboard (line 4) | def read_data_from_tensorboard(log_path, tag):
  function read_data_from_txt_2v (line 23) | def read_data_from_txt_2v(path, pattern, step_one=False):
  function read_data_from_txt_1v (line 48) | def read_data_from_txt_1v(path, pattern):
  function smooth_data (line 68) | def smooth_data(values, smooth_weight):

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, suffix=None):
    method register (line 46) | def register(self, obj=None, suffix=None):
    method get (line 65) | def get(self, name, suffix='basicsr'):
    method __contains__ (line 74) | def __contains__(self, name):
    method __iter__ (line 77) | def __iter__(self):
    method keys (line 80) | def keys(self):

FILE: docs/auto_generate_api.py
  function scandir (line 5) | def scandir(dir_path, suffix=None, recursive=False, full_path=False):

FILE: docs/conf.py
  function auto_generate_api (line 78) | def auto_generate_api(app):
  function setup (line 82) | def setup(app):

FILE: inference/inference_basicvsr.py
  function inference (line 13) | def inference(imgs, imgnames, model, save_path):
  function main (line 24) | def main():

FILE: inference/inference_basicvsrpp.py
  function inference (line 13) | def inference(imgs, imgnames, model, save_path):
  function main (line 24) | def main():

FILE: inference/inference_dfdnet.py
  function get_part_location (line 20) | def get_part_location(landmarks):

FILE: inference/inference_esrgan.py
  function main (line 11) | def main():

FILE: inference/inference_stylegan2.py
  function generate (line 11) | def generate(args, g_ema, device, mean_latent, randomize_noise):

FILE: inference/inference_swinir.py
  function main (line 13) | def main():
  function define_model (line 79) | def define_model(args):

FILE: scripts/data_preparation/create_lmdb.py
  function create_lmdb_for_div2k (line 8) | def create_lmdb_for_div2k():
  function prepare_keys_div2k (line 47) | def prepare_keys_div2k(folder_path):
  function create_lmdb_for_reds (line 64) | def create_lmdb_for_reds():
  function prepare_keys_reds (line 89) | def prepare_keys_reds(folder_path):
  function create_lmdb_for_vimeo90k (line 106) | def create_lmdb_for_vimeo90k():
  function prepare_keys_vimeo90k (line 127) | 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 10) | 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/plot/model_complexity_cmp_bsrn.py
  function main (line 1) | def main():

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 12) | def readme():
  function get_git_hash (line 18) | def get_git_hash():
  function get_hash (line 43) | def get_hash():
  function write_version_py (line 59) | def write_version_py():
  function get_version (line 76) | def get_version():
  function make_cuda_ext (line 82) | def make_cuda_ext(name, module, sources, sources_cuda=None):
  function get_requirements (line 108) | def get_requirements(filename='requirements.txt'):

FILE: test_scripts/test_discriminator_backward.py
  class ToyDiscriminator (line 7) | class ToyDiscriminator(nn.Module):
    method __init__ (line 9) | def __init__(self):
    method forward (line 18) | def forward(self, x):
  function main (line 26) | def main():

FILE: test_scripts/test_ffhq_dataset.py
  function main (line 9) | def main():

FILE: test_scripts/test_lr_scheduler.py
  function main (line 15) | def main():

FILE: test_scripts/test_niqe.py
  function main (line 7) | def main():

FILE: test_scripts/test_paired_image_dataset.py
  function main (line 8) | def main(mode='folder'):

FILE: test_scripts/test_reds_dataset.py
  function main (line 8) | def main(mode='folder'):

FILE: test_scripts/test_vimeo90k_dataset.py
  function main (line 8) | def main(mode='folder'):

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():