EzFlow

A modular PyTorch library for optical flow estimation using neural networks

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

• [x] DICL
• [x] FlowNetS
• [x] FlowNetC
• [x] PWCNet
• [x] RAFT
• [x] VCN

Datasets supported

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

References

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