SYMBOL INDEX (326 symbols across 30 files) FILE: dataloaders/stereo/KITTILoader.py function is_image_file (line 23) | def is_image_file(filename): function default_loader (line 26) | def default_loader(path): function disparity_loader (line 30) | def disparity_loader(path): class ImageLoader (line 34) | class ImageLoader(data.Dataset): method __init__ (line 35) | def __init__(self, left, right, left_disparity, calib, th=256, tw=512,... method __getitem__ (line 48) | def __getitem__(self, index): method __len__ (line 153) | def __len__(self): method train_aug (line 156) | def train_aug(self, img): method kitti_calib (line 168) | def kitti_calib(self, P2, P3): FILE: dataloaders/stereo/KITTIRawLoader.py function is_image_file (line 21) | def is_image_file(filename): function listfiles (line 25) | def listfiles(cfg, date=None, num=None, test=False): function listtrainfiles (line 70) | def listtrainfiles(filepath, train_dataname, date=None, num=None, test=F... function listvalfiles (line 123) | def listvalfiles(filepath): function default_loader (line 142) | def default_loader(path): function disparity_loader (line 146) | def disparity_loader(path): class ImageLoader (line 150) | class ImageLoader(data.Dataset): method __init__ (line 151) | def __init__(self, left, right, cfg, disp=None, training=True, demo=Fa... method __getitem__ (line 170) | def __getitem__(self, index): method __len__ (line 323) | def __len__(self): method kitti_calib (line 326) | def kitti_calib(self, P2, P3): function load_feature (line 337) | def load_feature(cfg): function match (line 348) | def match(kp1, des1, kp2, des2, matcher): FILE: dataloaders/stereo/KITTI_submission_loader.py function is_image_file (line 16) | def is_image_file(filename): function listfiles (line 20) | def listfiles(filepath, dataname): function default_loader (line 40) | def default_loader(path): function disparity_loader (line 45) | def disparity_loader(path): class ImageLoader (line 49) | class ImageLoader(data.Dataset): method __init__ (line 50) | def __init__(self, left, right, calib, loader=default_loader, dploader... method __getitem__ (line 58) | def __getitem__(self, index): method __len__ (line 92) | def __len__(self): method kitti_calib (line 95) | def kitti_calib(self, P2, P3): FILE: dataloaders/stereo/KITTIloader2012.py function is_image_file (line 8) | def is_image_file(filename): function dataloader (line 12) | def dataloader(filepath, returnCalib=False, trainAll=False): FILE: dataloaders/stereo/KITTIloader2015.py function is_image_file (line 8) | def is_image_file(filename): function dataloader (line 12) | def dataloader(filepath, returnCalib=False, trainAll=False): FILE: dataloaders/stereo/SceneFlowLoader.py function is_image_file (line 26) | def is_image_file(filename): function default_loader (line 30) | def default_loader(path): function disparity_loader (line 35) | def disparity_loader(path): class ImageLoader (line 39) | class ImageLoader(data.Dataset): method __init__ (line 40) | def __init__(self, left, right, method __getitem__ (line 55) | def __getitem__(self, index): method __len__ (line 117) | def __len__(self): method train_aug (line 120) | def train_aug(self, img): FILE: dataloaders/stereo/listflowfile.py function is_image_file (line 17) | def is_image_file(filename): function dataloader (line 21) | def dataloader(filepath): FILE: dataloaders/stereo/preprocess.py function scale_crop (line 23) | def scale_crop(input_size, scale_size=None, normalize=__imagenet_stats): function scale_random_crop (line 34) | def scale_random_crop(input_size, scale_size=None, normalize=__imagenet_... function pad_random_crop (line 46) | def pad_random_crop(input_size, scale_size=None, normalize=__imagenet_st... function inception_preproccess (line 56) | def inception_preproccess(input_size, normalize=__imagenet_stats): function inception_color_preproccess (line 65) | def inception_color_preproccess(input_size, normalize=__imagenet_stats): function get_transform (line 80) | def get_transform(name='imagenet', input_size=None, class Lighting (line 93) | class Lighting(object): method __init__ (line 96) | def __init__(self, alphastd, eigval, eigvec): method __call__ (line 101) | def __call__(self, img): class Grayscale (line 114) | class Grayscale(object): method __call__ (line 116) | def __call__(self, img): class Saturation (line 124) | class Saturation(object): method __init__ (line 126) | def __init__(self, var): method __call__ (line 129) | def __call__(self, img): class Brightness (line 135) | class Brightness(object): method __init__ (line 137) | def __init__(self, var): method __call__ (line 140) | def __call__(self, img): class Contrast (line 146) | class Contrast(object): method __init__ (line 148) | def __init__(self, var): method __call__ (line 151) | def __call__(self, img): class RandomOrder (line 158) | class RandomOrder(object): method __init__ (line 162) | def __init__(self, transforms): method __call__ (line 165) | def __call__(self, img): class ColorJitter (line 174) | class ColorJitter(RandomOrder): method __init__ (line 176) | def __init__(self, brightness=0.4, contrast=0.4, saturation=0.4): FILE: dataloaders/stereo/readpfm.py function readPFM (line 7) | def readPFM(file): FILE: dataloaders/stereo/stereo_albumentation.py function get_random_crop_coords (line 19) | def get_random_crop_coords(height, width, crop_height, crop_width): function crop (line 35) | def crop(img, x1, y1, x2, y2): function horizontal_flip (line 49) | def horizontal_flip(img_left, img_right, disp_left, disp_right): function random_crop (line 79) | def random_crop(min_crop_height, min_crop_width, input_data, split): class StereoTransform (line 123) | class StereoTransform(BasicTransform): method targets (line 129) | def targets(self): method update_params (line 135) | def update_params(self, params, **kwargs): class RightOnlyTransform (line 144) | class RightOnlyTransform(BasicTransform): method targets (line 150) | def targets(self): method update_params (line 155) | def update_params(self, params, **kwargs): class StereoTransformAsym (line 164) | class StereoTransformAsym(BasicTransform): method __init__ (line 169) | def __init__(self, always_apply=False, p=0.5, p_asym=0.2): method targets (line 174) | def targets(self): method update_params (line 180) | def update_params(self, params, **kwargs): method targets_as_params (line 189) | def targets_as_params(self): method asym (line 192) | def asym(self): class Normalize (line 202) | class Normalize(StereoTransform): method __init__ (line 214) | def __init__(self, mean=(0.485, 0.456, 0.406), std=(0.229, 0.224, 0.22... method apply (line 221) | def apply(self, image, **params): method get_transform_init_args_names (line 224) | def get_transform_init_args_names(self): class ToTensor (line 228) | class ToTensor(StereoTransform): method __init__ (line 236) | def __init__(self, always_apply=False, p=1.0): method apply (line 239) | def apply(self, image, **params): class ToGrayStereo (line 243) | class ToGrayStereo(StereoTransform, ToGray): method __init__ (line 244) | def __init__(self, always_apply=False, p=0.5): class GaussNoiseStereo (line 254) | class GaussNoiseStereo(StereoTransformAsym, GaussNoise): method __init__ (line 267) | def __init__(self, var_limit=(10.0, 50.0), mean=0, always_apply=False,... method apply_l (line 271) | def apply_l(self, img, gauss_l=None, **params): method apply_r (line 274) | def apply_r(self, img, gauss_r=None, **params): method get_params_dependent_on_targets (line 277) | def get_params_dependent_on_targets(self, params): class RGBShiftStereo (line 298) | class RGBShiftStereo(StereoTransformAsym, RGBShift): method __init__ (line 314) | def __init__(self, r_shift_limit=20, g_shift_limit=20, b_shift_limit=2... method apply_l (line 318) | def apply_l(self, image, r_shift_l=0, g_shift_l=0, b_shift_l=0, **para... method apply_r (line 321) | def apply_r(self, image, r_shift_r=0, g_shift_r=0, b_shift_r=0, **para... method get_params_dependent_on_targets (line 324) | def get_params_dependent_on_targets(self, params): class RandomBrightnessContrastStereo (line 348) | class RandomBrightnessContrastStereo(StereoTransformAsym, RandomBrightne... method __init__ (line 364) | def __init__(self, brightness_limit=0.1, contrast_limit=0.1, brightnes... method apply_l (line 369) | def apply_l(self, img, alpha_l=1.0, beta_l=0.0, **params): method apply_r (line 372) | def apply_r(self, img, alpha_r=1.0, beta_r=0.0, **params): method get_params_dependent_on_targets (line 375) | def get_params_dependent_on_targets(self, params): class RandomShiftRotate (line 399) | class RandomShiftRotate(RightOnlyTransform): method __init__ (line 410) | def __init__(self, max_shift=1.5, max_rotation=0.2, always_apply=False... method apply (line 415) | def apply(self, img, **params): FILE: dataloaders/stereo/transforms.py class Compose (line 9) | class Compose(object): method __init__ (line 10) | def __init__(self, transforms): method __call__ (line 13) | def __call__(self, sample): class ToTensor (line 19) | class ToTensor(object): method __call__ (line 22) | def __call__(self, sample): class Normalize (line 40) | class Normalize(object): method __init__ (line 43) | def __init__(self, mean, std): method __call__ (line 47) | def __call__(self, sample): class RandomCrop (line 59) | class RandomCrop(object): method __init__ (line 60) | def __init__(self, img_height, img_width, validate=False): method __call__ (line 65) | def __call__(self, sample): method crop_img (line 120) | def crop_img(self, img): class RandomVerticalFlip (line 125) | class RandomVerticalFlip(object): method __call__ (line 128) | def __call__(self, sample): class ToPILImage (line 141) | class ToPILImage(object): method __call__ (line 143) | def __call__(self, sample): class ToNumpyArray (line 150) | class ToNumpyArray(object): method __call__ (line 152) | def __call__(self, sample): class RandomContrast (line 160) | class RandomContrast(object): method __call__ (line 163) | def __call__(self, sample): class RandomGamma (line 173) | class RandomGamma(object): method __call__ (line 175) | def __call__(self, sample): class RandomBrightness (line 185) | class RandomBrightness(object): method __call__ (line 187) | def __call__(self, sample): class RandomHue (line 197) | class RandomHue(object): method __call__ (line 199) | def __call__(self, sample): class RandomSaturation (line 209) | class RandomSaturation(object): method __call__ (line 211) | def __call__(self, sample): class RandomColor (line 221) | class RandomColor(object): method __call__ (line 223) | def __call__(self, sample): FILE: demo.py function load_configs (line 26) | def load_configs(path): FILE: demo_tensorrt.py function load_configs (line 19) | def load_configs(path): FILE: demo_torchscript.py function load_configs (line 23) | def load_configs(path): FILE: models/stereo/CoEx.py class CoEx (line 16) | class CoEx(nn.Module): method __init__ (line 17) | def __init__(self, cfg): method forward (line 85) | def forward(self, imL, imR=None, u0=None, v0=None, training=False): FILE: models/stereo/CoExTRT.py class CoExTRT (line 13) | class CoExTRT(nn.Module): method __init__ (line 14) | def __init__(self, cfg): method forward (line 66) | def forward(self, imL): class FeatUp (line 103) | class FeatUp(SubModule): method __init__ (line 104) | def __init__(self, cfg): method forward (line 117) | def forward(self, x4, x8, x16, x32): class Feature (line 126) | class Feature(SubModule): method __init__ (line 127) | def __init__(self, cfg): method forward (line 146) | def forward(self, x): class Aggregation (line 159) | class Aggregation(SubModule): method __init__ (line 160) | def __init__(self, method forward (line 245) | def forward(self, x4, x8, x16, x32, cost): class Regression (line 300) | class Regression(SubModule): method __init__ (line 301) | def __init__(self, method forward (line 309) | def forward(self, cost, spg): method unfold (line 332) | def unfold(self, x): method topkpool (line 347) | def topkpool(self, cost): class CostVolume (line 355) | class CostVolume(nn.Module): method __init__ (line 356) | def __init__(self, maxdisp): method forward (line 360) | def forward(self, x, y): class AttentionCostVolume (line 379) | class AttentionCostVolume(nn.Module): method __init__ (line 380) | def __init__(self, max_disparity, in_chan, hidden_chan, head=1, weight... method forward (line 391) | def forward(self, imL, imR): class BasicConv (line 405) | class BasicConv(nn.Module): method __init__ (line 407) | def __init__(self, in_channels, out_channels, deconv=False, is_3d=Fals... method forward (line 424) | def forward(self, x): class BasicConvF (line 431) | class BasicConvF(nn.Module): method __init__ (line 433) | def __init__(self, in_channels, out_channels, deconv=False, is_3d=Fals... method forward (line 449) | def forward(self, x): class Conv2x (line 454) | class Conv2x(nn.Module): method __init__ (line 456) | def __init__(self, in_channels, out_channels, deconv=False, is_3d=Fals... method forward (line 481) | def forward(self, x, rem): FILE: models/stereo/PSMNet.py function convbn (line 17) | def convbn(in_planes, out_planes, kernel_size, stride, pad, dilation): function convbn_3d (line 23) | def convbn_3d(in_planes, out_planes, kernel_size, stride, pad): class BasicBlock (line 29) | class BasicBlock(nn.Module): method __init__ (line 31) | def __init__(self, inplanes, planes, stride, downsample, pad, dilation): method forward (line 42) | def forward(self, x): class disparityregression (line 54) | class disparityregression(nn.Module): method __init__ (line 55) | def __init__(self, maxdisp): method forward (line 59) | def forward(self, x): class feature_extraction (line 64) | class feature_extraction(nn.Module): method __init__ (line 65) | def __init__(self): method _make_layer (line 100) | def _make_layer(self, block, planes, blocks, stride, pad, dilation): method forward (line 116) | def forward(self, x): class hourglass (line 142) | class hourglass(nn.Module): method __init__ (line 143) | def __init__(self, inplanes): method forward (line 163) | def forward(self, x ,presqu, postsqu): class PSMNet (line 184) | class PSMNet(nn.Module): method __init__ (line 185) | def __init__(self, cfg): method forward (line 257) | def forward(self, left, right, u0=None, v0=None, training=False): method topkpool (line 358) | def topkpool(self, cost, k): FILE: models/stereo/submodules/Submodule.py class SubModule (line 11) | class SubModule(nn.Module): method __init__ (line 12) | def __init__(self): method weight_init (line 15) | def weight_init(self): FILE: models/stereo/submodules/aggregation.py class Aggregation (line 14) | class Aggregation(SubModule): method __init__ (line 15) | def __init__(self, method forward (line 96) | def forward(self, img, cost): FILE: models/stereo/submodules/feature.py function convbn (line 20) | def convbn(in_planes, out_planes, kernel_size, stride, pad, dilation): class PSMBasicBlock (line 26) | class PSMBasicBlock(nn.Module): method __init__ (line 28) | def __init__(self, inplanes, planes, stride, downsample, pad, dilation): method forward (line 39) | def forward(self, x): class FeatUp (line 51) | class FeatUp(SubModule): method __init__ (line 52) | def __init__(self, cfg): method forward (line 66) | def forward(self, featL, featR=None): class Feature (line 96) | class Feature(SubModule): method __init__ (line 97) | def __init__(self, cfg): method forward (line 225) | def forward(self, x): method _psm_make_layer (line 307) | def _psm_make_layer(self, block, planes, blocks, stride, pad, dilation): class unetUp (line 324) | class unetUp(nn.Module): method __init__ (line 325) | def __init__(self, in_c1, in_c2, out_c): method forward (line 333) | def forward(self, inputs1, inputs2): # small scale, large scale class FeatUp2 (line 340) | class FeatUp2(SubModule): method __init__ (line 341) | def __init__(self, cfg): method forward (line 360) | def forward(self, featL, featR=None): class Feature2 (line 406) | class Feature2(SubModule): method __init__ (line 407) | def __init__(self, cfg): method forward (line 471) | def forward(self, x): FILE: models/stereo/submodules/regression.py class Regression (line 11) | class Regression(SubModule): method __init__ (line 12) | def __init__(self, method forward (line 20) | def forward(self, cost, spg, training=False): method topkpool (line 39) | def topkpool(self, cost, k): class Regression2 (line 64) | class Regression2(SubModule): method __init__ (line 65) | def __init__(self, disp_pos, disp_neg, slant): method forward (line 74) | def forward(self, cost, disp_4b, spg=None, training=False): method topkpool (line 100) | def topkpool(self, cost, k): method init_grid (line 124) | def init_grid(self, ): FILE: models/stereo/submodules/spixel_utils/spixel.py class Args (line 14) | class Args: method __init__ (line 15) | def __init__(self): function init_spixel_grid (line 23) | def init_spixel_grid(train_img_height=256, train_img_width=512, function shift9pos (line 65) | def shift9pos(input, h_shift_unit=1, w_shift_unit=1): function poolfeat (line 89) | def poolfeat(input, prob, sp_h=2, sp_w=2): function poolfeat_ (line 144) | def poolfeat_(input, prob, sp_h=2, sp_w=2): function poolfeat_head (line 157) | def poolfeat_head(input, prob, sp_h=2, sp_w=2): function poolfeat3d (line 171) | def poolfeat3d(input, prob, sp_h=2, sp_w=2): function upfeat_original (line 184) | def upfeat_original(input, prob, up_h=2, up_w=2): function upfeat (line 225) | def upfeat(input, prob, up_h=2, up_w=2): function upfeatHW (line 237) | def upfeatHW(input, prob, tgt_h=2, tgt_w=2): function upfeat_slant (line 249) | def upfeat_slant(input, prob, slant, up_h=2, up_w=2): function upfeat_center (line 266) | def upfeat_center(input, up_h=2, up_w=2): function upfeat3d (line 275) | def upfeat3d(input, prob, up_h=2, up_w=2): function assign2uint8 (line 286) | def assign2uint8(assign): function val2uint8 (line 309) | def val2uint8(mat,maxVal): function update_spixl_map (line 315) | def update_spixl_map (spixl_map_idx_in, assig_map_in): function get_spixel_image (line 335) | def get_spixel_image(given_img, spix_index, n_spixels = 600, b_enforce_c... function spixlIdx (line 356) | def spixlIdx(args, b_train = False): class AverageMeter (line 373) | class AverageMeter(object): method __init__ (line 376) | def __init__(self): method reset (line 379) | def reset(self): method update (line 385) | def update(self, val, n=1): method __repr__ (line 391) | def __repr__(self): function batch2img (line 394) | def batch2img(img): function build_LABXY_feat (line 405) | def build_LABXY_feat(label_in, XY_feat): function rgb2Lab_torch (line 416) | def rgb2Lab_torch(img_in, mean_values = None): function label2one_hot_torch (line 455) | def label2one_hot_torch(labels, C=14): FILE: models/stereo/submodules/spixel_utils/spixel_conv.py class spixel_conv2_5d (line 16) | class spixel_conv2_5d(nn.Module): method __init__ (line 17) | def __init__(self, in_chan, out_chan, spixel_downsize=4): method forward (line 39) | def forward(self, cv, sp, xy): class spixel_upsample2_5d (line 143) | class spixel_upsample2_5d(nn.Module): method __init__ (line 144) | def __init__(self, chan, out_dim, bnrelu=True, spixel_downsize=4): method forward (line 171) | def forward(self, cv, sp): class spixel_upsample2d (line 211) | class spixel_upsample2d(nn.Module): method __init__ (line 212) | def __init__(self, ): method forward (line 216) | def forward(self, cv, sp): FILE: models/stereo/submodules/spixel_utils/spixel_loss.py function compute_semantic_pos_loss (line 16) | def compute_semantic_pos_loss(prob_in, labxy_feat, pos_weight = 0.003, ... FILE: models/stereo/submodules/spixel_utils/spixel_test.py function test (line 20) | def test(img, spx_pred, val_spixelID, save_path=None, im_num=None, downs... FILE: models/stereo/submodules/util_conv.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 33) | def forward(self, x): class Conv2x (line 42) | class Conv2x(nn.Module): method __init__ (line 44) | def __init__(self, in_channels, out_channels, deconv=False, is_3d=Fals... method forward (line 69) | def forward(self, x, rem): function BasicConv2d (line 84) | def BasicConv2d(in_channels, out_channels, kernel_size, stride, pad, dil... function BasicTransposeConv2d (line 93) | def BasicTransposeConv2d(in_channels, out_channels, kernel_size, stride,... function BasicConv3d (line 102) | def BasicConv3d(in_channels, out_channels, kernel_size, stride, pad, dil... function BasicTransposeConv3d (line 111) | def BasicTransposeConv3d(in_channels, out_channels, kernel_size, stride,... function conv3x3 (line 121) | def conv3x3(in_planes: int, out_planes: int, stride: int = 1, groups: in... function conv1x1 (line 127) | def conv1x1(in_planes: int, out_planes: int, stride: int = 1) -> nn.Conv2d: class BasicBlock (line 132) | class BasicBlock(nn.Module): method __init__ (line 135) | def __init__( method forward (line 162) | def forward(self, x: Tensor) -> Tensor: class ConvBNReLU3d (line 180) | class ConvBNReLU3d(nn.Sequential): method __init__ (line 181) | def __init__( class InvertedResidual3d (line 199) | class InvertedResidual3d(nn.Module): method __init__ (line 200) | def __init__( method forward (line 231) | def forward(self, x: Tensor) -> Tensor: class AtrousBlock (line 237) | class AtrousBlock(nn.Module): method __init__ (line 239) | def __init__(self, in_channels, out_channels, stride=1, bn=True, relu=... method forward (line 248) | def forward(self, x): FILE: models/stereo/submodules/utils.py class CostVolume (line 12) | class CostVolume(nn.Module): method __init__ (line 13) | def __init__(self, maxdisp, glue=False, group=1): method forward (line 21) | def forward(self, x, y, v=None): class AttentionCostVolume (line 40) | class AttentionCostVolume(nn.Module): method __init__ (line 41) | def __init__(self, max_disparity, in_chan, hidden_chan, head=1, weight... method forward (line 52) | def forward(self, imL, imR): class disparityregression (line 70) | class disparityregression(nn.Module): method __init__ (line 71) | def __init__(self): method forward (line 74) | def forward(self, x, reg): class channelAtt (line 80) | class channelAtt(SubModule): method __init__ (line 81) | def __init__(self, cv_chan, im_chan, D): method forward (line 90) | def forward(self, cv, im): FILE: stereo.py class Stereo (line 37) | class Stereo(LightningModule): method __init__ (line 39) | def __init__(self, cfg, dataname=None): method forward (line 48) | def forward(self, imgL, imgR=None, training=False): method training_step (line 64) | def training_step(self, batch, batch_idx): method validation_step (line 95) | def validation_step(self, batch, batch_idx): method test_step (line 148) | def test_step(self, batch, batch_idx): method configure_optimizers (line 233) | def configure_optimizers(self): method save_disp_imgs (line 243) | def save_disp_imgs(self, disp_img, filename, function load_configs (line 264) | def load_configs(path): function copy_dir (line 272) | def copy_dir(save_dir, name, save_version): class StereoTRT (line 465) | class StereoTRT(Stereo): method forward (line 467) | def forward(self, imgL): FILE: torch_to_tensorrt.py function load_configs (line 32) | def load_configs(path): function postprocess (line 40) | def postprocess(outputs): FILE: utils/load.py function make_list (line 4) | def make_list(var, n=None): function load_class (line 13) | def load_class(filename, paths, concat=True):