SYMBOL INDEX (231 symbols across 32 files)

FILE: config_utils/decode_args.py
  function obtain_decode_args (line 3) | def obtain_decode_args():

FILE: config_utils/predict_args.py
  function obtain_predict_args (line 3) | def obtain_predict_args():

FILE: config_utils/search_args.py
  function obtain_search_args (line 3) | def obtain_search_args():

FILE: config_utils/train_args.py
  function obtain_train_args (line 3) | def obtain_train_args():

FILE: dataloaders/__init__.py
  function make_data_loader (line 5) | def make_data_loader(args, **kwargs):

FILE: dataloaders/datasets/stereo.py
  function readPFM (line 14) | def readPFM(file):
  function train_transform (line 48) | def train_transform(temp_data, crop_height, crop_width, left_right=False...
  function test_transform (line 94) | def test_transform(temp_data, crop_height, crop_width, left_right=False):
  function load_data_sceneflow (line 114) | def load_data_sceneflow(data_path, current_file):
  function load_kitti2015_data (line 147) | def load_kitti2015_data(file_path, current_file):
  function load_kitti2012_data (line 186) | def load_kitti2012_data(file_path, current_file):
  function load_data_md (line 226) | def load_data_md(file_path, current_file, eth=False):
  class DatasetFromList (line 264) | class DatasetFromList(data.Dataset):
    method __init__ (line 265) | def __init__(self, args, file_list, crop_size=[256, 256], training=Tru...
    method __getitem__ (line 276) | def __getitem__(self, index):
    method __len__ (line 294) | def __len__(self):

FILE: decode.py
  class Loader (line 8) | class Loader(object):
    method __init__ (line 9) | def __init__(self, args):
    method retreive_alphas_betas (line 27) | def retreive_alphas_betas(self):
    method decode_architecture (line 30) | def decode_architecture(self):
    method decode_cell (line 35) | def decode_cell(self):
  function get_new_network_cell (line 40) | def get_new_network_cell():

FILE: models/build_model.py
  class AutoStereo (line 9) | class AutoStereo(nn.Module):
    method __init__ (line 10) | def __init__(self, maxdisp=192, Fea_Layers=6, Fea_Filter=8, Fea_Block=...
    method forward (line 40) | def forward(self, x, y):

FILE: models/build_model_2d.py
  class DispEntropy (line 9) | class DispEntropy(nn.Module):
    method __init__ (line 10) | def __init__(self, maxdisp):
    method forward (line 15) | def forward(self, x):
  class DisparityRegression (line 24) | class DisparityRegression(nn.Module):
    method __init__ (line 25) | def __init__(self, maxdisp):
    method forward (line 29) | def forward(self, x):
  class Disp (line 37) | class Disp(nn.Module):
    method __init__ (line 38) | def __init__(self, maxdisp=192):
    method forward (line 44) | def forward(self, x):
  class AutoFeature (line 52) | class AutoFeature(nn.Module):
    method __init__ (line 53) | def __init__(self, num_layers, filter_multiplier=8, block_multiplier=4...
    method forward (line 172) | def forward(self, x):
    method _initialize_alphas_betas (line 414) | def _initialize_alphas_betas(self):
    method arch_parameters (line 432) | def arch_parameters(self):
    method weight_parameters (line 435) | def weight_parameters(self):
    method genotype (line 438) | def genotype(self):

FILE: models/build_model_3d.py
  class AutoMatching (line 9) | class AutoMatching(nn.Module):
    method __init__ (line 10) | def __init__(self, num_layers, filter_multiplier=8, block_multiplier=2...
    method forward (line 128) | def forward(self, x):
    method _initialize_alphas_betas (line 367) | def _initialize_alphas_betas(self):
    method arch_parameters (line 386) | def arch_parameters(self):
    method weight_parameters (line 389) | def weight_parameters(self):
    method genotype (line 392) | def genotype(self):

FILE: models/cell_level_search_2d.py
  class MixedOp (line 5) | class MixedOp(nn.Module):
    method __init__ (line 7) | def __init__(self, C, stride):
    method forward (line 16) | def forward(self, x, weights):
  class Cell (line 20) | class Cell(nn.Module):
    method __init__ (line 22) | def __init__(self, steps, block_multiplier, prev_prev_fmultiplier,
    method scale_dimension (line 66) | def scale_dimension(self, dim, scale):
    method prev_feature_resize (line 70) | def prev_feature_resize(self, prev_feature, mode):
    method forward (line 80) | def forward(self, s0, s1_down, s1_same, s1_up, n_alphas):
    method _initialize_weights (line 139) | def _initialize_weights(self):

FILE: models/cell_level_search_3d.py
  class MixedOp (line 5) | class MixedOp(nn.Module):
    method __init__ (line 7) | def __init__(self, C, stride):
    method forward (line 16) | def forward(self, x, weights):
  class Cell (line 20) | class Cell(nn.Module):
    method __init__ (line 22) | def __init__(self, steps, block_multiplier, prev_prev_fmultiplier,
    method scale_dimension (line 66) | def scale_dimension(self, dim, scale):
    method prev_feature_resize (line 70) | def prev_feature_resize(self, prev_feature, mode):
    method forward (line 81) | def forward(self, s0, s1_down, s1_same, s1_up, n_alphas):
    method _initialize_weights (line 140) | def _initialize_weights(self):

FILE: models/decoding_formulas.py
  function network_layer_to_space (line 6) | def network_layer_to_space(net_arch):
  class Decoder (line 32) | class Decoder(object):
    method __init__ (line 33) | def __init__(self, alphas, betas, steps):
    method viterbi_decode (line 59) | def viterbi_decode(self):
    method genotype_decode (line 93) | def genotype_decode(self):

FILE: models/operations_2d.py
  class NaiveBN (line 10) | class NaiveBN(nn.Module):
    method __init__ (line 11) | def __init__(self, C_out, momentum=0.1):
    method forward (line 19) | def forward(self, x):
    method _initialize_weights (line 22) | def _initialize_weights(self):
  class ConvBR (line 31) | class ConvBR(nn.Module):
    method __init__ (line 32) | def __init__(self, C_in, C_out, kernel_size, stride, padding, bn=True,...
    method forward (line 41) | def forward(self, x):
    method _initialize_weights (line 48) | def _initialize_weights(self):
  class SepConv (line 56) | class SepConv(nn.Module):
    method __init__ (line 57) | def __init__(self, C_in, C_out, kernel_size, stride, padding):
    method forward (line 72) | def forward(self, x):
    method _initialize_weights (line 75) | def _initialize_weights(self):
  class Identity (line 83) | class Identity(nn.Module):
    method __init__ (line 84) | def __init__(self):
    method forward (line 88) | def forward(self, x):
    method init_weight (line 91) | def init_weight(self):
    method _initialize_weights (line 97) | def _initialize_weights(self):
  class FactorizedReduce (line 106) | class FactorizedReduce(nn.Module):
    method __init__ (line 107) | def __init__(self, C_in, C_out):
    method forward (line 116) | def forward(self, x):
    method init_weight (line 122) | def init_weight(self):
    method _initialize_weights (line 128) | def _initialize_weights(self):
  class DoubleFactorizedReduce (line 137) | class DoubleFactorizedReduce(nn.Module):
    method __init__ (line 138) | def __init__(self, C_in, C_out):
    method forward (line 147) | def forward(self, x):
    method _initialize_weights (line 153) | def _initialize_weights(self):
  class FactorizedIncrease (line 161) | class FactorizedIncrease(nn.Module):
    method __init__ (line 162) | def __init__(self, in_channel, out_channel):
    method forward (line 174) | def forward(self, x):
    method _initialize_weights (line 177) | def _initialize_weights(self):
  class DoubleFactorizedIncrease (line 186) | class DoubleFactorizedIncrease(nn.Module):
    method __init__ (line 187) | def __init__(self, in_channel, out_channel):
    method forward (line 203) | def forward(self, x):
    method _initialize_weights (line 206) | def _initialize_weights(self):

FILE: models/operations_3d.py
  class NaiveBN (line 10) | class NaiveBN(nn.Module):
    method __init__ (line 11) | def __init__(self, C_out, momentum=0.1):
    method forward (line 19) | def forward(self, x):
    method _initialize_weights (line 22) | def _initialize_weights(self):
  class ConvBR (line 31) | class ConvBR(nn.Module):
    method __init__ (line 32) | def __init__(self, C_in, C_out, kernel_size, stride, padding, bn=True,...
    method forward (line 41) | def forward(self, x):
    method _initialize_weights (line 49) | def _initialize_weights(self):
  class SepConv (line 57) | class SepConv(nn.Module):
    method __init__ (line 58) | def __init__(self, C_in, C_out, kernel_size, stride, padding):
    method forward (line 73) | def forward(self, x):
    method _initialize_weights (line 76) | def _initialize_weights(self):
  class Identity (line 84) | class Identity(nn.Module):
    method __init__ (line 85) | def __init__(self):
    method forward (line 89) | def forward(self, x):
    method init_weight (line 92) | def init_weight(self):
    method _initialize_weights (line 98) | def _initialize_weights(self):
  class FactorizedReduce (line 107) | class FactorizedReduce(nn.Module):
    method __init__ (line 108) | def __init__(self, C_in, C_out):
    method forward (line 117) | def forward(self, x):
    method init_weight (line 123) | def init_weight(self):
    method _initialize_weights (line 129) | def _initialize_weights(self):
  class DoubleFactorizedReduce (line 138) | class DoubleFactorizedReduce(nn.Module):
    method __init__ (line 139) | def __init__(self, C_in, C_out):
    method forward (line 148) | def forward(self, x):
    method _initialize_weights (line 154) | def _initialize_weights(self):
  class FactorizedIncrease (line 162) | class FactorizedIncrease(nn.Module):
    method __init__ (line 163) | def __init__(self, in_channel, out_channel):
    method forward (line 175) | def forward(self, x):
    method _initialize_weights (line 178) | def _initialize_weights(self):
  class DoubleFactorizedIncrease (line 187) | class DoubleFactorizedIncrease(nn.Module):
    method __init__ (line 188) | def __init__(self, in_channel, out_channel):
    method forward (line 204) | def forward(self, x):
    method _initialize_weights (line 207) | def _initialize_weights(self):

FILE: mypath.py
  class Path (line 1) | class Path(object):
    method db_root_dir (line 3) | def db_root_dir(dataset):

FILE: predict.py
  function RGBToPyCmap (line 64) | def RGBToPyCmap(rgbdata):
  function readPFM (line 86) | def readPFM(file):
  function save_pfm (line 122) | def save_pfm(filename, image, scale=1):
  function test_transform (line 150) | def test_transform(temp_data, crop_height, crop_width):
  function load_data (line 168) | def load_data(leftname, rightname):
  function test_md (line 192) | def test_md(leftname, rightname, savename, imgname):
  function test_kitti (line 230) | def test_kitti(leftname, rightname, savename):
  function test (line 252) | def test(leftname, rightname, savename):
  function plot_disparity (line 279) | def plot_disparity(savename, data, max_disp):

FILE: retrain/LEAStereo.py
  class LEAStereo (line 11) | class LEAStereo(nn.Module):
    method __init__ (line 12) | def __init__(self, args):
    method forward (line 27) | def forward(self, x, y):
    method get_params (line 45) | def get_params(self):

FILE: retrain/new_model_2d.py
  class Cell (line 12) | class Cell(nn.Module):
    method __init__ (line 13) | def __init__(self, steps, block_multiplier, prev_prev_fmultiplier,
    method scale_dimension (line 38) | def scale_dimension(self, dim, scale):
    method forward (line 41) | def forward(self, prev_prev_input, prev_input):
  class newFeature (line 78) | class newFeature(nn.Module):
    method __init__ (line 79) | def __init__(self, network_arch, cell_arch, cell=Cell, args=None):
    method forward (line 140) | def forward(self, x):
    method get_params (line 167) | def get_params(self):

FILE: retrain/new_model_3d.py
  class Cell (line 11) | class Cell(nn.Module):
    method __init__ (line 12) | def __init__(self, steps, block_multiplier, prev_prev_fmultiplier,
    method scale_dimension (line 37) | def scale_dimension(self, dim, scale):
    method forward (line 40) | def forward(self, prev_prev_input, prev_input):
  class newMatching (line 76) | class newMatching(nn.Module):
    method __init__ (line 77) | def __init__(self, network_arch, cell_arch, cell=Cell, args=None):
    method forward (line 136) | def forward(self, x):

FILE: retrain/skip_model_3d.py
  class Cell (line 11) | class Cell(nn.Module):
    method __init__ (line 12) | def __init__(self, steps, block_multiplier, prev_prev_fmultiplier,
    method scale_dimension (line 37) | def scale_dimension(self, dim, scale):
    method forward (line 40) | def forward(self, prev_prev_input, prev_input):
  class newMatching (line 76) | class newMatching(nn.Module):
    method __init__ (line 77) | def __init__(self, network_arch, cell_arch, cell=Cell, args=None):
    method forward (line 138) | def forward(self, x):

FILE: search.py
  class Trainer (line 50) | class Trainer(object):
    method __init__ (line 51) | def __init__(self, args):
    method training (line 151) | def training(self, epoch):
    method validation (line 231) | def validation(self, epoch):

FILE: thop/count_hooks.py
  function count_convNd (line 9) | def count_convNd(m, x, y):
  function count_conv2d (line 20) | def count_conv2d(m, x, y):
  function count_convtranspose2d (line 46) | def count_convtranspose2d(m, x, y):
  function count_bn (line 76) | def count_bn(m, x, y):
  function count_relu (line 86) | def count_relu(m, x, y):
  function count_softmax (line 95) | def count_softmax(m, x, y):
  function count_maxpool (line 108) | def count_maxpool(m, x, y):
  function count_adap_maxpool (line 116) | def count_adap_maxpool(m, x, y):
  function count_avgpool (line 125) | def count_avgpool(m, x, y):
  function count_adap_avgpool (line 135) | def count_adap_avgpool(m, x, y):
  function count_linear (line 146) | def count_linear(m, x, y):

FILE: thop/profile.py
  function profile (line 42) | def profile(model, input_size, custom_ops={}, device="cpu"):

FILE: thop/utils.py
  function clever_format (line 2) | def clever_format(num, format="%.2f"):

FILE: train.py
  function train (line 80) | def train(epoch):
  function val (line 117) | def val():
  function save_checkpoint (line 156) | def save_checkpoint(save_path, epoch,state, is_best):

FILE: utils/colorize.py
  function get_color_map (line 3) | def get_color_map():

FILE: utils/copy_state_dict.py
  function copy_state_dict (line 1) | def copy_state_dict(cur_state_dict, pre_state_dict, prefix = ''):

FILE: utils/lr_scheduler.py
  class LR_Scheduler (line 14) | class LR_Scheduler(object):
    method __init__ (line 32) | def __init__(self, mode, base_lr, num_epochs, iters_per_epoch=0,
    method __call__ (line 46) | def __call__(self, optimizer, i, epoch, best_pred):
    method _adjust_learning_rate (line 69) | def _adjust_learning_rate(self, optimizer, lr):

FILE: utils/multadds_count.py
  function count_parameters_in_MB (line 8) | def count_parameters_in_MB(model):
  function comp_multadds (line 12) | def comp_multadds(model, input_size=(3,224,224), half=False):
  function comp_multadds_fw (line 27) | def comp_multadds_fw(model, input_data):
  function add_flops_counting_methods (line 38) | def add_flops_counting_methods(net_main_module):
  function compute_average_flops_cost (line 92) | def compute_average_flops_cost(self):
  function start_flops_count (line 112) | def start_flops_count(self):
  function stop_flops_count (line 125) | def stop_flops_count(self):
  function reset_flops_count (line 138) | def reset_flops_count(self):
  function add_flops_mask (line 150) | def add_flops_mask(module, mask):
  function remove_flops_mask (line 158) | def remove_flops_mask(module):
  function conv_flops_counter_hook (line 165) | def conv_flops_counter_hook(conv_module, input, output):
  function linear_flops_counter_hook (line 197) | def linear_flops_counter_hook(linear_module, input, output):
  function batch_counter_hook (line 213) | def batch_counter_hook(module, input, output):
  function add_batch_counter_variables_or_reset (line 222) | def add_batch_counter_variables_or_reset(module):
  function add_batch_counter_hook_function (line 226) | def add_batch_counter_hook_function(module):
  function remove_batch_counter_hook_function (line 234) | def remove_batch_counter_hook_function(module):
  function add_flops_counter_variable_or_reset (line 241) | def add_flops_counter_variable_or_reset(module):
  function add_flops_counter_hook_function (line 246) | def add_flops_counter_hook_function(module):
  function remove_flops_counter_hook_function (line 262) | def remove_flops_counter_hook_function(module):
  function add_flops_mask_variable_or_reset (line 275) | def add_flops_mask_variable_or_reset(module):

FILE: utils/saver.py
  class Saver (line 8) | class Saver(object):
    method __init__ (line 10) | def __init__(self, args):
    method save_checkpoint (line 20) | def save_checkpoint(self, state, is_best, filename='checkpoint.pth.tar'):
    method save_experiment_config (line 45) | def save_experiment_config(self):

FILE: utils/summaries.py
  class TensorboardSummary (line 9) | class TensorboardSummary(object):
    method __init__ (line 10) | def __init__(self, directory):
    method create_summary (line 13) | def create_summary(self):
    method tensor2array (line 17) | def tensor2array(self, tensor, max_value=255, colormap='rainbow'):
    method visualize_image_stereo (line 45) | def visualize_image_stereo(self, writer, image, target, output, global...