SYMBOL INDEX (179 symbols across 29 files)

FILE: datasets/UPS_Custom_Dataset.py
  class UPS_Custom_Dataset (line 14) | class UPS_Custom_Dataset(data.Dataset):
    method __init__ (line 15) | def __init__(self, args, split='train'):
    method _getMask (line 22) | def _getMask(self, obj):
    method __getitem__ (line 28) | def __getitem__(self, index):
    method __len__ (line 75) | def __len__(self):

FILE: datasets/UPS_DiLiGenT_main.py
  class UPS_DiLiGenT_main (line 15) | class UPS_DiLiGenT_main(data.Dataset):
    method __init__ (line 16) | def __init__(self, args, split='train'):
    method _getMask (line 31) | def _getMask(self, obj):
    method __getitem__ (line 37) | def __getitem__(self, index):
    method __len__ (line 83) | def __len__(self):

FILE: datasets/UPS_Synth_Dataset.py
  class UPS_Synth_Dataset (line 14) | class UPS_Synth_Dataset(data.Dataset):
    method __init__ (line 15) | def __init__(self, args, root, split='train'):
    method _getInputPath (line 21) | def _getInputPath(self, index):
    method __getitem__ (line 35) | def __getitem__(self, index):
    method __len__ (line 79) | def __len__(self):

FILE: datasets/custom_data_loader.py
  function customDataloader (line 3) | def customDataloader(args):
  function benchmarkLoader (line 27) | def benchmarkLoader(args):

FILE: datasets/pms_transforms.py
  function arrayToTensor (line 8) | def arrayToTensor(array):
  function normalToMask (line 15) | def normalToMask(normal, thres=1e-2):
  function imgSizeToFactorOfK (line 24) | def imgSizeToFactorOfK(img, k):
  function randomCrop (line 32) | def randomCrop(inputs, target, size):
  function centerCrop (line 43) | def centerCrop(inputs, size):
  function rescale (line 52) | def rescale(inputs, target, size):
  function rescaleSingle (line 60) | def rescaleSingle(inputs, size, order=1):
  function randomNoiseAug (line 67) | def randomNoiseAug(inputs, noise_level=0.05):
  function getIntensity (line 73) | def getIntensity(num):

FILE: datasets/util.py
  function atoi (line 4) | def atoi(text):
  function natural_keys (line 7) | def natural_keys(text):
  function readList (line 15) | def readList(list_path,ignore_head=False, sort=True):
  function light_source_directions (line 25) | def light_source_directions():

FILE: eval/run_stage1.py
  function main (line 14) | def main(args):

FILE: eval/run_stage2.py
  function main (line 16) | def main(args):

FILE: main_stage1.py
  function main (line 13) | def main(args):

FILE: main_stage2.py
  function main (line 13) | def main(args):

FILE: models/LCNet.py
  class FeatExtractor (line 8) | class FeatExtractor(nn.Module):
    method __init__ (line 9) | def __init__(self, batchNorm, c_in, c_out=256):
    method forward (line 19) | def forward(self, inputs):
  class Classifier (line 29) | class Classifier(nn.Module):
    method __init__ (line 30) | def __init__(self, batchNorm, c_in, other):
    method forward (line 50) | def forward(self, inputs):
  class LCNet (line 63) | class LCNet(nn.Module):
    method __init__ (line 64) | def __init__(self, fuse_type='max', batchNorm=False, c_in=3, other={}):
    method prepareInputs (line 81) | def prepareInputs(self, x):
    method fuseFeatures (line 106) | def fuseFeatures(self, feats, fuse_type):
    method convertMidDirs (line 113) | def convertMidDirs(self, pred):
    method convertMidIntens (line 119) | def convertMidIntens(self, pred, img_num):
    method forward (line 126) | def forward(self, x):

FILE: models/NENet.py
  class FeatExtractor (line 6) | class FeatExtractor(nn.Module):
    method __init__ (line 7) | def __init__(self, batchNorm=False, c_in=3, other={}):
    method forward (line 18) | def forward(self, x):
  class Regressor (line 30) | class Regressor(nn.Module):
    method __init__ (line 31) | def __init__(self, batchNorm=False, other={}):
    method _make_output (line 40) | def _make_output(self, cin, cout, k=3, stride=1, pad=1):
    method forward (line 44) | def forward(self, x, shape):
  class NENet (line 53) | class NENet(nn.Module):
    method __init__ (line 54) | def __init__(self, fuse_type='max', batchNorm=False, c_in=3, other={}):
    method prepareInputs (line 71) | def prepareInputs(self, x):
    method forward (line 90) | def forward(self, x):

