optical-flow-python/setup.py at main · jordanshivers/optical-flow-python · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
from setuptools import setup, find_packages

setup(
    name='optical_flow',
    version='1.1.0',
    description='Optical flow estimation: Python reimplementation of Sun, Roth & Black (CVPR 2010)',
    author='Original: Deqing Sun, Stefan Roth, Michael J. Black; Python port',
    license='Research Use Only - See LICENSE file',
    packages=find_packages(include=['optical_flow', 'optical_flow.*',
                                    'flow_fast', 'flow_fast.*']),
    python_requires='>=3.9',
    install_requires=[
        'numpy>=1.21',
        'scipy>=1.7',
        'matplotlib>=3.4',
        'Pillow>=8.0',
        'scikit-image>=0.19',
    ],
    extras_require={
        'fast': [
            'numba>=0.58',
            'opencv-python>=4.5',
        ],
        'cholmod': [
            'scikit-sparse>=0.4.8',
        ],
        'deep': [
            'torch>=1.6',
            'torchvision>=0.10',
            'timm',
            'safetensors',
            'huggingface_hub',
            'gdown',
        ],
        'dev': [
            'pytest>=7.0',
            'jupyter',
        ],
    },
    package_data={
        '': ['data/**/*'],
    },
    include_package_data=True,
    test_suite='tests',
    tests_require=['pytest>=7.0'],
)