josesaribeiro/ezflow

A modular PyTorch library for optical flow estimation using neural networks

EzFlow

A modular PyTorch library for optical flow estimation using neural networks

Documentation | Tutorials

Installation

From source (recommended)

git clone https://github.com/neu-vig/ezflow
cd ezflow/
python setup.py install

From PyPI

pip install ezflow

Models supported

• DICL
• FlowNetS
• FlowNetC
• PWCNet
• RAFT
• VCN

Datasets supported

• AutoFlow
• FlyingChairs
• HD1K
• KITTI
• MPI Sintel
• SceneFlow Monkaa
• SceneFlow Driving
• SceneFlow FlyingThings3D
• SceneFlow FlyingThings3D subset

References

• RAFT
• DICL-Flow
• PWC-Net
• FlowNetPytorch
• VCN
• detectron2
• CorrelationLayer
• ptflow

Pixels icon by Icons8