Index

A

add_AutoFlow() (ezflow.data.dataloader.dataloader_creator.DataloaderCreator method) add_dataset() (ezflow.data.dataloader.dataloader_creator.DataloaderCreator method) add_Driving() (ezflow.data.dataloader.dataloader_creator.DataloaderCreator method) add_FlyingChairs() (ezflow.data.dataloader.dataloader_creator.DataloaderCreator method) add_FlyingThings3D() (ezflow.data.dataloader.dataloader_creator.DataloaderCreator method) add_FlyingThings3DSubset() (ezflow.data.dataloader.dataloader_creator.DataloaderCreator method)

add_HD1K() (ezflow.data.dataloader.dataloader_creator.DataloaderCreator method) add_Kitti() (ezflow.data.dataloader.dataloader_creator.DataloaderCreator method) add_Kubric() (ezflow.data.dataloader.dataloader_creator.DataloaderCreator method) add_Monkaa() (ezflow.data.dataloader.dataloader_creator.DataloaderCreator method) add_MPISintel() (ezflow.data.dataloader.dataloader_creator.DataloaderCreator method) add_SceneFlow() (ezflow.data.dataloader.dataloader_creator.DataloaderCreator method) AdvancedSpatialTransform (class in ezflow.functional.data_augmentation.operations)

B

BaseDataset (class in ezflow.data.dataset.base_dataset) BasicBlock (class in ezflow.modules.blocks) BasicConvEncoder (class in ezflow.encoder.conv_encoder) BasicEncoder (class in ezflow.encoder.residual) bilinear_sampler() (in module ezflow.utils.resampling) BottleneckBlock (class in ezflow.modules.blocks)

BottleneckEncoder (class in ezflow.encoder.residual) build_decoder() (in module ezflow.decoder.build) build_encoder() (in module ezflow.encoder.build) build_model() (in module ezflow.models.build) build_module() (in module ezflow.modules.build) build_similarity() (in module ezflow.similarity.build) Butterfly4D (class in ezflow.decoder.separable_conv)

C

CfgNode (class in ezflow.config.config) color_transform() (in module ezflow.functional.data_augmentation.operations) configurable() (in module ezflow.config.config) conv() (in module ezflow.decoder.conv_decoder) (in module ezflow.modules.units) Conv2DMatching (class in ezflow.similarity.learnable_cost) Conv2x (class in ezflow.modules.units)

ConvDecoder (class in ezflow.decoder.conv_decoder) convex_upsample_flow() (in module ezflow.utils.resampling) ConvGRU (class in ezflow.modules.models.recurrent) ConvNormRelu (class in ezflow.modules.units) CorrelationLayer (class in ezflow.similarity.correlation.layer) CosineWarmupScheduler (class in ezflow.functional.scheduler) crop() (in module ezflow.functional.data_augmentation.operations) Custom2DConvMatching (class in ezflow.similarity.learnable_cost)

D

DataloaderCreator (class in ezflow.data.dataloader.dataloader_creator) deconv() (in module ezflow.decoder.conv_decoder) (in module ezflow.modules.units) DeviceDataLoader (class in ezflow.data.dataloader.device_dataloader)

DICL (class in ezflow.models.dicl) DisplacementAwareProjection (class in ezflow.modules.dap) DistributedTrainer (class in ezflow.engine.trainer) dump() (ezflow.config.config.CfgNode method)

E

endpointerror() (in module ezflow.utils.metrics) eraser_transform() (in module ezflow.functional.data_augmentation.operations) eval_model() (in module ezflow.engine.eval) ezflow.config.config module ezflow.config.retrieve module ezflow.data.dataloader.dataloader_creator module ezflow.data.dataloader.device_dataloader module ezflow.data.dataset.base_dataset module ezflow.data.dataset.flying_chairs module ezflow.data.dataset.flying_things3d module ezflow.data.dataset.hd1k module ezflow.data.dataset.kitti module ezflow.data.dataset.mpi_sintel module ezflow.decoder.build module ezflow.decoder.conv_decoder module ezflow.decoder.iterative.recurrent_lookup module ezflow.decoder.noniterative.operators module ezflow.decoder.noniterative.soft_regression module ezflow.decoder.separable_conv module ezflow.encoder.build module ezflow.encoder.conv_encoder module ezflow.encoder.ganet module ezflow.encoder.pspnet module ezflow.encoder.pyramid module ezflow.encoder.residual module ezflow.engine.eval module ezflow.engine.profiler module ezflow.engine.pruning module ezflow.engine.retrieve module ezflow.engine.trainer module ezflow.functional.criterion.multiscale module

