SYMBOL INDEX (186 symbols across 20 files)

FILE: cog_predict.py
  class Predictor (line 27) | class Predictor(BasePredictor):
    method setup (line 29) | def setup(self):
    method predict (line 66) | def predict(
  function clean_folder (line 152) | def clean_folder(folder):

FILE: gfpgan/archs/arcface_arch.py
  function conv3x3 (line 5) | def conv3x3(inplanes, outplanes, stride=1):
  class BasicBlock (line 16) | class BasicBlock(nn.Module):
    method __init__ (line 27) | def __init__(self, inplanes, planes, stride=1, downsample=None):
    method forward (line 37) | def forward(self, x):
  class IRBlock (line 56) | class IRBlock(nn.Module):
    method __init__ (line 68) | def __init__(self, inplanes, planes, stride=1, downsample=None, use_se...
    method forward (line 82) | def forward(self, x):
  class Bottleneck (line 103) | class Bottleneck(nn.Module):
    method __init__ (line 114) | def __init__(self, inplanes, planes, stride=1, downsample=None):
    method forward (line 126) | def forward(self, x):
  class SEBlock (line 149) | class SEBlock(nn.Module):
    method __init__ (line 157) | def __init__(self, channel, reduction=16):
    method forward (line 164) | def forward(self, x):
  class ResNetArcFace (line 172) | class ResNetArcFace(nn.Module):
    method __init__ (line 183) | def __init__(self, block, layers, use_se=True):
    method _make_layer (line 214) | def _make_layer(self, block, planes, num_blocks, stride=1):
    method forward (line 229) | def forward(self, x):

FILE: gfpgan/archs/gfpgan_bilinear_arch.py
  class StyleGAN2GeneratorBilinearSFT (line 12) | class StyleGAN2GeneratorBilinearSFT(StyleGAN2GeneratorBilinear):
    method __init__ (line 28) | def __init__(self,
    method forward (line 45) | def forward(self,
  class GFPGANBilinear (line 132) | class GFPGANBilinear(nn.Module):
    method __init__ (line 156) | def __init__(
    method forward (line 264) | def forward(self, x, return_latents=False, return_rgb=True, randomize_...

FILE: gfpgan/archs/gfpganv1_arch.py
  class StyleGAN2GeneratorSFT (line 12) | class StyleGAN2GeneratorSFT(StyleGAN2Generator):
    method __init__ (line 27) | def __init__(self,
    method forward (line 46) | def forward(self,
  class ConvUpLayer (line 132) | class ConvUpLayer(nn.Module):
    method __init__ (line 146) | def __init__(self,
    method forward (line 180) | def forward(self, x):
  class ResUpBlock (line 197) | class ResUpBlock(nn.Module):
    method __init__ (line 205) | def __init__(self, in_channels, out_channels):
    method forward (line 212) | def forward(self, x):
  class GFPGANv1 (line 221) | class GFPGANv1(nn.Module):
    method __init__ (line 243) | def __init__(
    method forward (line 353) | def forward(self, x, return_latents=False, return_rgb=True, randomize_...
  class FacialComponentDiscriminator (line 405) | class FacialComponentDiscriminator(nn.Module):
    method __init__ (line 409) | def __init__(self):
    method forward (line 419) | def forward(self, x, return_feats=False, **kwargs):

FILE: gfpgan/archs/gfpganv1_clean_arch.py
  class StyleGAN2GeneratorCSFT (line 11) | class StyleGAN2GeneratorCSFT(StyleGAN2GeneratorClean):
    method __init__ (line 25) | def __init__(self, out_size, num_style_feat=512, num_mlp=8, channel_mu...
    method forward (line 34) | def forward(self,
  class ResBlock (line 120) | class ResBlock(nn.Module):
    method __init__ (line 129) | def __init__(self, in_channels, out_channels, mode='down'):
    method forward (line 140) | def forward(self, x):
  class GFPGANv1Clean (line 153) | class GFPGANv1Clean(nn.Module):
    method __init__ (line 174) | def __init__(
    method forward (line 277) | def forward(self, x, return_latents=False, return_rgb=True, randomize_...

FILE: gfpgan/archs/restoreformer_arch.py
  class VectorQuantizer (line 9) | class VectorQuantizer(nn.Module):
    method __init__ (line 21) | def __init__(self, n_e, e_dim, beta):
    method forward (line 30) | def forward(self, z):
    method get_codebook_entry (line 90) | def get_codebook_entry(self, indices, shape):
  function nonlinearity (line 109) | def nonlinearity(x):
  function Normalize (line 114) | def Normalize(in_channels):
  class Upsample (line 118) | class Upsample(nn.Module):
    method __init__ (line 120) | def __init__(self, in_channels, with_conv):
    method forward (line 126) | def forward(self, x):
  class Downsample (line 133) | class Downsample(nn.Module):
    method __init__ (line 135) | def __init__(self, in_channels, with_conv):
    method forward (line 142) | def forward(self, x):
  class ResnetBlock (line 152) | class ResnetBlock(nn.Module):
    method __init__ (line 154) | def __init__(self, *, in_channels, out_channels=None, conv_shortcut=Fa...
    method forward (line 174) | def forward(self, x, temb):
  class MultiHeadAttnBlock (line 197) | class MultiHeadAttnBlock(nn.Module):
    method __init__ (line 199) | def __init__(self, in_channels, head_size=1):
    method forward (line 215) | def forward(self, x, y=None):
  class MultiHeadEncoder (line 258) | class MultiHeadEncoder(nn.Module):
    method __init__ (line 260) | def __init__(self,
    method forward (line 324) | def forward(self, x):
  class MultiHeadDecoder (line 362) | class MultiHeadDecoder(nn.Module):
    method __init__ (line 364) | def __init__(self,
    method forward (line 432) | def forward(self, z):
  class MultiHeadDecoderTransformer (line 467) | class MultiHeadDecoderTransformer(nn.Module):
    method __init__ (line 469) | def __init__(self,
    method forward (line 537) | def forward(self, z, hs):
  class RestoreFormer (line 573) | class RestoreFormer(nn.Module):
    method __init__ (line 575) | def __init__(self,
    method encode (line 641) | def encode(self, x):
    method decode (line 648) | def decode(self, quant, hs):
    method forward (line 654) | def forward(self, input, **kwargs):

FILE: gfpgan/archs/stylegan2_bilinear_arch.py
  class NormStyleCode (line 10) | class NormStyleCode(nn.Module):
    method forward (line 12) | def forward(self, x):
  class EqualLinear (line 24) | class EqualLinear(nn.Module):
    method __init__ (line 38) | def __init__(self, in_channels, out_channels, bias=True, bias_init_val...
    method forward (line 55) | def forward(self, x):
    method __repr__ (line 67) | def __repr__(self):
  class ModulatedConv2d (line 72) | class ModulatedConv2d(nn.Module):
    method __init__ (line 90) | def __init__(self,
    method forward (line 120) | def forward(self, x, style):
    method __repr__ (line 155) | def __repr__(self):
  class StyleConv (line 162) | class StyleConv(nn.Module):
    method __init__ (line 175) | def __init__(self,
    method forward (line 195) | def forward(self, x, style, noise=None):
  class ToRGB (line 208) | class ToRGB(nn.Module):
    method __init__ (line 217) | def __init__(self, in_channels, num_style_feat, upsample=True, interpo...
    method forward (line 235) | def forward(self, x, style, skip=None):
  class ConstantInput (line 256) | class ConstantInput(nn.Module):
    method __init__ (line 264) | def __init__(self, num_channel, size):
    method forward (line 268) | def forward(self, batch):
  class StyleGAN2GeneratorBilinear (line 274) | class StyleGAN2GeneratorBilinear(nn.Module):
    method __init__ (line 287) | def __init__(self,
    method make_noise (line 369) | def make_noise(self):
    method get_latent (line 380) | def get_latent(self, x):
    method mean_latent (line 383) | def mean_latent(self, num_latent):
    method forward (line 388) | def forward(self,
  class ScaledLeakyReLU (line 465) | class ScaledLeakyReLU(nn.Module):
    method __init__ (line 472) | def __init__(self, negative_slope=0.2):
    method forward (line 476) | def forward(self, x):
  class EqualConv2d (line 481) | class EqualConv2d(nn.Module):
    method __init__ (line 496) | def __init__(self, in_channels, out_channels, kernel_size, stride=1, p...
    method forward (line 511) | def forward(self, x):
    method __repr__ (line 522) | def __repr__(self):
  class ConvLayer (line 530) | class ConvLayer(nn.Sequential):
    method __init__ (line 543) | def __init__(self,
  class ResBlock (line 579) | class ResBlock(nn.Module):
    method __init__ (line 587) | def __init__(self, in_channels, out_channels, interpolation_mode='bili...
    method forward (line 608) | def forward(self, x):

FILE: gfpgan/archs/stylegan2_clean_arch.py
  class NormStyleCode (line 10) | class NormStyleCode(nn.Module):
    method forward (line 12) | def forward(self, x):
  class ModulatedConv2d (line 24) | class ModulatedConv2d(nn.Module):
    method __init__ (line 39) | def __init__(self,
    method forward (line 65) | def forward(self, x, style):
    method __repr__ (line 101) | def __repr__(self):
  class StyleConv (line 106) | class StyleConv(nn.Module):
    method __init__ (line 118) | def __init__(self, in_channels, out_channels, kernel_size, num_style_f...
    method forward (line 126) | def forward(self, x, style, noise=None):
  class ToRGB (line 141) | class ToRGB(nn.Module):
    method __init__ (line 150) | def __init__(self, in_channels, num_style_feat, upsample=True):
    method forward (line 157) | def forward(self, x, style, skip=None):
  class ConstantInput (line 177) | class ConstantInput(nn.Module):
    method __init__ (line 185) | def __init__(self, num_channel, size):
    method forward (line 189) | def forward(self, batch):
  class StyleGAN2GeneratorClean (line 195) | class StyleGAN2GeneratorClean(nn.Module):
    method __init__ (line 206) | def __init__(self, out_size, num_style_feat=512, num_mlp=8, channel_mu...
    method make_noise (line 279) | def make_noise(self):
    method get_latent (line 290) | def get_latent(self, x):
    method mean_latent (line 293) | def mean_latent(self, num_latent):
    method forward (line 298) | def forward(self,

FILE: gfpgan/data/ffhq_degradation_dataset.py
  class FFHQDegradationDataset (line 17) | class FFHQDegradationDataset(data.Dataset):
    method __init__ (line 32) | def __init__(self, opt):
    method color_jitter (line 91) | def color_jitter(img, shift):
    method color_jitter_pt (line 99) | def color_jitter_pt(img, brightness, contrast, saturation, hue):
    method get_component_coordinates (line 120) | def get_component_coordinates(self, index, status):
    method __getitem__ (line 145) | def __getitem__(self, index):
    method __len__ (line 229) | def __len__(self):

FILE: gfpgan/models/gfpgan_model.py
  class GFPGANModel (line 18) | class GFPGANModel(BaseModel):
    method __init__ (line 21) | def __init__(self, opt):
    method init_training_settings (line 41) | def init_training_settings(self):
    method setup_optimizers (line 150) | def setup_optimizers(self):
    method feed_data (line 201) | def feed_data(self, data):
    method construct_img_pyramid (line 225) | def construct_img_pyramid(self):
    method get_roi_regions (line 234) | def get_roi_regions(self, eye_out_size=80, mouth_out_size=120):
    method _gram_mat (line 266) | def _gram_mat(self, x):
    method gray_resize_for_identity (line 281) | def gray_resize_for_identity(self, out, size=128):
    method optimize_parameters (line 287) | def optimize_parameters(self, current_iter):
    method test (line 473) | def test(self):
    method dist_validation (line 485) | def dist_validation(self, dataloader, current_iter, tb_logger, save_img):
    method nondist_validation (line 489) | def nondist_validation(self, dataloader, current_iter, tb_logger, save...
    method _log_validation_metric_values (line 554) | def _log_validation_metric_values(self, current_iter, dataset_name, tb...
    method save (line 569) | def save(self, epoch, current_iter):

FILE: gfpgan/utils.py
  class GFPGANer (line 16) | class GFPGANer():
    method __init__ (line 32) | def __init__(self, model_path, upscale=2, arch='clean', channel_multip...
    method enhance (line 102) | def enhance(self, img, has_aligned=False, only_center_face=False, past...

FILE: inference_gfpgan.py
  function main (line 12) | def main():

FILE: scripts/convert_gfpganv_to_clean.py
  function modify_checkpoint (line 8) | def modify_checkpoint(checkpoint_bilinear, checkpoint_clean):

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 52) | def write_version_py():
  function get_version (line 69) | def get_version():
  function get_requirements (line 75) | def get_requirements(filename='requirements.txt'):

FILE: tests/test_arcface_arch.py
  function test_resnetarcface (line 6) | def test_resnetarcface():
  function test_basicblock (line 22) | def test_basicblock():
  function test_bottleneck (line 37) | def test_bottleneck():

FILE: tests/test_ffhq_degradation_dataset.py
  function test_ffhq_degradation_dataset (line 7) | def test_ffhq_degradation_dataset():

FILE: tests/test_gfpgan_arch.py
  function test_stylegan2generatorsft (line 7) | def test_stylegan2generatorsft():
  function test_gfpganv1 (line 48) | def test_gfpganv1():
  function test_facialcomponentdiscriminator (line 101) | def test_facialcomponentdiscriminator():
  function test_stylegan2generatorcsft (line 122) | def test_stylegan2generatorcsft():
  function test_gfpganv1clean (line 156) | def test_gfpganv1clean():

FILE: tests/test_gfpgan_model.py
  function test_gfpgan_model (line 13) | def test_gfpgan_model():

FILE: tests/test_stylegan2_clean_arch.py
  function test_stylegan2generatorclean (line 6) | def test_stylegan2generatorclean():

FILE: tests/test_utils.py
  function test_gfpganer (line 9) | def test_gfpganer():