ezflow
A modular PyTorch library for optical flow estimation using neural networks
Science Score: 64.0%
This score indicates how likely this project is to be science-related based on various indicators:
-
✓CITATION.cff file
Found CITATION.cff file -
✓codemeta.json file
Found codemeta.json file -
✓.zenodo.json file
Found .zenodo.json file -
○DOI references
-
✓Academic publication links
Links to: arxiv.org -
✓Committers with academic emails
1 of 6 committers (16.7%) from academic institutions -
○Institutional organization owner
-
○JOSS paper metadata
-
○Scientific vocabulary similarity
Low similarity (10.3%) to scientific vocabulary
Keywords
Keywords from Contributors
Repository
A modular PyTorch library for optical flow estimation using neural networks
Basic Info
- Host: GitHub
- Owner: neu-vi
- License: mit
- Language: Python
- Default Branch: main
- Homepage: https://ezflow.readthedocs.io/
- Size: 595 KB
Statistics
- Stars: 136
- Watchers: 5
- Forks: 8
- Open Issues: 25
- Releases: 6
Topics
Metadata Files
README.md
EzFlow
A modular PyTorch library for optical flow estimation using neural networks
Installation
From source (recommended)
```shell
git clone https://github.com/neu-vig/ezflow cd ezflow/ python setup.py install
```
From PyPI
```shell
pip install ezflow
```
Models supported
- [x] DICL
- [x] DCVNet (1 checkpoint)
- [x] FlowNetS
- [x] FlowNetC (3 checkpoints)
- [x] PWCNet (3 checkpoints)
- [x] RAFT (3 checkpoints)
- [x] VCN
Datasets supported
- [x] AutoFlow
- [x] FlyingChairs
- [x] HD1K
- [x] KITTI
- [x] Kubric
- [x] MPI Sintel
- [x] SceneFlow Monkaa
- [x] SceneFlow Driving
- [x] SceneFlow FlyingThings3D
- [x] 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
Owner
- Name: neu-vi
- Login: neu-vi
- Kind: organization
- Location: United States of America
- Repositories: 3
- Profile: https://github.com/neu-vi
Northeastern University, Visual Intelligence Lab
Citation (CITATION.cff)
cff-version: 1.2.0 message: "If you find this software useful, please cite it as below." authors: - family-names: "Shah" given-names: "Neelay" - family-names: "Goswami" given-names: "Prajnan" - family-names: "Jiang" given-names: "Huaizu" title: "EzFlow: A modular PyTorch library for optical flow estimation using neural networks" date-released: 2021-11-18 url: "https://github.com/neu-vig/ezflow" license: MIT
GitHub Events
Total
- Watch event: 6
- Fork event: 1
Last Year
- Watch event: 6
- Fork event: 1
Committers
Last synced: about 2 years ago
Top Committers
| Name | Commits | |
|---|---|---|
| Neelay Shah | s****9@g****m | 170 |
| Prajnan Goswami | 8****3 | 36 |
| dependabot[bot] | 4****] | 29 |
| Prajnan Goswami | p****s@g****m | 15 |
| Neelay Shah | n****h@t****e | 1 |
| Huaizu Jiang | p****k | 1 |
Committer Domains (Top 20 + Academic)
Issues and Pull Requests
Last synced: 8 months ago
All Time
- Total issues: 24
- Total pull requests: 126
- Average time to close issues: 5 months
- Average time to close pull requests: 21 days
- Total issue authors: 5
- Total pull request authors: 4
- Average comments per issue: 0.92
- Average comments per pull request: 0.84
- Merged pull requests: 73
- Bot issues: 0
- Bot pull requests: 54
Past Year
- Issues: 0
- Pull requests: 0
- Average time to close issues: N/A
- Average time to close pull requests: N/A
- Issue authors: 0
- Pull request authors: 0
- Average comments per issue: 0
- Average comments per pull request: 0
- Merged pull requests: 0
- Bot issues: 0
- Bot pull requests: 0
Top Authors
Issue Authors
- NeelayS (14)
- prajnan93 (6)
- Salvatore-tech (2)
- AruniRC (1)
- YanjieGuo (1)
Pull Request Authors
- dependabot[bot] (56)
- NeelayS (36)
- prajnan93 (33)
- kardevroop (1)
Top Labels
Issue Labels
Pull Request Labels
Dependencies
- Babel ==2.9.1
- Jinja2 ==3.0.1
- Markdown ==3.3.4
- MarkupSafe ==2.0.1
- Pillow *
- PySocks ==1.7.1
- PyYAML ==5.4.1
- Pygments ==2.9.0
- Sphinx ==4.1.2
- Unidecode ==1.3.2
- Werkzeug ==2.0.1
- absl-py ==0.13.0
- alabaster ==0.7.12
- antlr4-python3-runtime ==4.8
- appdirs ==1.4.4
- argh ==0.26.2
- arrow ==0.15.1
- attrs *
- backports.entry-points-selectable ==1.1.0
- binaryornot ==0.4.4
- black ==21.7b0
- bleach ==3.3.0
- brotlipy ==0.7.0
- bump2version ==0.5.11
- cachetools ==4.2.2
- certifi ==2021.5.30
- cffi ==1.14.6
- cfgv ==3.3.0
- chardet ==4.0.0
- charset-normalizer ==2.0.1
- click ==8.0.1
- colorama ==0.4.4
- cookiecutter ==2.1.1
- coverage ==5.5
- cryptography ==3.4.8
- cycler ==0.10.0
- distlib ==0.3.2
- dnspython ==1.16.0
- docopt ==0.6.2
- docutils ==0.17.1
- easydict ==1.9
- entrypoints ==0.3
- fett ==0.3.2
- filelock ==3.0.12
- flake8 ==3.9.2
- fvcore ==0.1.5.post20210915
- google-auth ==1.35.0
- google-auth-oauthlib ==0.4.5
- grpcio ==1.39.0
- identify ==2.2.13
- idna ==3.2
- imagesize ==1.2.0
- importlib-metadata ==0.23
- iniconfig ==1.1.1
- iopath ==0.1.9
- jinja2-time ==0.2.0
- kiwisolver *
- matplotlib *
- mccabe ==0.6.1
- mkl-fft *
- mkl-random *
- mkl-service *
- mypy-extensions ==0.4.3
- networkx ==2.6.2
- nodeenv ==1.6.0
- numpy *
- oauthlib ==3.1.1
- olefile ==0.46
- omegaconf *
- opencv-python ==4.5.3.56
- packaging ==21.0
- pathspec ==0.9.0
- pathtools ==0.1.2
- pkginfo ==1.7.1
- platformdirs ==2.0.0
- pluggy ==0.13.1
- portalocker *
- poyo ==0.5.0
- pre-commit ==2.14.0
- protobuf ==3.17.3
- py ==1.10.0
- pyOpenSSL ==20.0.1
- pyasn1 ==0.4.8
- pyasn1-modules ==0.2.8
- pycodestyle ==2.7.0
- pycparser ==2.20
- pyflakes ==2.3.1
- pymongo ==3.11.4
- pyparsing ==2.4.7
- pytest ==6.2.4
- python-dateutil ==2.8.1
- python-slugify ==5.0.2
- pytz ==2021.1
- readme-renderer ==29.0
- regex *
- requests ==2.26.0
- requests-oauthlib ==1.3.0
- requests-toolbelt ==0.9.1
- rsa ==4.7.2
- scipy ==1.7.0
- six *
- snooty-lextudio ==1.11.1.dev0
- snowballstemmer ==2.1.0
- sphinx-rtd-theme ==1.0.0
- sphinxcontrib-applehelp ==1.0.2
- sphinxcontrib-devhelp ==1.0.2
- sphinxcontrib-htmlhelp ==2.0.0
- sphinxcontrib-jsmath ==1.0.1
- sphinxcontrib-qthelp ==1.0.3
- sphinxcontrib-serializinghtml ==1.1.5
- sphinxcontrib-websupport ==1.2.4
- tabulate ==0.8.9
- tensorboard ==2.6.0
- tensorboard-data-server ==0.6.1
- tensorboard-plugin-wit ==1.8.0
- termcolor ==1.1.0
- text-unidecode ==1.3
- toml *
- tomli ==1.2.1
- torch ==1.9.0
- torchmetrics ==0.5.0
- torchvision ==0.10.0
- tornado *
- tox ==3.14.0
- tqdm *
- twine ==1.14.0
- typed-ast *
- typing-extensions *
- ujson *
- urllib3 ==1.26.6
- virtualenv ==20.5.0
- watchdog ==1.0.2
- webencodings ==0.5.1
- whichcraft ==0.6.1
- yacs ==0.1.8
- zipp *
- Babel ==2.9.1
- Jinja2 ==3.0.1
- Markdown ==3.3.4
- MarkupSafe ==2.0.1
- Pillow *
- PySocks ==1.7.1
- PyYAML ==5.4.1
- Pygments ==2.9.0
- Unidecode ==1.3.2
- Werkzeug ==2.0.1
- absl-py ==0.13.0
- alabaster ==0.7.12
- antlr4-python3-runtime ==4.8
- appdirs ==1.4.4
- argh ==0.26.2
- arrow ==0.15.1
- attrs *
- backports.entry-points-selectable ==1.1.0
- binaryornot ==0.4.4
- black ==21.7b0
- bleach ==3.3.0
- brotlipy ==0.7.0
- bump2version ==0.5.11
- cachetools ==4.2.2
- certifi ==2021.5.30
- cffi ==1.14.6
- cfgv ==3.3.0
- chardet ==4.0.0
- charset-normalizer ==2.0.1
- click ==8.0.1
- colorama ==0.4.4
- coverage ==5.5
- cryptography ==3.4.8
- cycler ==0.10.0
- distlib ==0.3.2
- dnspython ==1.16.0
- docopt ==0.6.2
- docutils ==0.17.1
- easydict ==1.9
- entrypoints ==0.3
- fett ==0.3.2
- filelock ==3.0.12
- flake8 ==3.9.2
- fvcore ==0.1.5.post20210915
- grpcio ==1.39.0
- identify ==2.2.13
- idna ==3.2
- imagesize ==1.2.0
- iniconfig ==1.1.1
- iopath ==0.1.9
- jinja2-time ==0.2.0
- kiwisolver *
- matplotlib *
- mccabe ==0.6.1
- mkl-fft ==1.3.0
- mkl-random *
- mkl-service *
- mypy-extensions ==0.4.3
- networkx ==2.6.2
- nodeenv ==1.6.0
- numpy *
- oauthlib ==3.1.1
- olefile ==0.46
- omegaconf *
- opencv-python ==4.5.3.56
- packaging ==21.0
- pathspec ==0.9.0
- pathtools ==0.1.2
- pkginfo ==1.7.1
- platformdirs ==2.0.0
- pluggy ==0.13.1
- portalocker *
- poyo ==0.5.0
- pre-commit ==2.14.0
- protobuf ==3.17.3
- py ==1.10.0
- pyOpenSSL ==20.0.1
- pyasn1 ==0.4.8
- pyasn1-modules ==0.2.8
- pycodestyle ==2.7.0
- pycparser ==2.20
- pyflakes ==2.3.1
- pymongo ==3.11.4
- pyparsing ==2.4.7
- pytest ==6.2.4
- python-dateutil ==2.8.1
- python-slugify ==5.0.2
- pytz ==2021.1
- readme-renderer ==29.0
- regex *
- requests ==2.26.0
- requests-oauthlib ==1.3.0
- requests-toolbelt ==0.9.1
- rsa ==4.7.2
- scipy ==1.7.0
- six *
- snooty-lextudio ==1.11.1.dev0
- snowballstemmer ==2.1.0
- tabulate ==0.8.9
- tensorboard ==2.6.0
- tensorboard-data-server ==0.6.1
- tensorboard-plugin-wit ==1.8.0
- termcolor ==1.1.0
- text-unidecode ==1.3
- toml *
- tomli ==1.2.1
- torch >=1.9.0
- torchmetrics >=0.5.0
- torchvision >=0.10.0
- tornado *
- tox ==3.14.0
- tqdm *
- twine ==1.14.0
- typed-ast *
- typing-extensions *
- ujson *
- urllib3 ==1.26.6
- virtualenv ==20.5.0
- watchdog ==1.0.2
- webencodings ==0.5.1
- whichcraft ==0.6.1
- yacs ==0.1.8
- zipp *
- actions/checkout v2 composite
- actions/setup-python v2 composite
- codecov/codecov-action v1 composite
- actions/checkout v2 composite
- actions/setup-python v2 composite
- actions/checkout v2 composite
- actions/setup-python v2 composite
- actions/checkout v2 composite
- actions/setup-python v2 composite
- pypa/gh-action-pypi-publish 27b31702a0e7fc50959f5ad993c78deac1bdfc29 composite
- _libgcc_mutex 0.1.*
- attrs 21.2.0.*
- blas 1.0.*
- brotlipy 0.7.0.*
- bzip2 1.0.8.*
- ca-certificates 2021.10.8.*
- cffi 1.14.6.*
- click 8.0.1.*
- colorama 0.4.4.*
- cudatoolkit 11.3.*
- cycler 0.10.0.*
- dnspython 1.16.0.*
- docopt 0.6.2.*
- einops 0.3.2.*
- expat 2.4.1.*
- ffmpeg 4.3.*
- fontconfig 2.13.1.*
- freetype 2.10.4.*
- future 0.18.2.*
- icu 58.2.*
- intel-openmp 2021.2.0.*
- jpeg 9b.*
- kiwisolver 1.3.1.*
- lcms2 2.12.*
- libffi 3.3.*
- libiconv 1.15.*
- libpng 1.6.37.*
- libtiff 4.2.0.*
- libuv 1.40.0.*
- libwebp-base 1.2.0.*
- libxcb 1.13.*
- libxml2 2.9.12.*
- lz4-c 1.9.3.*
- matplotlib 3.4.2.*
- matplotlib-base 3.4.2.*
- mkl 2021.2.0.*
- mkl-service 2.3.0.*
- mkl_fft 1.3.0.*
- mkl_random 1.2.1.*
- mypy_extensions 0.4.3.*
- networkx 2.6.2.*
- ninja 1.10.2.*
- numpy 1.20.2.*
- numpy-base 1.20.2.*
- olefile 0.46.*
- openjpeg 2.3.0.*
- openssl 1.1.1l.*
- pcre 8.45.*
- pillow 8.3.1.*
- pip 21.2.4.*
- pthread-stubs 0.4.*
- pycparser 2.20.*
- pymongo 3.11.4.*
- pyopenssl 20.0.1.*
- pyparsing 2.4.7.*
- pyqt 5.9.2.*
- pysocks 1.7.1.*
- python 3.7.10.*
- python-dateutil 2.8.1.*
- python-jsonrpc-server 0.3.4.*
- python_abi 3.7.*
- pytorch 1.10.1.*
- qt 5.9.7.*
- regex 2021.7.6.*
- setuptools 52.0.0.*
- sip 4.19.8.*
- six 1.16.0.*
- sqlite 3.36.0.*
- tk 8.6.10.*
- toml 0.10.2.*
- torchvision 0.11.2.*
- tornado 6.1.*
- typed-ast 1.4.3.*
- typing_extensions 3.10.0.0.*
- ujson 1.35.*
- whichcraft 0.6.1.*
- xorg-libxau 1.0.9.*
- xorg-libxdmcp 1.1.3.*
- xz 5.2.5.*
- yaml 0.2.5.*
- zipp 3.5.0.*
- zlib 1.2.11.*
- zstd 1.4.9.*