ezflow.functional.criterion.sequence module ezflow.functional.data_augmentation.augmentor module ezflow.functional.data_augmentation.operations module ezflow.functional.registry module ezflow.functional.scheduler module ezflow.models.build module ezflow.models.dicl module ezflow.models.flownet_c module ezflow.models.flownet_s module ezflow.models.predictor module ezflow.models.pwcnet module ezflow.models.raft module ezflow.models.vcn module ezflow.modules.blocks module ezflow.modules.build module ezflow.modules.dap module ezflow.modules.models.recurrent module ezflow.modules.units module ezflow.similarity.build module ezflow.similarity.correlation.layer module ezflow.similarity.correlation.pairwise module ezflow.similarity.correlation.sampler module ezflow.similarity.learnable_cost module ezflow.utils.io module ezflow.utils.metrics module ezflow.utils.registry module ezflow.utils.resampling module ezflow.utils.viz module ezflow.utils.warp module

F

FeatureProjection4D (class in ezflow.decoder.separable_conv) flip_transform() (in module ezflow.functional.data_augmentation.operations) FlowAugmentor (class in ezflow.functional.data_augmentation.augmentor) FlowEntropy (class in ezflow.decoder.noniterative.operators) FlowHead (class in ezflow.decoder.iterative.recurrent_lookup) FlowNetC (class in ezflow.models.flownet_c) FlowNetConvDecoder (class in ezflow.decoder.conv_decoder) FlowNetConvEncoder (class in ezflow.encoder.conv_encoder) FlowNetS (class in ezflow.models.flownet_s) FlyingChairs (class in ezflow.data.dataset.flying_chairs) FlyingThings3D (class in ezflow.data.dataset.flying_things3d) FlyingThings3DClean (class in ezflow.data.dataset.flying_things3d) FlyingThings3DFinal (class in ezflow.data.dataset.flying_things3d) FlyingThings3DSubset (class in ezflow.data.dataset.flying_things3d) forward() (ezflow.decoder.conv_decoder.ConvDecoder method) (ezflow.decoder.conv_decoder.FlowNetConvDecoder method) (ezflow.decoder.iterative.recurrent_lookup.FlowHead method) (ezflow.decoder.iterative.recurrent_lookup.MotionEncoder method) (ezflow.decoder.iterative.recurrent_lookup.RecurrentLookupUpdateBlock method) (ezflow.decoder.iterative.recurrent_lookup.SepConvGRU method) (ezflow.decoder.iterative.recurrent_lookup.SmallMotionEncoder method) (ezflow.decoder.iterative.recurrent_lookup.SmallRecurrentLookupUpdateBlock method) (ezflow.decoder.noniterative.operators.FlowEntropy method) (ezflow.decoder.noniterative.soft_regression.Soft4DFlowRegression method) (ezflow.decoder.noniterative.soft_regression.SoftArg2DFlowRegression method) (ezflow.decoder.separable_conv.Butterfly4D method) (ezflow.decoder.separable_conv.FeatureProjection4D method) (ezflow.decoder.separable_conv.SeparableConv4D method) (ezflow.decoder.separable_conv.SeparableConv4DBlock method) (ezflow.encoder.conv_encoder.BasicConvEncoder method) (ezflow.encoder.ganet.GANetBackbone method) (ezflow.encoder.pspnet.PSPNetBackbone method) (ezflow.encoder.pspnet.PyramidPooling method) (ezflow.encoder.pspnet.ResidualBlock method) (ezflow.encoder.pyramid.PyramidEncoder method) (ezflow.encoder.residual.BasicEncoder method) (ezflow.encoder.residual.BottleneckEncoder method) (ezflow.functional.criterion.multiscale.MultiScaleLoss method) (ezflow.functional.criterion.sequence.SequenceLoss method) (ezflow.models.dicl.DICL method) (ezflow.models.flownet_c.FlowNetC method) (ezflow.models.flownet_s.FlowNetS method) (ezflow.models.pwcnet.PWCNet method) (ezflow.models.raft.RAFT method) (ezflow.models.vcn.VCN method) (ezflow.modules.blocks.BasicBlock method) (ezflow.modules.blocks.BottleneckBlock method) (ezflow.modules.dap.DisplacementAwareProjection method) (ezflow.modules.models.recurrent.ConvGRU method) (ezflow.modules.units.Conv2x method) (ezflow.modules.units.ConvNormRelu method) (ezflow.similarity.correlation.layer.CorrelationLayer method) (ezflow.similarity.correlation.sampler.IterSpatialCorrelationSampler method) (ezflow.similarity.learnable_cost.Conv2DMatching method) (ezflow.similarity.learnable_cost.Custom2DConvMatching method) (ezflow.similarity.learnable_cost.LearnableMatchingCost method) (ezflow.similarity.learnable_cost.MatryoshkaDilatedCostVolume method) (ezflow.similarity.learnable_cost.MatryoshkaDilatedCostVolumeList method)

