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.KubricDataset method) (ptlflow.data.datasets.MiddleburyDataset method) (ptlflow.data.datasets.MiddleburySTDataset method) (ptlflow.data.datasets.MonkaaDataset method) (ptlflow.data.datasets.SintelDataset method) (ptlflow.data.datasets.SpringDataset method) (ptlflow.data.datasets.TartanAirDataset method) (ptlflow.data.datasets.ViperDataset method) (ptlflow.data.flow_transforms.CenterCrop method) (ptlflow.data.flow_transforms.ColorJitter method) (ptlflow.data.flow_transforms.Compose method) (ptlflow.data.flow_transforms.GaussianNoise method) (ptlflow.data.flow_transforms.GenerateFBCheckFlowOcclusion 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)

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) CenterCrop (class in ptlflow.data.flow_transforms) 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)

GenerateFBCheckFlowOcclusion (class in ptlflow.data.flow_transforms) 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)

KubricDataset (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) MiddleburySTDataset (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

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)

read_kubric_flow() (in module ptlflow.utils.flow_utils) read_viper_flow() (in module ptlflow.utils.flow_utils) 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

TartanAirDataset (class in ptlflow.data.datasets) tensor_dict_to_numpy() (in module ptlflow.utils.utils) test_step() (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 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

validate module validate() (in module validate)

validate_list_of_models() (in module validate) validate_one_dataloader() (in module validate) validation_step() (ptlflow.models.base_model.base_model.BaseModel method) ViperDataset (class in ptlflow.data.datasets)

W

write_autoflow() (in module ptlflow.utils.dummy_datasets) 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_kubric() (in module ptlflow.utils.dummy_datasets) write_middlebury_st() (in module ptlflow.utils.dummy_datasets)

write_outputs() (in module infer) write_sintel() (in module ptlflow.utils.dummy_datasets) write_spring() (in module ptlflow.utils.dummy_datasets) write_tartanair() (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) write_viper() (in module ptlflow.utils.dummy_datasets)