FILE: models/custom_model.py
  function buildModel (line 4) | def buildModel(args):
  function buildModelStage2 (line 31) | def buildModelStage2(args):

FILE: models/model_utils.py
  function getInput (line 5) | def getInput(args, data):
  function parseData (line 11) | def parseData(args, sample, timer=None, split='train'):
  function getInputChanel (line 28) | def getInputChanel(args):
  function get_n_params (line 40) | def get_n_params(model):
  function loadCheckpoint (line 49) | def loadCheckpoint(path, model, cuda=True):
  function saveCheckpoint (line 56) | def saveCheckpoint(save_path, epoch=-1, model=None, optimizer=None, reco...
  function conv_ReLU (line 62) | def conv_ReLU(batchNorm, cin, cout, k=3, stride=1, pad=-1):
  function conv (line 77) | def conv(batchNorm, cin, cout, k=3, stride=1, pad=-1):
  function outputConv (line 92) | def outputConv(cin, cout, k=3, stride=1, pad=1):
  function deconv (line 96) | def deconv(cin, cout):
  function upconv (line 102) | def upconv(cin, cout):

FILE: models/solver_utils.py
  class Stage1ClsCrit (line 5) | class Stage1ClsCrit(object): # First Stage, Light classification criterion
    method __init__ (line 6) | def __init__(self, args):
    method setupLightCrit (line 13) | def setupLightCrit(self, args):
    method forward (line 27) | def forward(self, output, target):
    method backward (line 51) | def backward(self):
  class Stage2Crit (line 54) | class Stage2Crit(object): # Second stage
    method __init__ (line 55) | def __init__(self, args):
    method setupLightCrit (line 63) | def setupLightCrit(self, args):
    method setupNormalCrit (line 74) | def setupNormalCrit(self, args):
    method forward (line 86) | def forward(self, output, target):
    method backward (line 117) | def backward(self):
  function getOptimizer (line 120) | def getOptimizer(args, params):
  function getLrScheduler (line 130) | def getLrScheduler(args, optimizer):
  function loadRecords (line 135) | def loadRecords(path, model, optimizer):
  function configOptimizer (line 147) | def configOptimizer(args, model):

FILE: options/base_opts.py
  class BaseOpts (line 5) | class BaseOpts(object):
    method __init__ (line 6) | def __init__(self):
    method initialize (line 9) | def initialize(self):
    method setDefault (line 88) | def setDefault(self):
    method collectInfo (line 96) | def collectInfo(self):
    method parse (line 109) | def parse(self):

FILE: options/run_model_opts.py
  class RunModelOpts (line 2) | class RunModelOpts(BaseOpts):
    method __init__ (line 3) | def __init__(self):
    method initialize (line 7) | def initialize(self):
    method collectInfo (line 23) | def collectInfo(self):
    method setDefault (line 28) | def setDefault(self):
    method parse (line 31) | def parse(self):

FILE: options/stage1_opts.py
  class TrainOpts (line 2) | class TrainOpts(BaseOpts):
    method __init__ (line 3) | def __init__(self):
    method initialize (line 7) | def initialize(self):
    method collectInfo (line 31) | def collectInfo(self):
    method setDefault (line 41) | def setDefault(self):
    method parse (line 47) | def parse(self):

FILE: options/stage2_opts.py
  class TrainOpts (line 2) | class TrainOpts(BaseOpts):
    method __init__ (line 3) | def __init__(self):
    method initialize (line 7) | def initialize(self):
    method collectInfo (line 31) | def collectInfo(self):
    method setDefault (line 41) | def setDefault(self):
    method parse (line 47) | def parse(self):

FILE: scripts/cropDiLiGenTData.py
  function getSaveDir (line 21) | def getSaveDir():
  function getBBoxCompact (line 28) | def getBBoxCompact(mask):
  function loadMaskNormal (line 44) | def loadMaskNormal(d):
  function copyTXT (line 57) | def copyTXT(d):

FILE: test_stage1.py
  function get_itervals (line 7) | def get_itervals(args, split):
  function test (line 16) | def test(args, split, loader, model, log, epoch, recorder):
  function prepareRes (line 51) | def prepareRes(args, data, pred, recorder, log, split):
  function prepareSave (line 76) | def prepareSave(args, data, pred):

FILE: test_stage2.py
  function get_itervals (line 7) | def get_itervals(args, split):
  function test (line 16) | def test(args, split, loader, models, log, epoch, recorder):
  function prepareRes (line 59) | def prepareRes(args, data, pred_c, pred, recorder, log, split):
  function prepareSave (line 82) | def prepareSave(args, data, pred_c, pred):

FILE: train_stage1.py
  function train (line 4) | def train(args, loader, model, criterion, optimizer, log, epoch, recorder):
  function prepareSave (line 39) | def prepareSave(args, data, pred, recorder, log):

FILE: train_stage2.py
  function train (line 5) | def train(args, loader, models, criterion, optimizers, log, epoch, recor...
  function prepareSave (line 45) | def prepareSave(args, data, pred_c, pred, recorder, log):

FILE: utils/eval_utils.py
  function colorMap (line 6) | def colorMap(diff):
  function calDirsAcc (line 12) | def calDirsAcc(gt_l, pred_l, data_batch=1):
  function calIntsAcc (line 21) | def calIntsAcc(gt_i, pred_i, data_batch=1):
  function calNormalAcc (line 31) | def calNormalAcc(gt_n, pred_n, mask=None):
  function SphericalDirsToClass (line 52) | def SphericalDirsToClass(dirs, cls_num):
  function SphericalClassToDirs (line 62) | def SphericalClassToDirs(x_cls, y_cls, cls_num):
  function LightIntsToClass (line 81) | def LightIntsToClass(ints, cls_num):
  function ClassToLightInts (line 86) | def ClassToLightInts(cls, cls_num):

FILE: utils/logger.py
  class Logger (line 16) | class Logger(object):
    method __init__ (line 17) | def __init__(self, args):
    method printArgs (line 25) | def printArgs(self):
    method _addArguments (line 31) | def _addArguments(self, args):
    method _setupDirs (line 45) | def _setupDirs(self, args):
    method _checkPath (line 57) | def _checkPath(self, args, dir_name):
    method printWrite (line 76) | def printWrite(self, strs):
    method getTimeInfo (line 82) | def getTimeInfo(self, epoch, iters, batch):
    method printItersSummary (line 89) | def printItersSummary(self, opt):
    method printEpochSummary (line 103) | def printEpochSummary(self, opt):
    method convertToSameSize (line 109) | def convertToSameSize(self, t_list):
    method getSaveDir (line 120) | def getSaveDir(self, split, epoch):
    method splitMulitChannel (line 128) | def splitMulitChannel(self, t_list, max_save_n = 8):
    method saveSplit (line 138) | def saveSplit(self, res, save_prefix):
    method saveImgResults (line 143) | def saveImgResults(self, results, split, epoch, iters, nrow, error=''):
    method plotCurves (line 156) | def plotCurves(self, recorder, split='train', epoch=-1, intv=1):

FILE: utils/recorders.py
  class Records (line 4) | class Records(object):
    method __init__ (line 9) | def __init__(self, log_dir, records=None):
    method resetIter (line 18) | def resetIter(self):
    method checkDict (line 21) | def checkDict(self, a_dict, key, sub_type='dict'):
    method updateIter (line 28) | def updateIter(self, split, keys, values):
    method saveIterRecord (line 34) | def saveIterRecord(self, epoch, reset=True):
    method insertRecord (line 44) | def insertRecord(self, split, key, epoch, value):
    method iterRecToString (line 50) | def iterRecToString(self, split, epoch):
    method epochRecToString (line 62) | def epochRecToString(self, split, epoch):
    method recordToDictOfArray (line 73) | def recordToDictOfArray(self, splits, epoch=-1, intv=1):

FILE: utils/time_utils.py
  class Timer (line 5) | class Timer(object):
    method __init__ (line 6) | def __init__(self, cuda_sync=False):
    method startTimer (line 11) | def startTimer(self):
    method resetTimer (line 15) | def resetTimer(self):
    method updateTime (line 20) | def updateTime(self, key):
    method timeToString (line 26) | def timeToString(self, reset=True):
  class AverageMeter (line 34) | class AverageMeter(object):
    method __init__ (line 35) | def __init__(self):
    method reset (line 38) | def reset(self):
    method update (line 43) | def update(self, val, n=1):
    method __repr__ (line 48) | def __repr__(self):

FILE: utils/utils.py
  function makeFile (line 5) | def makeFile(f):
  function makeFiles (line 10) | def makeFiles(f_list):
  function emptyFile (line 14) | def emptyFile(name):
  function dictToString (line 18) | def dictToString(dicts, start='\t', end='\n'):
  function checkIfInList (line 24) | def checkIfInList(list1, list2):
  function atoi (line 33) | def atoi(text):
  function natural_keys (line 36) | def natural_keys(text):
  function readList (line 44) | def readList(list_path,ignore_head=False, sort=False):