forward_interpolate() (in module ezflow.utils.resampling)

G

GANetBackbone (class in ezflow.encoder.ganet) get() (ezflow.utils.registry.Registry method) get_cfg() (in module ezflow.config.retrieve) get_cfg_obj() (in module ezflow.config.retrieve) get_cfg_path() (in module ezflow.config.retrieve)

get_dataloader() (ezflow.data.dataloader.dataloader_creator.DataloaderCreator method) get_functional() (in module ezflow.functional.registry) get_list() (ezflow.utils.registry.Registry method) get_lr_factor() (ezflow.functional.scheduler.CosineWarmupScheduler method) get_training_cfg() (in module ezflow.engine.retrieve)

H

HD1K (class in ezflow.data.dataset.hd1k)

I

InputPadder (class in ezflow.utils.io)

iter_spatial_correlation_sample() (in module ezflow.similarity.correlation.sampler) IterSpatialCorrelationSampler (class in ezflow.similarity.correlation.sampler)

K

Kitti (class in ezflow.data.dataset.kitti)

L

LearnableMatchingCost (class in ezflow.similarity.learnable_cost)

M

MatryoshkaDilatedCostVolume (class in ezflow.similarity.learnable_cost) MatryoshkaDilatedCostVolumeList (class in ezflow.similarity.learnable_cost) module ezflow.config.config ezflow.config.retrieve ezflow.data.dataloader.dataloader_creator ezflow.data.dataloader.device_dataloader ezflow.data.dataset.base_dataset ezflow.data.dataset.flying_chairs ezflow.data.dataset.flying_things3d ezflow.data.dataset.hd1k ezflow.data.dataset.kitti ezflow.data.dataset.mpi_sintel ezflow.decoder.build ezflow.decoder.conv_decoder ezflow.decoder.iterative.recurrent_lookup ezflow.decoder.noniterative.operators ezflow.decoder.noniterative.soft_regression ezflow.decoder.separable_conv ezflow.encoder.build ezflow.encoder.conv_encoder ezflow.encoder.ganet ezflow.encoder.pspnet ezflow.encoder.pyramid ezflow.encoder.residual ezflow.engine.eval ezflow.engine.profiler ezflow.engine.pruning ezflow.engine.retrieve ezflow.engine.trainer ezflow.functional.criterion.multiscale ezflow.functional.criterion.sequence ezflow.functional.data_augmentation.augmentor ezflow.functional.data_augmentation.operations ezflow.functional.registry ezflow.functional.scheduler ezflow.models.build ezflow.models.dicl ezflow.models.flownet_c ezflow.models.flownet_s ezflow.models.predictor ezflow.models.pwcnet ezflow.models.raft ezflow.models.vcn ezflow.modules.blocks ezflow.modules.build ezflow.modules.dap ezflow.modules.models.recurrent ezflow.modules.units ezflow.similarity.build ezflow.similarity.correlation.layer ezflow.similarity.correlation.pairwise ezflow.similarity.correlation.sampler ezflow.similarity.learnable_cost ezflow.utils.io ezflow.utils.metrics ezflow.utils.registry ezflow.utils.resampling ezflow.utils.viz ezflow.utils.warp

