Index

_

__init__() (ptlflow.data.datasets.AutoFlowDataset method) (ptlflow.data.datasets.BaseFlowDataset method) (ptlflow.data.datasets.FlyingChairs2Dataset method) (ptlflow.data.datasets.FlyingChairsDataset method) (ptlflow.data.datasets.FlyingThings3DDataset method) (ptlflow.data.datasets.FlyingThings3DSubsetDataset method) (ptlflow.data.datasets.Hd1kDataset method) (ptlflow.data.datasets.KittiDataset method) (ptlflow.data.datasets.MiddleburyDataset method) (ptlflow.data.datasets.MonkaaDataset method) (ptlflow.data.datasets.SintelDataset method) (ptlflow.data.datasets.SpringDataset method) (ptlflow.data.flow_transforms.ColorJitter method) (ptlflow.data.flow_transforms.Compose method) (ptlflow.data.flow_transforms.GaussianNoise method) (ptlflow.data.flow_transforms.RandomFlip method) (ptlflow.data.flow_transforms.RandomPatchEraser method) (ptlflow.data.flow_transforms.RandomRotate method) (ptlflow.data.flow_transforms.RandomScaleAndCrop method) (ptlflow.data.flow_transforms.RandomTranslate method) (ptlflow.data.flow_transforms.Resize method) (ptlflow.data.flow_transforms.ToTensor method) (ptlflow.models.base_model.base_model.BaseModel method) (ptlflow.utils.callbacks.logger.LoggerCallback method) (ptlflow.utils.correlation.IterativeCorrBlock method) (ptlflow.utils.correlation.IterSpatialCorrelationSampler method) (ptlflow.utils.correlation.IterTranslatedSpatialCorrelationSampler method) (ptlflow.utils.flow_metrics.FlowMetrics method) (ptlflow.utils.io_adapter.IOAdapter method) (ptlflow.utils.timer.Timer method) (ptlflow.utils.timer.TimerManager method)

A

add_datasets_to_parser() (in module ptlflow.utils.utils) add_model_specific_args() (ptlflow.models.base_model.base_model.BaseModel static method)

are_shapes_compatible() (in module ptlflow.utils.utils) AutoFlowDataset (class in ptlflow.data.datasets)

B

BaseFlowDataset (class in ptlflow.data.datasets) BaseModel (class in ptlflow.models.base_model.base_model)

benchmark() (in module model_benchmark) bgr_val_as_tensor() (in module ptlflow.utils.utils)

C

calculate_metrics() (ptlflow.utils.flow_metrics.FlowMetrics method) clear() (ptlflow.utils.timer.TimerManager method) ColorJitter (class in ptlflow.data.flow_transforms) Compose (class in ptlflow.data.flow_transforms)

compute() (ptlflow.utils.flow_metrics.FlowMetrics method) config_logging() (in module ptlflow.utils.utils) configure_optimizers() (ptlflow.models.base_model.base_model.BaseModel method) count_parameters() (in module ptlflow.utils.utils)

E

estimate_inference_time() (in module model_benchmark)

F

fill() (ptlflow.utils.utils.InputScaler method) flow_read() (in module ptlflow.utils.flow_utils) flow_to_rgb() (in module ptlflow.utils.flow_utils) (in module ptlflow.utils.flowpy_torch) flow_write() (in module ptlflow.utils.flow_utils) FlowMetrics (class in ptlflow.utils.flow_metrics) FlyingChairs2Dataset (class in ptlflow.data.datasets)

FlyingChairsDataset (class in ptlflow.data.datasets) FlyingThings3DDataset (class in ptlflow.data.datasets) FlyingThings3DSubsetDataset (class in ptlflow.data.datasets) forward() (ptlflow.models.base_model.base_model.BaseModel method) (ptlflow.utils.correlation.IterativeCorrBlock method) (ptlflow.utils.correlation.IterSpatialCorrelationSampler method) (ptlflow.utils.correlation.IterTranslatedSpatialCorrelationSampler method) forward_interpolate_batch() (in module ptlflow.utils.utils)

G

GaussianNoise (class in ptlflow.data.flow_transforms)

generate_outputs() (in module validate) get_list_of_available_models_list() (in module ptlflow.utils.utils)

H

Hd1kDataset (class in ptlflow.data.datasets)

I

infer module infer() (in module infer) init_input() (in module infer) InputPadder (class in ptlflow.utils.utils) InputScaler (class in ptlflow.utils.utils)

IOAdapter (class in ptlflow.utils.io_adapter) iter_spatial_correlation_sample() (in module ptlflow.utils.correlation) iter_translated_spatial_correlation_sample() (in module ptlflow.utils.correlation) IterativeCorrBlock (class in ptlflow.utils.correlation) IterSpatialCorrelationSampler (class in ptlflow.utils.correlation) IterTranslatedSpatialCorrelationSampler (class in ptlflow.utils.correlation)

K

