SYMBOL INDEX (472 symbols across 39 files)

FILE: augmentations.py
  function denormalize_coords (line 12) | def denormalize_coords(xx, yy, width, height):
  function normalize_coords (line 19) | def normalize_coords(xx, yy, width, height):
  function apply_transform_to_params (line 26) | def apply_transform_to_params(theta0, theta_transform):
  class _IdentityParams (line 52) | class _IdentityParams(nn.Module):
    method __init__ (line 53) | def __init__(self):
    method _update (line 59) | def _update(self, batch_size):
    method forward (line 64) | def forward(self, batch_size):
  class RandomMirror (line 71) | class RandomMirror(nn.Module):
    method __init__ (line 72) | def __init__(self, vertical=True, p=0.5):
    method update_probs (line 79) | def update_probs(self, batch_size):
    method forward (line 83) | def forward(self, theta1, theta2):
  class RandomCrop (line 106) | class RandomCrop(nn.Module):
    method __init__ (line 112) | def __init__(self, crop):
    method forward (line 118) | def forward(self, im1, im2, flo):
  class RandomAffineFlow (line 135) | class RandomAffineFlow(nn.Module):
    method __init__ (line 136) | def __init__(self, args, addnoise=True):
    method inverse_transform_coords (line 150) | def inverse_transform_coords(self, width, height, thetas, offset_x=Non...
    method transform_coords (line 174) | def transform_coords(self, width, height, thetas):
    method find_invalid (line 202) | def find_invalid(self, width, height, thetas):
    method apply_random_transforms_to_params (line 231) | def apply_random_transforms_to_params(self,
    method transform_image (line 282) | def transform_image(self, images, thetas):
    method transform_flow (line 288) | def transform_flow(self, flow, theta1, theta2):
    method forward (line 312) | def forward(self, example_dict):
  class RandomAffineFlowOcc (line 368) | class RandomAffineFlowOcc(nn.Module):
    method __init__ (line 369) | def __init__(self, args, addnoise=True, crop=None):
    method inverse_transform_coords (line 387) | def inverse_transform_coords(self, width, height, thetas, offset_x=Non...
    method transform_coords (line 411) | def transform_coords(self, width, height, thetas):
    method find_invalid (line 439) | def find_invalid(self, width, height, thetas):
    method apply_random_transforms_to_params (line 468) | def apply_random_transforms_to_params(self,
    method transform_image (line 519) | def transform_image(self, images, thetas):
    method transform_flow (line 525) | def transform_flow(self, flow, theta1, theta2):
    method check_out_of_bound (line 549) | def check_out_of_bound(self, flow, occ, batch_size):
    method random_crop (line 564) | def random_crop(self, im1, im2, flo_f, flo_b, occ1, occ2):
    method forward (line 586) | def forward(self, example_dict):
  class RandomAffineFlowOccSintel (line 656) | class RandomAffineFlowOccSintel(nn.Module):
    method __init__ (line 657) | def __init__(self, args, addnoise=True, crop=None):
    method inverse_transform_coords (line 675) | def inverse_transform_coords(self, width, height, thetas, offset_x=Non...
    method transform_coords (line 699) | def transform_coords(self, width, height, thetas):
    method find_invalid (line 727) | def find_invalid(self, width, height, thetas):
    method apply_random_transforms_to_params (line 756) | def apply_random_transforms_to_params(self,
    method transform_image (line 807) | def transform_image(self, images, thetas):
    method transform_flow (line 813) | def transform_flow(self, flow, theta1, theta2):
    method check_out_of_bound (line 837) | def check_out_of_bound(self, flow, occ, batch_size):
    method random_crop (line 852) | def random_crop(self, im1, im2, flo_f, occ1):
    method forward (line 872) | def forward(self, example_dict):
  class RandomAffineFlowOccKITTI (line 935) | class RandomAffineFlowOccKITTI(nn.Module):
    method __init__ (line 936) | def __init__(self, args, addnoise=True, crop=None):
    method inverse_transform_coords (line 954) | def inverse_transform_coords(self, width, height, thetas, offset_x=Non...
    method transform_coords (line 978) | def transform_coords(self, width, height, thetas):
    method find_invalid (line 1006) | def find_invalid(self, width, height, thetas):
    method apply_random_transforms_to_params (line 1035) | def apply_random_transforms_to_params(self,
    method transform_image (line 1086) | def transform_image(self, images, thetas):
    method transform_flow (line 1092) | def transform_flow(self, flow, theta1, theta2, valid_mask):
    method check_out_of_bound (line 1117) | def check_out_of_bound(self, flow, occ, batch_size):
    method random_crop (line 1132) | def random_crop(self, im1, im2, flo_f, valid_mask):
    method forward (line 1152) | def forward(self, example_dict):

FILE: commandline.py
  function _get_type_from_arg (line 22) | def _get_type_from_arg(arg):
  function _add_arguments_for_module (line 29) | def _add_arguments_for_module(parser,
  function _add_special_arguments (line 138) | def _add_special_arguments(parser):
  function _parse_arguments (line 179) | def _parse_arguments():
  function postprocess_args (line 341) | def postprocess_args(args):
  function setup_logging_and_parse_arguments (line 383) | def setup_logging_and_parse_arguments(blocktitle):

FILE: configuration.py
  class ModelAndLoss (line 20) | class ModelAndLoss(nn.Module):
    method __init__ (line 21) | def __init__(self, args, model, training_loss, evaluation_loss=None):
    method training_loss (line 28) | def training_loss(self):
    method evaluation_loss (line 32) | def evaluation_loss(self):
    method model (line 36) | def model(self):
    method num_parameters (line 39) | def num_parameters(self):
    method forward (line 45) | def forward(self, example_dict):
  function configure_runtime_augmentations (line 65) | def configure_runtime_augmentations(args):
  function configure_model_and_loss (line 108) | def configure_model_and_loss(args):
  function configure_random_seed (line 169) | def configure_random_seed(args):
  class CheckpointSaver (line 192) | class CheckpointSaver:
    method __init__ (line 193) | def __init__(self,
    method _load_state_dict_into_module (line 211) | def _load_state_dict_into_module(self, state_dict, module, strict=True):
    method restore (line 235) | def restore(self, filename, model_and_loss, include_params="*", exclud...
    method restore_latest (line 271) | def restore_latest(self, directory, model_and_loss, include_params="*"...
    method restore_best (line 276) | def restore_best(self, directory, model_and_loss, include_params="*", ...
    method save_latest (line 281) | def save_latest(self, directory, model_and_loss, stats_dict, store_as_...
  function configure_checkpoint_saver (line 317) | def configure_checkpoint_saver(args, model_and_loss):
  function configure_data_loaders (line 362) | def configure_data_loaders(args):
  function _print_trainable_params (line 456) | def _print_trainable_params(model_and_loss, match="*"):
  function _generate_trainable_params (line 468) | def _generate_trainable_params(model_and_loss, match="*"):
  function _param_names_and_trainable_generator (line 475) | def _param_names_and_trainable_generator(model_and_loss, match="*"):
  function configure_optimizer (line 488) | def configure_optimizer(args, model_and_loss):
  function configure_lr_scheduler (line 579) | def configure_lr_scheduler(args, optimizer):

FILE: datasets/common.py
  function numpy2torch (line 10) | def numpy2torch(array):
  function read_flo_as_float32 (line 19) | def read_flo_as_float32(filename):
  function read_occ_image_as_float32 (line 30) | def read_occ_image_as_float32(filename):
  function read_image_as_float32 (line 37) | def read_image_as_float32(filename):
  function read_image_as_byte (line 41) | def read_image_as_byte(filename):

FILE: datasets/flyingThings3D.py
  function fillingInNaN (line 15) | def fillingInNaN(flow):
  class FlyingThings3d (line 41) | class FlyingThings3d(data.Dataset):
    method __init__ (line 42) | def __init__(self,
    method __getitem__ (line 135) | def __getitem__(self, index):
    method __len__ (line 187) | def __len__(self):
  class FlyingThings3dFinalTrain (line 191) | class FlyingThings3dFinalTrain(FlyingThings3d):
    method __init__ (line 192) | def __init__(self,
  class FlyingThings3dFinalTest (line 207) | class FlyingThings3dFinalTest(FlyingThings3d):
    method __init__ (line 208) | def __init__(self,
  class FlyingThings3dCleanTrain (line 223) | class FlyingThings3dCleanTrain(FlyingThings3d):
    method __init__ (line 224) | def __init__(self,
  class FlyingThings3dCleanTest (line 239) | class FlyingThings3dCleanTest(FlyingThings3d):
    method __init__ (line 240) | def __init__(self,

FILE: datasets/flyingchairs.py
  class FlyingChairs (line 81) | class FlyingChairs(data.Dataset):
    method __init__ (line 82) | def __init__(self,
    method __getitem__ (line 150) | def __getitem__(self, index):
    method __len__ (line 185) | def __len__(self):
  class FlyingChairsTrain (line 189) | class FlyingChairsTrain(FlyingChairs):
    method __init__ (line 190) | def __init__(self,
  class FlyingChairsValid (line 201) | class FlyingChairsValid(FlyingChairs):
    method __init__ (line 202) | def __init__(self,
  class FlyingChairsFull (line 213) | class FlyingChairsFull(FlyingChairs):
    method __init__ (line 214) | def __init__(self,

FILE: datasets/flyingchairsOcc.py
  class FlyingChairsOcc (line 81) | class FlyingChairsOcc(data.Dataset):
    method __init__ (line 82) | def __init__(self,
    method __getitem__ (line 165) | def __getitem__(self, index):
    method __len__ (line 210) | def __len__(self):
  class FlyingChairsOccTrain (line 214) | class FlyingChairsOccTrain(FlyingChairsOcc):
    method __init__ (line 215) | def __init__(self,
  class FlyingChairsOccValid (line 226) | class FlyingChairsOccValid(FlyingChairsOcc):
    method __init__ (line 227) | def __init__(self,
  class FlyingChairsOccFull (line 238) | class FlyingChairsOccFull(FlyingChairsOcc):
    method __init__ (line 239) | def __init__(self,

FILE: datasets/kitti_combined.py
  function read_png_flow (line 19) | def read_png_flow(flow_file):
  function kitti_random_crop (line 37) | def kitti_random_crop(im1, im2, flo_f, valid_mask, crop_height=370, crop...
  class Kitti_comb_test (line 55) | class Kitti_comb_test(data.Dataset):
    method __init__ (line 56) | def __init__(self,
    method __getitem__ (line 156) | def __getitem__(self, index):
    method __len__ (line 181) | def __len__(self):
  class Kitti_comb (line 185) | class Kitti_comb(data.Dataset):
    method __init__ (line 186) | def __init__(self,
    method __getitem__ (line 337) | def __getitem__(self, index):
    method __len__ (line 374) | def __len__(self):
  class KittiCombTrain (line 378) | class KittiCombTrain(Kitti_comb):
    method __init__ (line 379) | def __init__(self,
  class KittiCombVal (line 399) | class KittiCombVal(Kitti_comb):
    method __init__ (line 400) | def __init__(self,
  class KittiCombFull (line 420) | class KittiCombFull(Kitti_comb):
    method __init__ (line 421) | def __init__(self,
  class KittiComb2015Train (line 441) | class KittiComb2015Train(Kitti_comb):
    method __init__ (line 442) | def __init__(self,
  class KittiComb2015Val (line 458) | class KittiComb2015Val(Kitti_comb):
    method __init__ (line 459) | def __init__(self,
  class KittiComb2015Full (line 475) | class KittiComb2015Full(Kitti_comb):
    method __init__ (line 476) | def __init__(self,
  class KittiComb2015Test (line 492) | class KittiComb2015Test(Kitti_comb_test):
    method __init__ (line 493) | def __init__(self,
  class KittiComb2012Train (line 506) | class KittiComb2012Train(Kitti_comb):
    method __init__ (line 507) | def __init__(self,
  class KittiComb2012Val (line 523) | class KittiComb2012Val(Kitti_comb):
    method __init__ (line 524) | def __init__(self,
  class KittiComb2012Full (line 540) | class KittiComb2012Full(Kitti_comb):
    method __init__ (line 541) | def __init__(self,
  class KittiComb2012Test (line 557) | class KittiComb2012Test(Kitti_comb_test):
    method __init__ (line 558) | def __init__(self,

FILE: datasets/sintel.py
  class _Sintel (line 30) | class _Sintel(data.Dataset):
    method __init__ (line 31) | def __init__(self,
    method __getitem__ (line 168) | def __getitem__(self, index):
    method __len__ (line 205) | def __len__(self):
  class _Sintel_test (line 209) | class _Sintel_test(data.Dataset):
    method __init__ (line 210) | def __init__(self,
    method __getitem__ (line 285) | def __getitem__(self, index):
    method __len__ (line 314) | def __len__(self):
  class SintelTrainingCleanTrain (line 318) | class SintelTrainingCleanTrain(_Sintel):
    method __init__ (line 319) | def __init__(self, args, root, photometric_augmentations=True):
  class SintelTrainingCleanValid (line 329) | class SintelTrainingCleanValid(_Sintel):
    method __init__ (line 330) | def __init__(self, args, root, photometric_augmentations=False):
  class SintelTrainingCleanFull (line 340) | class SintelTrainingCleanFull(_Sintel):
    method __init__ (line 341) | def __init__(self, args, root, photometric_augmentations=True):
  class SintelTrainingFinalTrain (line 351) | class SintelTrainingFinalTrain(_Sintel):
    method __init__ (line 352) | def __init__(self, args, root, photometric_augmentations=True):
  class SintelTrainingFinalValid (line 362) | class SintelTrainingFinalValid(_Sintel):
    method __init__ (line 363) | def __init__(self, args, root, photometric_augmentations=False):
  class SintelTrainingFinalFull (line 373) | class SintelTrainingFinalFull(_Sintel):
    method __init__ (line 374) | def __init__(self, args, root, photometric_augmentations=True):
  class SintelTrainingCombTrain (line 384) | class SintelTrainingCombTrain(_Sintel):
    method __init__ (line 385) | def __init__(self, args, root, photometric_augmentations=True):
  class SintelTrainingCombValid (line 395) | class SintelTrainingCombValid(_Sintel):
    method __init__ (line 396) | def __init__(self, args, root, photometric_augmentations=False):
  class SintelTrainingCombFull (line 406) | class SintelTrainingCombFull(_Sintel):
    method __init__ (line 407) | def __init__(self, args, root, photometric_augmentations=True):
  class SintelTestClean (line 417) | class SintelTestClean(_Sintel_test):
    method __init__ (line 418) | def __init__(self, args, root, photometric_augmentations=False):
  class SintelTestFinal (line 427) | class SintelTestFinal(_Sintel_test):
    method __init__ (line 428) | def __init__(self, args, root, photometric_augmentations=False):

FILE: datasets/transforms.py
  function image_random_gamma (line 9) | def image_random_gamma(image, min_gamma=0.7, max_gamma=1.5, clip_image=F...
  class RandomGamma (line 17) | class RandomGamma:
    method __init__ (line 18) | def __init__(self, min_gamma=0.7, max_gamma=1.5, clip_image=False):
    method __call__ (line 23) | def __call__(self, image):
  class TransformChainer (line 35) | class TransformChainer:
    method __init__ (line 36) | def __init__(self, list_of_transforms):
    method __call__ (line 39) | def __call__(self, *args):
  class ConcatTransformSplitChainer (line 53) | class ConcatTransformSplitChainer:
    method __init__ (line 54) | def __init__(self, list_of_transforms, from_numpy=True, to_numpy=False):
    method __call__ (line 59) | def __call__(self, *args):

FILE: logger.py
  function get_default_logging_format (line 13) | def get_default_logging_format(colorize=False, brackets=False):
  function get_default_logging_datefmt (line 25) | def get_default_logging_datefmt():
  function log_module_info (line 29) | def log_module_info(module):
  class LogbookFormatter (line 35) | class LogbookFormatter(logging.Formatter):
    method __init__ (line 36) | def __init__(self, fmt=None, datefmt=None):
    method remove_colors_from_msg (line 40) | def remove_colors_from_msg(self, msg):
    method format (line 44) | def format(self, record=None):
  class ConsoleFormatter (line 49) | class ConsoleFormatter(logging.Formatter):
    method __init__ (line 50) | def __init__(self, fmt=None, datefmt=None):
    method format (line 53) | def format(self, record=None):
  class SkipLogbookFilter (line 59) | class SkipLogbookFilter(logging.Filter):
    method filter (line 60) | def filter(self, record):
  function configure_logging (line 64) | def configure_logging(filename=None):
  class LoggingBlock (line 108) | class LoggingBlock:
    method __init__ (line 109) | def __init__(self, title, emph=False):
    method __enter__ (line 119) | def __enter__(self):
    method __exit__ (line 123) | def __exit__(self, exc_type, exc_value, traceback):

FILE: losses.py
  function _elementwise_epe (line 8) | def _elementwise_epe(input_flow, target_flow):
  function _elementwise_robust_epe_char (line 12) | def _elementwise_robust_epe_char(input_flow, target_flow):
  function _downsample2d_as (line 16) | def _downsample2d_as(inputs, target_as):
  function _upsample2d_as (line 20) | def _upsample2d_as(inputs, target_as, mode="bilinear"):
  function f1_score (line 24) | def f1_score(y_true, y_pred):
  function fbeta_score (line 27) | def fbeta_score(y_true, y_pred, beta, eps=1e-8):
  function f1_score_bal_loss (line 39) | def f1_score_bal_loss(y_pred, y_true):
  class MultiScaleEPE_FlowNet (line 51) | class MultiScaleEPE_FlowNet(nn.Module):
    method __init__ (line 52) | def __init__(self,
    method forward (line 60) | def forward(self, output_dict, target_dict):
  class MultiScaleEPE_FlowNet_IRR (line 84) | class MultiScaleEPE_FlowNet_IRR(nn.Module):
    method __init__ (line 85) | def __init__(self,
    method forward (line 94) | def forward(self, output_dict, target_dict):
  class MultiScaleEPE_FlowNet_IRR_Bi (line 120) | class MultiScaleEPE_FlowNet_IRR_Bi(nn.Module):
    method __init__ (line 121) | def __init__(self,
    method forward (line 130) | def forward(self, output_dict, target_dict):
  class MultiScaleEPE_FlowNet_IRR_Occ (line 157) | class MultiScaleEPE_FlowNet_IRR_Occ(nn.Module):
    method __init__ (line 158) | def __init__(self,
    method forward (line 170) | def forward(self, output_dict, target_dict):
  class MultiScaleEPE_FlowNet_IRR_Bi_Occ (line 213) | class MultiScaleEPE_FlowNet_IRR_Bi_Occ(nn.Module):
    method __init__ (line 214) | def __init__(self,
    method forward (line 226) | def forward(self, output_dict, target_dict):
  class MultiScaleEPE_FlowNet_IRR_Bi_Occ_upsample (line 278) | class MultiScaleEPE_FlowNet_IRR_Bi_Occ_upsample(nn.Module):
    method __init__ (line 279) | def __init__(self,
    method forward (line 289) | def forward(self, output_dict, target_dict):
  class MultiScaleEPE_PWC (line 344) | class MultiScaleEPE_PWC(nn.Module):
    method __init__ (line 345) | def __init__(self,
    method forward (line 353) | def forward(self, output_dict, target_dict):
  class MultiScaleEPE_PWC_Bi (line 374) | class MultiScaleEPE_PWC_Bi(nn.Module):
    method __init__ (line 375) | def __init__(self,
    method forward (line 383) | def forward(self, output_dict, target_dict):
  class MultiScaleEPE_PWC_Occ (line 405) | class MultiScaleEPE_PWC_Occ(nn.Module):
    method __init__ (line 406) | def __init__(self,
    method forward (line 417) | def forward(self, output_dict, target_dict):
  class MultiScaleEPE_PWC_Bi_Occ (line 457) | class MultiScaleEPE_PWC_Bi_Occ(nn.Module):
    method __init__ (line 458) | def __init__(self,
    method forward (line 469) | def forward(self, output_dict, target_dict):
  class MultiScaleEPE_PWC_Bi_Occ_upsample (line 515) | class MultiScaleEPE_PWC_Bi_Occ_upsample(nn.Module):
    method __init__ (line 516) | def __init__(self,
    method forward (line 527) | def forward(self, output_dict, target_dict):
  class MultiScaleEPE_PWC_Bi_Occ_upsample_Sintel (line 579) | class MultiScaleEPE_PWC_Bi_Occ_upsample_Sintel(nn.Module):
    method __init__ (line 580) | def __init__(self,
    method forward (line 591) | def forward(self, output_dict, target_dict):
  class MultiScaleEPE_PWC_Bi_Occ_upsample_KITTI (line 640) | class MultiScaleEPE_PWC_Bi_Occ_upsample_KITTI(nn.Module):
    method __init__ (line 641) | def __init__(self,
    method forward (line 651) | def forward(self, output_dict, target_dict):

FILE: main.py
  function main (line 14) | def main():

FILE: models/IRR_FlowNet.py
  class FlowNetS (line 11) | class FlowNetS(nn.Module):
    method __init__ (line 12) | def __init__(self, args):
    method forward (line 72) | def forward(self, conv2_im1, conv3_im1, conv3_im2):
  class FlowNet1S (line 130) | class FlowNet1S(nn.Module):
    method __init__ (line 131) | def __init__(self, args, div_flow=0.05):
    method forward (line 153) | def forward(self, input_dict):

FILE: models/IRR_PWC.py
  class PWCNet (line 14) | class PWCNet(nn.Module):
    method __init__ (line 15) | def __init__(self, args, div_flow=0.05):
    method forward (line 51) | def forward(self, input_dict):

FILE: models/correlation_package/correlation.py
  class CorrelationFunction (line 6) | class CorrelationFunction(Function):
    method __init__ (line 8) | def __init__(self, pad_size=3, kernel_size=3, max_displacement=20, str...
    method forward (line 18) | def forward(self, input1, input2):
    method backward (line 31) | def backward(self, grad_output):
  class Correlation (line 47) | class Correlation(Module):
    method __init__ (line 48) | def __init__(self, pad_size=0, kernel_size=0, max_displacement=0, stri...
    method forward (line 57) | def forward(self, input1, input2):

FILE: models/correlation_package/correlation_cuda.cc
  function correlation_forward_cuda (line 8) | int correlation_forward_cuda(at::Tensor& input1, at::Tensor& input2, at:...
  function correlation_backward_cuda (line 86) | int correlation_backward_cuda(at::Tensor& input1, at::Tensor& input2, at...
  function PYBIND11_MODULE (line 165) | PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {

FILE: models/correlation_package_cu9/correlation.py
  class CorrelationFunction (line 6) | class CorrelationFunction(Function):
    method __init__ (line 8) | def __init__(self, pad_size=3, kernel_size=3, max_displacement=20, str...
    method forward (line 18) | def forward(self, input1, input2):
    method backward (line 31) | def backward(self, grad_output):
  class Correlation (line 47) | class Correlation(Module):
    method __init__ (line 48) | def __init__(self, pad_size=0, kernel_size=0, max_displacement=0, stri...
    method forward (line 57) | def forward(self, input1, input2):

FILE: models/correlation_package_cu9/correlation_cuda.cc
  function correlation_forward_cuda (line 10) | int correlation_forward_cuda(at::Tensor& input1, at::Tensor& input2, at:...
  function correlation_backward_cuda (line 89) | int correlation_backward_cuda(at::Tensor& input1, at::Tensor& input2, at...
  function PYBIND11_MODULE (line 169) | PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {

FILE: models/flownet1s.py
  class FlowNetS (line 10) | class FlowNetS(nn.Module):
    method __init__ (line 11) | def __init__(self, args):
    method forward (line 60) | def forward(self, inputs):
  class FlowNet1S (line 96) | class FlowNet1S(nn.Module):
    method __init__ (line 97) | def __init__(self, args, div_flow=0.05):
    method forward (line 102) | def forward(self, input_dict):

FILE: models/flownet1s_irr.py
  class FlowNetS (line 10) | class FlowNetS(nn.Module):
    method __init__ (line 11) | def __init__(self, args):
    method forward (line 55) | def forward(self, conv2_im1, conv3_im1, conv3_im2):
  class FlowNet1S (line 89) | class FlowNet1S(nn.Module):
    method __init__ (line 90) | def __init__(self, args, div_flow=0.05):
    method forward (line 108) | def forward(self, input_dict):

FILE: models/flownet1s_irr_bi.py
  class FlowNetS (line 10) | class FlowNetS(nn.Module):
    method __init__ (line 11) | def __init__(self, args):
    method forward (line 55) | def forward(self, conv2_im1, conv3_im1, conv3_im2):
  class FlowNet1S (line 90) | class FlowNet1S(nn.Module):
    method __init__ (line 91) | def __init__(self, args, div_flow=0.05):
    method forward (line 109) | def forward(self, input_dict):

FILE: models/flownet1s_irr_occ.py
  class FlowNetS (line 10) | class FlowNetS(nn.Module):
    method __init__ (line 11) | def __init__(self, args):
    method forward (line 71) | def forward(self, conv2_im1, conv3_im1, conv3_im2):
  class FlowNet1S (line 129) | class FlowNet1S(nn.Module):
    method __init__ (line 130) | def __init__(self, args, div_flow=0.05):
    method forward (line 148) | def forward(self, input_dict):

FILE: models/flownet1s_irr_occ_bi.py
  class FlowNetS (line 10) | class FlowNetS(nn.Module):
    method __init__ (line 11) | def __init__(self, args):
    method forward (line 71) | def forward(self, conv2_im1, conv3_im1, conv3_im2):
  class FlowNet1S (line 129) | class FlowNet1S(nn.Module):
    method __init__ (line 130) | def __init__(self, args, div_flow=0.05):
    method forward (line 148) | def forward(self, input_dict):

FILE: models/flownet_modules.py
  function conv (line 9) | def conv(in_planes, out_planes, kernel_size, stride, pad, nonlinear, bias):
  function deconv (line 22) | def deconv(in_planes, out_planes, kernel_size, stride, pad, nonlinear, b...
  function resize2D (line 35) | def resize2D(inputs, size_targets, mode="bilinear"):
  function resize2D_as (line 47) | def resize2D_as(inputs, output_as, mode="bilinear"):
  function concatenate_as (line 52) | def concatenate_as(tensor_list, tensor_as, dim, mode="bilinear"):
  function upsample2d_as (line 57) | def upsample2d_as(inputs, target_as, mode="bilinear"):
  function initialize_msra (line 62) | def initialize_msra(modules):
  function get_grid (line 85) | def get_grid(x):
  class WarpingLayer (line 93) | class WarpingLayer(nn.Module):
    method __init__ (line 94) | def __init__(self):
    method forward (line 97) | def forward(self, x, flow, height_im, width_im, div_flow):

FILE: models/irr_modules.py
  function conv (line 7) | def conv(in_planes, out_planes, kernel_size=3, stride=1, dilation=1, isR...
  function upsample_factor2 (line 21) | def upsample_factor2(inputs, target_as):
  class OccUpsampleNetwork (line 30) | class OccUpsampleNetwork(nn.Module):
    method __init__ (line 31) | def __init__(self, ch_in, ch_out):
    method forward (line 46) | def forward(self, occ, x):
  function subtract_mean (line 59) | def subtract_mean(input):
  class RefineFlow (line 63) | class RefineFlow(nn.Module):
    method __init__ (line 64) | def __init__(self, ch_in):
    method forward (line 85) | def forward(self, flow, diff_img, feature):
  class RefineOcc (line 107) | class RefineOcc(nn.Module):
    method __init__ (line 108) | def __init__(self, ch_in):
    method forward (line 129) | def forward(self, occ, feat1, feat2):

FILE: models/pwc_modules.py
  function conv (line 8) | def conv(in_planes, out_planes, kernel_size=3, stride=1, dilation=1, isR...
  function initialize_msra (line 22) | def initialize_msra(modules):
  function compute_cost_volume (line 42) | def compute_cost_volume(feat1, feat2, param_dict):
  function upsample2d_as (line 65) | def upsample2d_as(inputs, target_as, mode="bilinear"):
  function rescale_flow (line 70) | def rescale_flow(flow, div_flow, width_im, height_im, to_local=True):
  class FeatureExtractor (line 85) | class FeatureExtractor(nn.Module):
    method __init__ (line 86) | def __init__(self, num_chs):
    method forward (line 98) | def forward(self, x):
  function get_grid (line 107) | def get_grid(x):
  class WarpingLayer (line 115) | class WarpingLayer(nn.Module):
    method __init__ (line 116) | def __init__(self):
    method forward (line 119) | def forward(self, x, flow, height_im, width_im, div_flow):
  class OpticalFlowEstimator (line 135) | class OpticalFlowEstimator(nn.Module):
    method __init__ (line 136) | def __init__(self, ch_in):
    method forward (line 148) | def forward(self, x):
  class FlowEstimatorDense (line 153) | class FlowEstimatorDense(nn.Module):
    method __init__ (line 154) | def __init__(self, ch_in):
    method forward (line 163) | def forward(self, x):
  class OcclusionEstimator (line 173) | class OcclusionEstimator(nn.Module):
    method __init__ (line 174) | def __init__(self, ch_in):
    method forward (line 185) | def forward(self, x):
  class OccEstimatorDense (line 190) | class OccEstimatorDense(nn.Module):
    method __init__ (line 191) | def __init__(self, ch_in):
    method forward (line 200) | def forward(self, x):
  class ContextNetwork (line 210) | class ContextNetwork(nn.Module):
    method __init__ (line 211) | def __init__(self, ch_in):
    method forward (line 224) | def forward(self, x):
  class OccContextNetwork (line 228) | class OccContextNetwork(nn.Module):
    method __init__ (line 229) | def __init__(self, ch_in):
    method forward (line 242) | def forward(self, x):

FILE: models/pwcnet.py
  class PWCNet (line 9) | class PWCNet(nn.Module):
    method __init__ (line 10) | def __init__(self, args, div_flow=0.05):
    method forward (line 43) | def forward(self, input_dict):

FILE: models/pwcnet_bi.py
  class PWCNet (line 9) | class PWCNet(nn.Module):
    method __init__ (line 10) | def __init__(self, args, div_flow=0.05):
    method forward (line 42) | def forward(self, input_dict):

FILE: models/pwcnet_irr.py
  class PWCNet (line 9) | class PWCNet(nn.Module):
    method __init__ (line 10) | def __init__(self, args, div_flow=0.05):
    method forward (line 40) | def forward(self, input_dict):

FILE: models/pwcnet_irr_bi.py
  class PWCNet (line 9) | class PWCNet(nn.Module):
    method __init__ (line 10) | def __init__(self, args, div_flow=0.05):
    method forward (line 40) | def forward(self, input_dict):

FILE: models/pwcnet_irr_occ.py
  class PWCNet (line 9) | class PWCNet(nn.Module):
    method __init__ (line 10) | def __init__(self, args, div_flow=0.05):
    method forward (line 43) | def forward(self, input_dict):

FILE: models/pwcnet_irr_occ_bi.py
  class PWCNet (line 9) | class PWCNet(nn.Module):
    method __init__ (line 10) | def __init__(self, args, div_flow=0.05):
    method forward (line 43) | def forward(self, input_dict):

FILE: models/pwcnet_occ.py
  class PWCNet (line 9) | class PWCNet(nn.Module):
    method __init__ (line 10) | def __init__(self, args, div_flow=0.05):
    method forward (line 49) | def forward(self, input_dict):

FILE: models/pwcnet_occ_bi.py
  class PWCNet (line 9) | class PWCNet(nn.Module):
    method __init__ (line 10) | def __init__(self, args, div_flow=0.05):
    method forward (line 49) | def forward(self, input_dict):

FILE: runtime.py
  function create_progressbar (line 32) | def create_progressbar(iterable,
  function tensor2float_dict (line 93) | def tensor2float_dict(tensor_dict):
  function format_moving_averages_as_progress_dict (line 97) | def format_moving_averages_as_progress_dict(moving_averages_dict={},
  function format_learning_rate (line 106) | def format_learning_rate(lr):
  class TrainingEpoch (line 113) | class TrainingEpoch:
    method __init__ (line 114) | def __init__(self,
    method _step (line 131) | def _step(self, example_dict):
    method run (line 196) | def run(self, offset=0):
  class EvaluationEpoch (line 258) | class EvaluationEpoch:
    method __init__ (line 259) | def __init__(self,
    method save_outputs (line 276) | def save_outputs(self, example_dict, output_dict):
    method _step (line 354) | def _step(self, example_dict):
    method run (line 391) | def run(self, offset=0):
  function exec_runtime (line 472) | def exec_runtime(args,

FILE: tools.py
  function x2module (line 25) | def x2module(module_or_data_parallel):
  function addLoggingLevel (line 37) | def addLoggingLevel(level_name, level_num, method_name=None):
  function kwargs_from_args (line 67) | def kwargs_from_args(args, name, exclude=[]):
  function instance_from_kwargs (line 84) | def instance_from_kwargs(class_constructor, kwargs):
  function module_classes_to_dict (line 92) | def module_classes_to_dict(module, include_classes="*", exclude_classes=...
  function ensure_dir (line 127) | def ensure_dir(file_path):
  function search_and_replace (line 133) | def search_and_replace(string, regex, replace):
  function hostname (line 143) | def hostname():
  function get_filenames (line 151) | def get_filenames(directory, match='*.*', not_match=()):
  function str2bool (line 170) | def str2bool(v):
  function str2str_or_none (line 179) | def str2str_or_none(v):
  function str2dict (line 185) | def str2dict(v):
  function str2intlist (line 189) | def str2intlist(v):
  function str2list (line 193) | def str2list(v):
  function read_json (line 197) | def read_json(filename):
  function write_json (line 224) | def write_json(data_dict, filename):
  function datestr (line 229) | def datestr():
  function filter_list_of_strings (line 235) | def filter_list_of_strings(lst, include="*", exclude=()):
  function write_dictionary_to_file (line 246) | def write_dictionary_to_file(arguments_dict, filename):
  class MovingAverage (line 284) | class MovingAverage:
    method __init__ (line 287) | def __init__(self):
    method add_value (line 291) | def add_value(self, sigma, addcount=1):
    method add_average (line 295) | def add_average(self, avg, addcount):
    method mean (line 299) | def mean(self):
  class ExponentialMovingAverage (line 303) | class ExponentialMovingAverage:
    method __init__ (line 306) | def __init__(self, alpha=0.7):
    method add_value (line 311) | def add_value(self, sigma, addcount=1):
    method add_average (line 315) | def add_average(self, avg, addcount):
    method mean (line 319) | def mean(self):
  class TqdmToLogger (line 328) | class TqdmToLogger(tqdm.tqdm):
    method __init__ (line 329) | def __init__(self, iterable=None, desc=None, total=None, leave=True,
    method format_meter (line 351) | def format_meter(n, total, elapsed, ncols=None, prefix='', ascii=False,
    method update (line 367) | def update(self, n=1):
    method close (line 373) | def close(self):
  function tqdm_with_logging (line 381) | def tqdm_with_logging(iterable=None, desc=None, total=None, leave=True,
  function cd_dotdot (line 401) | def cd_dotdot(path_or_filename):
  function cd_dotdotdot (line 405) | def cd_dotdotdot(path_or_filename):
  function cd_dotdotdotdot (line 409) | def cd_dotdotdotdot(path_or_filename):
  function tensor2numpy (line 413) | def tensor2numpy(tensor):

FILE: utils/flow.py
  function write_flow (line 11) | def write_flow(filename, uv, v=None):
  function write_flow_png (line 37) | def write_flow_png(filename, uv, v=None, mask=None):
  function flow_to_png (line 65) | def flow_to_png(flow_map, max_value=None):
  function compute_color (line 79) | def compute_color(u, v):
  function make_color_wheel (line 123) | def make_color_wheel():
  function flow_to_png_middlebury (line 173) | def flow_to_png_middlebury(flow):

FILE: utils/interpolation.py
  function _bchw2bhwc (line 10) | def _bchw2bhwc(tensor):
  function _bhwc2bchw (line 14) | def _bhwc2bchw(tensor):
  class Meshgrid (line 18) | class Meshgrid(nn.Module):
    method __init__ (line 19) | def __init__(self):
    method _compute_meshgrid (line 26) | def _compute_meshgrid(self, width, height):
    method forward (line 32) | def forward(self, width, height, device=None, dtype=None):
  class BatchSub2Ind (line 42) | class BatchSub2Ind(nn.Module):
    method __init__ (line 43) | def __init__(self):
    method forward (line 47) | def forward(self, shape, row_sub, col_sub, out=None):
  class Interp2 (line 60) | class Interp2(nn.Module):
    method __init__ (line 61) | def __init__(self, clamp=False):
    method forward (line 80) | def forward(self, v, xq, yq):
  class Interp2MaskBinary (line 144) | class Interp2MaskBinary(nn.Module):
    method __init__ (line 145) | def __init__(self, clamp=False):
    method forward (line 168) | def forward(self, v, xq, yq, mask):
  function resize2D (line 247) | def resize2D(inputs, size_targets, mode="bilinear"):
  function resize2D_as (line 261) | def resize2D_as(inputs, output_as, mode="bilinear"):