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
- DICL
- DCVNet (1 checkpoint)
- FlowNetS
- FlowNetC (3 checkpoints)
- PWCNet (3 checkpoints)
- RAFT (3 checkpoints)
- VCN
Datasets supported
- AutoFlow
- FlyingChairs
- HD1K
- KITTI
- Kubric
- MPI Sintel
- SceneFlow Monkaa
- SceneFlow Driving
- SceneFlow FlyingThings3D
- SceneFlow FlyingThings3D subset
Results and Pre-trained checkpoints
-
DCVNet | model config | paper
| Training Dataset | Training Config | ckpts | Sintel Clean (training) | Sintel Final(training) | KITTI2015 AEPE | KITTI2015 F1-all |
|---|---|---|---|---|---|---|
| FlyingThings3DSubset + Monkaa + Driving | config | download | 1.90 | 3.35 | 4.75 | 23.41% |
-
FlowNetC | model config | arXiv
| Training Dataset | Training Config | ckpts | Sintel Clean (training) | Sintel Final(training) | KITTI2015 AEPE | KITTI2015 F1-all |
|---|---|---|---|---|---|---|
| Chairs | config | download | 3.41 | 4.94 | 14.84 | 54.23% |
| Chairs -> Things | config | download | 2.93 | 4.48 | 12.47 | 45.89% |
| Kubric | config | download | 3.57 | 3.96 | 12.11 | 36.35% |
-
PWC-Net | model config | arXiv
| Training Dataset | Training Config | ckpts | Sintel Clean (training) | Sintel Final(training) | KITTI2015 AEPE | KITTI2015 F1-all |
|---|---|---|---|---|---|---|
| Chairs | config | download | 3.5 | 4.73 | 17.81 | 51.76% |
| Chairs -> Things | config | download | 2.06 | 3.43 | 11.04 | 32.68% |
| Kubric | config | download | 3.08 | 3.31 | 9.83 | 21.94% |
-
RAFT | model config | arXiv
| Training Dataset | Training Config | ckpts | Sintel Clean (training) | Sintel Final(training) | KITTI2015 AEPE | KITTI2015 F1-all |
|---|---|---|---|---|---|---|
| Chairs | config | download | 2.23 | 4.56 | 10.45 | 38.93% |
| Chairs -> Things | config | download | 1.66 | 2.75 | 5.01 | 16.87% |
| Kubric | config | download | 2.12 | 2.54 | 6.01 | 17.35% |
Additional Information
- KITTI dataset has been evaluated with a center crop of size
1224 x 370. - FlowNetC and PWC-Net uses
paddingof size64for evaluating the KITTI2015 dataset. - RAFT and DCVNet uses
paddingof size8for evaluating the Sintel and KITTI2015 datasets.
References
Pixels icon by Icons8
1 maintainer
Metadata
1 maintainer
Release files for ezflow 0.2.5
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Source distribution (sdist)
| File | Size | Uploaded | |
|---|---|---|---|
| ezflow-0.2.5.tar.gz | 130.6 kB | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| ezflow-0.2.5-py2.py3-none-any.whl | Python 2, Python 3 | none | any | Details |
Total release size: 271.2 kB
Release files / ezflow-0.2.5.tar.gz
| Download URL | ezflow-0.2.5.tar.gz |
|---|---|
| Size | 130.6 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
3a1d69365d77303843247a38aabe954c797e5797f22b6e9bc6264905774ce997
|
|
BLAKE2b-256 checksum How to use checksums |
341f7a2195dca8c05eac4e40f7c1529e9fbbf99e0b4ca74cb84a9619fe936493
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/4.0.2 CPython/3.9.18
|
Release files / ezflow-0.2.5-py2.py3-none-any.whl
| Download URL | ezflow-0.2.5-py2.py3-none-any.whl |
|---|---|
| Size | 140.6 kB |
| Tags | Python 2 Python 3 |
|
SHA-256 checksum How to use checksums |
b54d9ecaf18767f6dacb4916a39bc56a00092ebbb8beb34b325eba71990c3b7b
|
|
BLAKE2b-256 checksum How to use checksums |
bd7df342244a73d764599f296b910db23e40b3012208e3253f5b262476ab6b06
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/4.0.2 CPython/3.9.18
|