KittiDataset (class in ptlflow.data.datasets)

L

load_summarized_table() (in module summary_metrics)

log_image() (ptlflow.utils.callbacks.logger.LoggerCallback method) LoggerCallback (class in ptlflow.utils.callbacks.logger)

M

make_divisible() (in module ptlflow.utils.utils) mean() (ptlflow.utils.timer.Timer method) MiddleburyDataset (class in ptlflow.data.datasets) model_benchmark module module infer model_benchmark ptlflow.data.datasets ptlflow.data.flow_transforms ptlflow.models.base_model.base_model ptlflow.utils.callbacks.logger ptlflow.utils.correlation ptlflow.utils.dummy_datasets ptlflow.utils.flow_metrics ptlflow.utils.flow_utils ptlflow.utils.flowpy_torch ptlflow.utils.io_adapter ptlflow.utils.timer ptlflow.utils.utils summary_metrics train validate

MonkaaDataset (class in ptlflow.data.datasets)

O

on_train_batch_end() (ptlflow.utils.callbacks.logger.LoggerCallback method) on_train_epoch_end() (ptlflow.utils.callbacks.logger.LoggerCallback method) on_train_epoch_start() (ptlflow.utils.callbacks.logger.LoggerCallback method)

on_validation_batch_end() (ptlflow.utils.callbacks.logger.LoggerCallback method) on_validation_epoch_end() (ptlflow.utils.callbacks.logger.LoggerCallback method) on_validation_epoch_start() (ptlflow.utils.callbacks.logger.LoggerCallback method)

P

parse_dataset_selection() (ptlflow.models.base_model.base_model.BaseModel method) postprocess_predictions() (ptlflow.models.base_model.base_model.BaseModel method) prepare_inputs() (ptlflow.utils.io_adapter.IOAdapter method) preprocess_images() (ptlflow.models.base_model.base_model.BaseModel method) ptlflow.data.datasets module ptlflow.data.flow_transforms module ptlflow.models.base_model.base_model module ptlflow.utils.callbacks.logger module ptlflow.utils.correlation module

ptlflow.utils.dummy_datasets module ptlflow.utils.flow_metrics module ptlflow.utils.flow_utils module ptlflow.utils.flowpy_torch module ptlflow.utils.io_adapter module ptlflow.utils.timer module ptlflow.utils.utils module

R

RandomFlip (class in ptlflow.data.flow_transforms) RandomPatchEraser (class in ptlflow.data.flow_transforms) RandomRotate (class in ptlflow.data.flow_transforms) RandomScaleAndCrop (class in ptlflow.data.flow_transforms)

RandomTranslate (class in ptlflow.data.flow_transforms) release_gpu() (in module ptlflow.utils.utils) reset() (ptlflow.utils.timer.Timer method) (ptlflow.utils.timer.TimerManager method) Resize (class in ptlflow.data.flow_transforms)

S

save_plot() (in module model_benchmark) save_plots() (in module summary_metrics) show_outputs() (in module infer) SintelDataset (class in ptlflow.data.datasets)

SpringDataset (class in ptlflow.data.datasets) summarize() (in module summary_metrics) summary_metrics module

T

tensor_dict_to_numpy() (in module ptlflow.utils.utils) test_dataloader() (ptlflow.models.base_model.base_model.BaseModel method) tic() (ptlflow.utils.timer.Timer method) Timer (class in ptlflow.utils.timer) TimerManager (class in ptlflow.utils.timer) toc() (ptlflow.utils.timer.Timer method)

total() (ptlflow.utils.timer.Timer method) ToTensor (class in ptlflow.data.flow_transforms) train module train() (in module train) train_dataloader() (ptlflow.models.base_model.base_model.BaseModel method) training_step() (ptlflow.models.base_model.base_model.BaseModel method)

U

unfill() (ptlflow.utils.utils.InputScaler method)

unscale() (ptlflow.utils.io_adapter.IOAdapter method) update() (ptlflow.utils.flow_metrics.FlowMetrics method)

V

val_dataloader() (ptlflow.models.base_model.base_model.BaseModel method) validate module validate() (in module validate)

validate_list_of_models() (in module validate) validate_one_dataloader() (in module validate) validation_epoch_end() (ptlflow.models.base_model.base_model.BaseModel method) validation_step() (ptlflow.models.base_model.base_model.BaseModel method)

W

write_flying_chairs() (in module ptlflow.utils.dummy_datasets) write_flying_chairs2() (in module ptlflow.utils.dummy_datasets) write_hd1k() (in module ptlflow.utils.dummy_datasets) write_kitti() (in module ptlflow.utils.dummy_datasets)

write_outputs() (in module infer) write_sintel() (in module ptlflow.utils.dummy_datasets) write_things() (in module ptlflow.utils.dummy_datasets) write_things_subset() (in module ptlflow.utils.dummy_datasets) write_to_log() (ptlflow.utils.timer.TimerManager method)