SYMBOL INDEX (219 symbols across 27 files) FILE: datasets/MiddleburyLoader.py function is_image_file (line 17) | def is_image_file(filename): function default_loader (line 21) | def default_loader(path): function disparity_loader (line 25) | def disparity_loader(path): class myImageFloder (line 36) | class myImageFloder(data.Dataset): method __init__ (line 38) | def __init__(self, left, right, left_disparity, training, right_dispar... method __getitem__ (line 48) | def __getitem__(self, index): method __len__ (line 175) | def __len__(self): FILE: datasets/data_io.py function get_transform (line 6) | def get_transform(): function get_transform_aug (line 15) | def get_transform_aug(): function read_all_lines (line 25) | def read_all_lines(filename): function pfm_imread (line 32) | def pfm_imread(filename): FILE: datasets/eth3dLoader.py function is_image_file (line 17) | def is_image_file(filename): function default_loader (line 21) | def default_loader(path): function disparity_loader (line 25) | def disparity_loader(path): class myImageFloder (line 36) | class myImageFloder(data.Dataset): method __init__ (line 38) | def __init__(self, left, right, left_disparity, training, right_dispar... method __getitem__ (line 47) | def __getitem__(self, index): method __len__ (line 165) | def __len__(self): FILE: datasets/eth3d_dataset.py class ETH3DStereoDataset (line 25) | class ETH3DStereoDataset(BaseDataset): method __init__ (line 27) | def __init__(self, method load_images (line 42) | def load_images(self, idx, do_flip=False): method load_disparity (line 50) | def load_disparity(self, idx, do_flip=False): method __getitem__ (line 58) | def __getitem__(self, idx): FILE: datasets/flow_transforms.py class Compose (line 10) | class Compose(object): method __init__ (line 14) | def __init__(self, co_transforms): method __call__ (line 17) | def __call__(self, input, target): class Compose_mask (line 22) | class Compose_mask(object): method __init__ (line 26) | def __init__(self, co_transforms): method __call__ (line 29) | def __call__(self, input, target, mask): class Scale (line 36) | class Scale(object): method __init__ (line 40) | def __init__(self, size, order=2): method __call__ (line 50) | def __call__(self, inputs, target): class RandomCrop (line 61) | class RandomCrop(object): method __init__ (line 65) | def __init__(self, size): method __call__ (line 71) | def __call__(self, inputs,target): class RandomCrop_mask (line 83) | class RandomCrop_mask(object): method __init__ (line 87) | def __init__(self, size): method __call__ (line 93) | def __call__(self, inputs,target,mask): class RandomVdisp (line 106) | class RandomVdisp(object): method __init__ (line 110) | def __init__(self, angle, px, diff_angle=0, order=2, reshape=False): method __call__ (line 117) | def __call__(self, inputs,target): FILE: datasets/kitti_dataset_1215.py class KITTIDataset (line 13) | class KITTIDataset(Dataset): method __init__ (line 14) | def __init__(self, kitti15_datapath, kitti12_datapath, list_filename, ... method load_path (line 22) | def load_path(self, list_filename): method load_image (line 33) | def load_image(self, filename): method load_disp (line 36) | def load_disp(self, filename): method __len__ (line 41) | def __len__(self): method __getitem__ (line 44) | def __getitem__(self, index): FILE: datasets/kitti_dataset_1215_augmentation.py class KITTIDataset (line 15) | class KITTIDataset(Dataset): method __init__ (line 16) | def __init__(self, kitti15_datapath, kitti12_datapath, list_filename, ... method load_path (line 24) | def load_path(self, list_filename): method load_image (line 35) | def load_image(self, filename): method load_disp (line 38) | def load_disp(self, filename): method __len__ (line 43) | def __len__(self): method __getitem__ (line 46) | def __getitem__(self, index): FILE: datasets/kittiraw_loader.py function listfiles (line 14) | def listfiles(filepath): class ImageLoader (line 32) | class ImageLoader(Dataset): method __init__ (line 33) | def __init__(self, left, right): method load_path (line 37) | def load_path(self, list_filename): method load_image (line 48) | def load_image(self, filename): method load_disp (line 51) | def load_disp(self, filename): method __len__ (line 56) | def __len__(self): method __getitem__ (line 59) | def __getitem__(self, index): method __len__ (line 86) | def __len__(self): FILE: datasets/listfiles.py function dataloader (line 11) | def dataloader(filepath): FILE: datasets/middlebury_data.py function pil_loader (line 9) | def pil_loader(path): function readlines (line 16) | def readlines(filename): function read_pfm (line 22) | def read_pfm(file): class MiddleburyStereoDataset (line 31) | class MiddleburyStereoDataset: method __init__ (line 33) | def __init__(self, method load_images (line 53) | def load_images(self, idx, do_flip=False): method load_disparity (line 62) | def load_disparity(self, idx, do_flip=False): method preprocess (line 70) | def preprocess(self, inputs): method __len__ (line 78) | def __len__(self): method __getitem__ (line 81) | def __getitem__(self, idx): FILE: datasets/middlebury_data_our.py function pil_loader (line 10) | def pil_loader(path): function readlines (line 17) | def readlines(filename): function read_pfm (line 23) | def read_pfm(file): class MiddleburyStereoDataset (line 32) | class MiddleburyStereoDataset: method __init__ (line 34) | def __init__(self, method load_images (line 55) | def load_images(self, idx, do_flip=False): method load_disparity (line 64) | def load_disparity(self, idx, do_flip=False): method preprocess (line 72) | def preprocess(self, inputs): method __len__ (line 80) | def __len__(self): method __getitem__ (line 83) | def __getitem__(self, idx): FILE: datasets/middlebury_loader.py function mb_loader (line 10) | def mb_loader(filepath, res): function img_loader (line 41) | def img_loader(path): function disparity_loader (line 45) | def disparity_loader(path): class myDataset (line 49) | class myDataset(data.Dataset): method __init__ (line 51) | def __init__(self, left, right, left_disp, training, imgloader=img_loa... method __getitem__ (line 62) | def __getitem__(self, index): method __len__ (line 116) | def __len__(self): function horizontal_flip (line 120) | def horizontal_flip(img): FILE: datasets/readpfm.py function readPFM (line 6) | def readPFM(file): FILE: datasets/sceneflow_dataset.py class SceneFlowDatset (line 14) | class SceneFlowDatset(Dataset): method __init__ (line 15) | def __init__(self, datapath, list_filename, training): method load_path (line 20) | def load_path(self, list_filename): method load_image (line 28) | def load_image(self, filename): method load_disp (line 31) | def load_disp(self, filename): method __len__ (line 36) | def __len__(self): method __getitem__ (line 39) | def __getitem__(self, index): FILE: datasets/sceneflow_dataset_augmentation.py class SceneFlowDatset (line 12) | class SceneFlowDatset(Dataset): method __init__ (line 13) | def __init__(self, datapath, list_filename, training): method load_path (line 18) | def load_path(self, list_filename): method load_image (line 26) | def load_image(self, filename): method load_disp (line 29) | def load_disp(self, filename): method __len__ (line 34) | def __len__(self): method __getitem__ (line 37) | def __getitem__(self, index): FILE: main_kitti.py function train (line 95) | def train(): function train_sample (line 147) | def train_sample(sample, compute_metrics=False): function test_sample (line 179) | def test_sample(sample, compute_metrics=True): FILE: main_sceneflow.py function train (line 92) | def train(): function train_sample (line 148) | def train_sample(sample, compute_metrics=False): function test_sample (line 179) | def test_sample(sample, compute_metrics=True): FILE: models/Fast_ACV.py class SubModule (line 13) | class SubModule(nn.Module): method __init__ (line 14) | def __init__(self): method weight_init (line 17) | def weight_init(self): class Feature (line 33) | class Feature(SubModule): method __init__ (line 34) | def __init__(self): method forward (line 50) | def forward(self, x): class FeatUp (line 59) | class FeatUp(SubModule): method __init__ (line 60) | def __init__(self): method forward (line 70) | def forward(self, featL, featR=None): class channelAtt (line 88) | class channelAtt(SubModule): method __init__ (line 89) | def __init__(self, cv_chan, im_chan): method forward (line 98) | def forward(self, cv, im): class hourglass (line 104) | class hourglass(nn.Module): method __init__ (line 105) | def __init__(self, in_channels): method forward (line 133) | def forward(self, x, imgs): class hourglass_att (line 150) | class hourglass_att(nn.Module): method __init__ (line 151) | def __init__(self, in_channels): method forward (line 179) | def forward(self, x, imgs): class Fast_ACVNet (line 196) | class Fast_ACVNet(nn.Module): method __init__ (line 197) | def __init__(self, maxdisp, att_weights_only): method concat_volume_generator (line 243) | def concat_volume_generator(self, left_input, right_input, disparity_s... method forward (line 249) | def forward(self, left, right): FILE: models/Fast_ACV_plus.py class SubModule (line 13) | class SubModule(nn.Module): method __init__ (line 14) | def __init__(self): method weight_init (line 17) | def weight_init(self): class Feature (line 33) | class Feature(SubModule): method __init__ (line 34) | def __init__(self): method forward (line 49) | def forward(self, x): class FeatUp (line 58) | class FeatUp(SubModule): method __init__ (line 59) | def __init__(self): method forward (line 69) | def forward(self, featL, featR=None): class channelAtt (line 84) | class channelAtt(SubModule): method __init__ (line 85) | def __init__(self, cv_chan, im_chan): method forward (line 94) | def forward(self, cv, im): class hourglass (line 99) | class hourglass(nn.Module): method __init__ (line 100) | def __init__(self, in_channels): method forward (line 127) | def forward(self, x, imgs): class hourglass_att (line 143) | class hourglass_att(nn.Module): method __init__ (line 144) | def __init__(self, in_channels): method forward (line 186) | def forward(self, x, imgs): class Fast_ACVNet_plus (line 212) | class Fast_ACVNet_plus(nn.Module): method __init__ (line 213) | def __init__(self, maxdisp, att_weights_only): method concat_volume_generator (line 248) | def concat_volume_generator(self, left_input, right_input, disparity_s... method forward (line 256) | def forward(self, left, right): FILE: models/loss.py function model_loss_train (line 4) | def model_loss_train(disp_ests, disp_gts, img_masks): function model_loss_test (line 11) | def model_loss_test(disp_ests, disp_gts,img_masks): FILE: models/submodule.py class BasicConv (line 12) | class BasicConv(nn.Module): method __init__ (line 14) | def __init__(self, in_channels, out_channels, deconv=False, is_3d=Fals... method forward (line 32) | def forward(self, x): class Conv2x (line 41) | class Conv2x(nn.Module): method __init__ (line 43) | def __init__(self, in_channels, out_channels, deconv=False, is_3d=Fals... method forward (line 68) | def forward(self, x, rem): function disparity_regression (line 83) | def disparity_regression(x, maxdisp): function build_concat_volume (line 90) | def build_concat_volume(refimg_fea, targetimg_fea, maxdisp): function groupwise_correlation (line 104) | def groupwise_correlation(fea1, fea2, num_groups): function build_gwc_volume (line 112) | def build_gwc_volume(refimg_fea, targetimg_fea, maxdisp, num_groups): function groupwise_correlation_norm (line 124) | def groupwise_correlation_norm(fea1, fea2, num_groups): function build_gwc_volume_norm (line 135) | def build_gwc_volume_norm(refimg_fea, targetimg_fea, maxdisp, num_groups): function norm_correlation (line 148) | def norm_correlation(fea1, fea2): function build_norm_correlation_volume (line 152) | def build_norm_correlation_volume(refimg_fea, targetimg_fea, maxdisp): function disparity_variance (line 163) | def disparity_variance(x, maxdisp, disparity): function SpatialTransformer_grid (line 171) | def SpatialTransformer_grid(x, y, disp_range_samples): class Propagation (line 196) | class Propagation(nn.Module): method __init__ (line 197) | def __init__(self): method forward (line 201) | def forward(self, disparity_samples): class Propagation_prob (line 216) | class Propagation_prob(nn.Module): method __init__ (line 217) | def __init__(self): method forward (line 221) | def forward(self, prob_volume): function context_upsample (line 236) | def context_upsample(depth_low, up_weights): function regression_topk (line 251) | def regression_topk(cost, disparity_samples, k): FILE: save_disp.py function test (line 59) | def test(): function test_sample (line 82) | def test_sample(sample): FILE: test_mid.py function test_trainset (line 50) | def test_trainset(): FILE: utils/experiment.py function make_iterative_func (line 13) | def make_iterative_func(func): function make_nograd_func (line 27) | def make_nograd_func(func): function tensor2float (line 37) | def tensor2float(vars): function tensor2numpy (line 47) | def tensor2numpy(vars): function check_allfloat (line 57) | def check_allfloat(vars): function save_scalars (line 61) | def save_scalars(logger, mode_tag, scalar_dict, global_step): function save_images (line 73) | def save_images(logger, mode_tag, images_dict, global_step): function adjust_learning_rate (line 91) | def adjust_learning_rate(optimizer, epoch, base_lr, lrepochs): class AverageMeter (line 112) | class AverageMeter(object): method __init__ (line 113) | def __init__(self): method update (line 117) | def update(self, x): method mean (line 122) | def mean(self): class AverageMeterDict (line 126) | class AverageMeterDict(object): method __init__ (line 127) | def __init__(self): method update (line 131) | def update(self, x): method mean (line 146) | def mean(self): function get_world_size (line 155) | def get_world_size(): function reduce_scalar_outputs (line 164) | def reduce_scalar_outputs(scalar_outputs): FILE: utils/metrics.py function check_shape_for_metric_computation (line 15) | def check_shape_for_metric_computation(*vars): function compute_metric_for_each_image (line 22) | def compute_metric_for_each_image(metric_func): function D1_metric (line 45) | def D1_metric(D_est, D_gt, mask): function Thres_metric (line 53) | def Thres_metric(D_est, D_gt, mask, thres): function EPE_metric (line 63) | def EPE_metric(D_est, D_gt, mask): function D1_metric_mask (line 71) | def D1_metric_mask(D_est, D_gt, mask, mask_img): function Thres_metric_mask (line 80) | def Thres_metric_mask(D_est, D_gt, mask, thres, mask_img): function EPE_metric_mask (line 91) | def EPE_metric_mask(D_est, D_gt, mask, mask_img): FILE: utils/misc.py function setup_for_distributed (line 5) | def setup_for_distributed(is_master): function init_distributed_mode (line 20) | def init_distributed_mode(args): FILE: utils/visualization.py function gen_error_colormap (line 11) | def gen_error_colormap(): class disp_error_image_func (line 30) | class disp_error_image_func(Function): method forward (line 31) | def forward(self, D_est_tensor, D_gt_tensor, abs_thres=3., rel_thres=0... method backward (line 57) | def backward(self, grad_output):