dannce/setup.py at master · jessedmarshall/dannce · 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
"""Setup file for dannce."""
from setuptools import setup, find_packages

setup(
    name="dannce",
    version="1.1.0",
    packages=find_packages(),
    install_requires=[
        "six",
        "pyyaml",
        "imageio==2.8.0",
        "imageio-ffmpeg",
        "scikit-image",
        "matplotlib",
        "opencv-python",
        "tensorflow==2.3.1",
        "torch",
        "numpy==1.18"
    ],
    entry_points={
        "console_scripts": [
            "dannce-train = dannce.cli:dannce_train_cli",
            "dannce-predict = dannce.cli:dannce_predict_cli",
            "com-train = dannce.cli:com_train_cli",
            "com-predict = dannce.cli:com_predict_cli",
        ]
    },
)