SYMBOL INDEX (207 symbols across 33 files)

FILE: src/data/__init__.py
  class MyConcatDataset (line 7) | class MyConcatDataset(ConcatDataset):
    method __init__ (line 8) | def __init__(self, datasets):
    method set_scale (line 12) | def set_scale(self, idx_scale):
  class Data (line 16) | class Data:
    method __init__ (line 17) | def __init__(self, args):

FILE: src/data/benchmark.py
  class Benchmark (line 11) | class Benchmark(srdata.SRData):
    method __init__ (line 12) | def __init__(self, args, name='', train=True, benchmark=True):
    method _set_filesystem (line 17) | def _set_filesystem(self, dir_data):

FILE: src/data/common.py
  function get_patch (line 8) | def get_patch(*args, patch_size=96, scale=2, multi=False, input_large=Fa...
  function set_channel (line 34) | def set_channel(*args, n_channels=3):
  function np2Tensor (line 49) | def np2Tensor(*args, rgb_range=255):
  function augment (line 59) | def augment(*args, hflip=True, rot=True):

FILE: src/data/demo.py
  class Demo (line 11) | class Demo(data.Dataset):
    method __init__ (line 12) | def __init__(self, args, name='Demo', train=False, benchmark=False):
    method __getitem__ (line 26) | def __getitem__(self, idx):
    method __len__ (line 34) | def __len__(self):
    method set_scale (line 37) | def set_scale(self, idx_scale):

FILE: src/data/div2k.py
  class DIV2K (line 4) | class DIV2K(srdata.SRData):
    method __init__ (line 5) | def __init__(self, args, name='DIV2K', train=True, benchmark=False):
    method _scan (line 20) | def _scan(self):
    method _set_filesystem (line 27) | def _set_filesystem(self, dir_data):

FILE: src/data/div2kjpeg.py
  class DIV2KJPEG (line 5) | class DIV2KJPEG(div2k.DIV2K):
    method __init__ (line 6) | def __init__(self, args, name='', train=True, benchmark=False):
    method _set_filesystem (line 12) | def _set_filesystem(self, dir_data):

FILE: src/data/sr291.py
  class SR291 (line 3) | class SR291(srdata.SRData):
    method __init__ (line 4) | def __init__(self, args, name='SR291', train=True, benchmark=False):

FILE: src/data/srdata.py
  class SRData (line 13) | class SRData(data.Dataset):
    method __init__ (line 14) | def __init__(self, args, name='', train=True, benchmark=False):
    method _scan (line 68) | def _scan(self):
    method _set_filesystem (line 84) | def _set_filesystem(self, dir_data):
    method _check_and_load (line 91) | def _check_and_load(self, ext, img, f, verbose=True):
    method __getitem__ (line 98) | def __getitem__(self, idx):
    method __len__ (line 106) | def __len__(self):
    method _get_index (line 112) | def _get_index(self, idx):
    method _load_file (line 118) | def _load_file(self, idx):
    method get_patch (line 135) | def get_patch(self, lr, hr):
    method set_scale (line 152) | def set_scale(self, idx_scale):

FILE: src/data/video.py
  class Video (line 12) | class Video(data.Dataset):
    method __init__ (line 13) | def __init__(self, args, name='Video', train=False, benchmark=False):
    method __getitem__ (line 27) | def __getitem__(self, idx):
    method __len__ (line 39) | def __len__(self):
    method set_scale (line 42) | def set_scale(self, idx_scale):

FILE: src/dataloader.py
  function _ms_loop (line 22) | def _ms_loop(dataset, index_queue, data_queue, done_event, collate_fn, s...
  class _MSDataLoaderIter (line 68) | class _MSDataLoaderIter(_DataLoaderIter):
    method __init__ (line 70) | def __init__(self, loader):
  class MSDataLoader (line 148) | class MSDataLoader(DataLoader):
    method __init__ (line 150) | def __init__(self, cfg, *args, **kwargs):
    method __iter__ (line 156) | def __iter__(self):

FILE: src/loss/__init__.py
  class Loss (line 14) | class Loss(nn.modules.loss._Loss):
    method __init__ (line 15) | def __init__(self, args, ckp):
    method forward (line 67) | def forward(self, sr, hr):
    method step (line 84) | def step(self):
    method start_log (line 89) | def start_log(self):
    method end_log (line 92) | def end_log(self, n_batches):
    method display_loss (line 95) | def display_loss(self, batch):
    method plot_loss (line 103) | def plot_loss(self, apath, epoch):
    method get_loss_module (line 117) | def get_loss_module(self):
    method save (line 123) | def save(self, apath):
    method load (line 127) | def load(self, apath, cpu=False):

FILE: src/loss/adversarial.py
  class Adversarial (line 12) | class Adversarial(nn.Module):
    method __init__ (line 13) | def __init__(self, args, gan_type):
    method forward (line 35) | def forward(self, fake, real):
    method state_dict (line 95) | def state_dict(self, *args, **kwargs):
    method bce (line 101) | def bce(self, real, fake):

FILE: src/loss/discriminator.py
  class Discriminator (line 5) | class Discriminator(nn.Module):
    method __init__ (line 9) | def __init__(self, args):
    method forward (line 50) | def forward(self, x):

FILE: src/loss/hash.py
  class HASH (line 8) | class HASH(nn.Module):
    method __init__ (line 9) | def __init__(self):
    method forward (line 12) | def forward(self, sr, qk, orders, hr, m=3):

FILE: src/loss/vgg.py
  class VGG (line 8) | class VGG(nn.Module):
    method __init__ (line 9) | def __init__(self, conv_index, rgb_range=1):
    method forward (line 24) | def forward(self, sr, hr):

FILE: src/main.py
  function main (line 13) | def main():

FILE: src/model/__init__.py
  class Model (line 8) | class Model(nn.Module):
    method __init__ (line 9) | def __init__(self, args, ckp):
    method forward (line 38) | def forward(self, x, idx_scale):
    method get_model (line 56) | def get_model(self):
    method state_dict (line 62) | def state_dict(self, **kwargs):
    method save (line 66) | def save(self, apath, epoch, is_best=False):
    method load (line 84) | def load(self, apath, pre_train='.', resume=-1, cpu=False):
    method forward_chop (line 114) | def forward_chop(self, x, shave=10, min_size=120000):
    method forward_x8 (line 158) | def forward_x8(self, x, forward_function):

FILE: src/model/attention.py
  class NonLocalSparseAttention (line 6) | class NonLocalSparseAttention(nn.Module):
    method __init__ (line 7) | def __init__( self, n_hashes=4, channels=64, k_size=3, reduction=4, ch...
    method LSH (line 16) | def LSH(self, hash_buckets, x):
    method add_adjacent_buckets (line 39) | def add_adjacent_buckets(self, x):
    method forward (line 44) | def forward(self, input):
  class NonLocalAttention (line 117) | class NonLocalAttention(nn.Module):
    method __init__ (line 118) | def __init__(self, channel=128, reduction=2, ksize=1, scale=3, stride=...
    method forward (line 125) | def forward(self, input):

FILE: src/model/common.py
  function batched_index_select (line 8) | def batched_index_select(values, indices):
  function default_conv (line 12) | def default_conv(in_channels, out_channels, kernel_size,stride=1, bias=T...
  class MeanShift (line 17) | class MeanShift(nn.Conv2d):
    method __init__ (line 18) | def __init__(
  class BasicBlock (line 29) | class BasicBlock(nn.Sequential):
    method __init__ (line 30) | def __init__(
  class ResBlock (line 42) | class ResBlock(nn.Module):
    method __init__ (line 43) | def __init__(
    method forward (line 59) | def forward(self, x):
  class Upsampler (line 65) | class Upsampler(nn.Sequential):
    method __init__ (line 66) | def __init__(self, conv, scale, n_feats, bn=False, act=False, bias=True):

FILE: src/model/ddbpn.py
  function make_model (line 10) | def make_model(args, parent=False):
  function projection_conv (line 13) | def projection_conv(in_channels, out_channels, scale, up=True):
  class DenseProjection (line 29) | class DenseProjection(nn.Module):
    method __init__ (line 30) | def __init__(self, in_channels, nr, scale, up=True, bottleneck=True):
    method forward (line 55) | def forward(self, x):
  class DDBPN (line 68) | class DDBPN(nn.Module):
    method __init__ (line 69) | def __init__(self, args):
    method forward (line 112) | def forward(self, x):

FILE: src/model/edsr.py
  function make_model (line 5) | def make_model(args, parent=False):
  class EDSR (line 12) | class EDSR(nn.Module):
    method __init__ (line 13) | def __init__(self, args, conv=common.default_conv):
    method forward (line 57) | def forward(self, x):
    method load_state_dict (line 69) | def load_state_dict(self, state_dict, strict=True):

FILE: src/model/mdsr.py
  function make_model (line 5) | def make_model(args, parent=False):
  class MDSR (line 8) | class MDSR(nn.Module):
    method __init__ (line 9) | def __init__(self, args, conv=common.default_conv):
    method forward (line 52) | def forward(self, x):
    method set_scale (line 66) | def set_scale(self, scale_idx):

FILE: src/model/mssr.py
  function make_model (line 5) | def make_model(args, parent=False):
  class MultisourceProjection (line 8) | class MultisourceProjection(nn.Module):
    method __init__ (line 9) | def __init__(self, in_channel,kernel_size = 3, conv=common.default_conv):
    method forward (line 16) | def forward(self,x):
  class RecurrentProjection (line 25) | class RecurrentProjection(nn.Module):
    method __init__ (line 26) | def __init__(self, in_channel,kernel_size = 3, conv=common.default_conv):
    method forward (line 39) | def forward(self, x):
  class MSSR (line 58) | class MSSR(nn.Module):
    method __init__ (line 59) | def __init__(self, args, conv=common.default_conv):
    method forward (line 94) | def forward(self,input):

FILE: src/model/nlsn.py
  function make_model (line 5) | def make_model(args, parent=False):
  class NLSN (line 13) | class NLSN(nn.Module):
    method __init__ (line 14) | def __init__(self, args, conv=common.default_conv):
    method forward (line 56) | def forward(self, x):
    method load_state_dict (line 68) | def load_state_dict(self, state_dict, strict=True):

FILE: src/model/rcan.py
  function make_model (line 7) | def make_model(args, parent=False):
  class CALayer (line 11) | class CALayer(nn.Module):
    method __init__ (line 12) | def __init__(self, channel, reduction=16):
    method forward (line 27) | def forward(self, x):
  class RCAB (line 33) | class RCAB(nn.Module):
    method __init__ (line 34) | def __init__(
    method forward (line 48) | def forward(self, x):
  class ResidualGroup (line 55) | class ResidualGroup(nn.Module):
    method __init__ (line 56) | def __init__(self, conv, n_feat, kernel_size, reduction, act, res_scal...
    method forward (line 66) | def forward(self, x):
  class RCAN (line 72) | class RCAN(nn.Module):
    method __init__ (line 73) | def __init__(self, args, conv=common.default_conv):
    method forward (line 111) | def forward(self, x):
    method load_state_dict (line 122) | def load_state_dict(self, state_dict, strict=False):

FILE: src/model/rdn.py
  function make_model (line 10) | def make_model(args, parent=False):
  class RDB_Conv (line 13) | class RDB_Conv(nn.Module):
    method __init__ (line 14) | def __init__(self, inChannels, growRate, kSize=3):
    method forward (line 23) | def forward(self, x):
  class RDB (line 27) | class RDB(nn.Module):
    method __init__ (line 28) | def __init__(self, growRate0, growRate, nConvLayers, kSize=3):
    method forward (line 42) | def forward(self, x):
  class RDN (line 45) | class RDN(nn.Module):
    method __init__ (line 46) | def __init__(self, args):
    method forward (line 93) | def forward(self, x):

FILE: src/model/utils/tools.py
  function normalize (line 8) | def normalize(x):
  function same_padding (line 11) | def same_padding(images, ksizes, strides, rates):
  function extract_image_patches (line 30) | def extract_image_patches(images, ksizes, strides, rates, padding='same'):
  function reduce_mean (line 59) | def reduce_mean(x, axis=None, keepdim=False):
  function reduce_std (line 67) | def reduce_std(x, axis=None, keepdim=False):
  function reduce_sum (line 75) | def reduce_sum(x, axis=None, keepdim=False):

FILE: src/model/vdsr.py
  function make_model (line 10) | def make_model(args, parent=False):
  class VDSR (line 13) | class VDSR(nn.Module):
    method __init__ (line 14) | def __init__(self, args, conv=common.default_conv):
    method forward (line 39) | def forward(self, x):

FILE: src/template.py
  function set_template (line 1) | def set_template(args):

FILE: src/trainer.py
  class Trainer (line 11) | class Trainer():
    method __init__ (line 12) | def __init__(self, args, loader, my_model, my_loss, ckp):
    method train (line 28) | def train(self):
    method test (line 74) | def test(self):
    method prepare (line 131) | def prepare(self, *args):
    method terminate (line 139) | def terminate(self):

FILE: src/utility.py
  class timer (line 19) | class timer():
    method __init__ (line 20) | def __init__(self):
    method tic (line 24) | def tic(self):
    method toc (line 27) | def toc(self, restart=False):
    method hold (line 32) | def hold(self):
    method release (line 35) | def release(self):
    method reset (line 41) | def reset(self):
  class checkpoint (line 44) | class checkpoint():
    method __init__ (line 45) | def __init__(self, args):
    method get_path (line 82) | def get_path(self, *subdir):
    method save (line 85) | def save(self, trainer, epoch, is_best=False):
    method add_log (line 94) | def add_log(self, log):
    method write_log (line 97) | def write_log(self, log, refresh=False):
    method done (line 104) | def done(self):
    method plot_psnr (line 107) | def plot_psnr(self, epoch):
    method begin_background (line 126) | def begin_background(self):
    method end_background (line 143) | def end_background(self):
    method save_results (line 148) | def save_results(self, dataset, filename, save_list, scale):
  function quantize (line 161) | def quantize(img, rgb_range):
  function calc_psnr (line 165) | def calc_psnr(sr, hr, scale, rgb_range, dataset=None):
  function make_optimizer (line 183) | def make_optimizer(args, target):

FILE: src/utils/tools.py
  function normalize (line 8) | def normalize(x):
  function same_padding (line 11) | def same_padding(images, ksizes, strides, rates):
  function extract_image_patches (line 30) | def extract_image_patches(images, ksizes, strides, rates, padding='same'):
  function reduce_mean (line 59) | def reduce_mean(x, axis=None, keepdim=False):
  function reduce_std (line 67) | def reduce_std(x, axis=None, keepdim=False):
  function reduce_sum (line 75) | def reduce_sum(x, axis=None, keepdim=False):

FILE: src/videotester.py
  class VideoTester (line 12) | class VideoTester():
    method __init__ (line 13) | def __init__(self, args, my_model, ckp):
    method test (line 22) | def test(self):
    method prepare (line 65) | def prepare(self, *args):