MotionEncoder (class in ezflow.decoder.iterative.recurrent_lookup) MPISintel (class in ezflow.data.dataset.mpi_sintel) MPISintelClean (class in ezflow.data.dataset.mpi_sintel) MPISintelFinal (class in ezflow.data.dataset.mpi_sintel) MultiScaleLoss (class in ezflow.functional.criterion.multiscale) MutliScalePairwise4DCorr (class in ezflow.similarity.correlation.pairwise)

N

noise_transform() (in module ezflow.functional.data_augmentation.operations)

Normalize (class in ezflow.functional.data_augmentation.operations)

P

pad() (ezflow.utils.io.InputPadder method) Predictor (class in ezflow.models.predictor) profile_inference() (in module ezflow.engine.eval) Profiler (class in ezflow.engine.profiler) prune_l1_structured() (in module ezflow.engine.pruning)

prune_l1_unstructured() (in module ezflow.engine.pruning) PSPNetBackbone (class in ezflow.encoder.pspnet) PWCNet (class in ezflow.models.pwcnet) PyramidEncoder (class in ezflow.encoder.pyramid) PyramidPooling (class in ezflow.encoder.pspnet)

R

RAFT (class in ezflow.models.raft) read_flow() (in module ezflow.utils.io) read_flow_middlebury() (in module ezflow.utils.io) read_flow_pfm() (in module ezflow.utils.io) read_flow_png() (in module ezflow.utils.io) read_image() (in module ezflow.utils.io)

RecurrentLookupUpdateBlock (class in ezflow.decoder.iterative.recurrent_lookup) register() (ezflow.utils.registry.Registry method) Registry (class in ezflow.utils.registry) ResidualBlock (class in ezflow.encoder.pspnet) resize_sparse_flow_map() (in module ezflow.functional.data_augmentation.operations) run_inference() (in module ezflow.engine.eval)

S

SeparableConv4D (class in ezflow.decoder.separable_conv) SeparableConv4DBlock (class in ezflow.decoder.separable_conv) SepConvGRU (class in ezflow.decoder.iterative.recurrent_lookup) SequenceLoss (class in ezflow.functional.criterion.sequence) SmallMotionEncoder (class in ezflow.decoder.iterative.recurrent_lookup)

SmallRecurrentLookupUpdateBlock (class in ezflow.decoder.iterative.recurrent_lookup) Soft4DFlowRegression (class in ezflow.decoder.noniterative.soft_regression) SoftArg2DFlowRegression (class in ezflow.decoder.noniterative.soft_regression) sparse_spatial_transform() (in module ezflow.functional.data_augmentation.operations) SparseFlowAugmentor (class in ezflow.functional.data_augmentation.augmentor) spatial_transform() (in module ezflow.functional.data_augmentation.operations)

T

to_dict() (ezflow.config.config.CfgNode method) train() (ezflow.engine.trainer.DistributedTrainer method) (ezflow.engine.trainer.Trainer method)

Trainer (class in ezflow.engine.trainer)

U

unpad() (ezflow.utils.io.InputPadder method)

upflow() (in module ezflow.utils.resampling)

V

VCN (class in ezflow.models.vcn)

W

warmup() (in module ezflow.engine.eval)

warp() (in module ezflow.utils.warp) write_flow() (in module ezflow.utils.io)