SYMBOL INDEX (24 symbols across 4 files)

FILE: datasets.py
  class TrainDataset (line 6) | class TrainDataset(Dataset):
    method __init__ (line 7) | def __init__(self, h5_file):
    method __getitem__ (line 11) | def __getitem__(self, idx):
    method __len__ (line 15) | def __len__(self):
  class EvalDataset (line 20) | class EvalDataset(Dataset):
    method __init__ (line 21) | def __init__(self, h5_file):
    method __getitem__ (line 25) | def __getitem__(self, idx):
    method __len__ (line 29) | def __len__(self):

FILE: models.py
  class FSRCNN (line 5) | class FSRCNN(nn.Module):
    method __init__ (line 6) | def __init__(self, scale_factor, num_channels=1, d=56, s=12, m=4):
    method _initialize_weights (line 22) | def _initialize_weights(self):
    method forward (line 34) | def forward(self, x):

FILE: prepare.py
  function train (line 10) | def train(args):
  function eval (line 53) | def eval(args):

FILE: utils.py
  function calc_patch_size (line 5) | def calc_patch_size(func):
  function convert_rgb_to_y (line 19) | def convert_rgb_to_y(img, dim_order='hwc'):
  function convert_rgb_to_ycbcr (line 26) | def convert_rgb_to_ycbcr(img, dim_order='hwc'):
  function convert_ycbcr_to_rgb (line 38) | def convert_ycbcr_to_rgb(img, dim_order='hwc'):
  function preprocess (line 50) | def preprocess(img, device):
  function calc_psnr (line 60) | def calc_psnr(img1, img2):
  class AverageMeter (line 64) | class AverageMeter(object):
    method __init__ (line 65) | def __init__(self):
    method reset (line 68) | def reset(self):
    method update (line 74) | def update(self, val, n=1):