DeepRegFromMain20200714/setup.py at master · agrimwood/DeepRegFromMain20200714 · 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
from setuptools import setup

setup(
    name="deepreg",
    version="0.1.5",
    description="Registration with Deep Learning",
    author="Yunguan Fu",
    packages=["deepreg"],
    zip_safe=False,
    install_requires=[
        "h5py",
        "numpy>=1.16",
        "nibabel",
        "pyyaml",
        "matplotlib",
        "argparse",
        "tqdm",
        "tensorflow==2.2",
        "pytest>=4.6",
        "pytest-cov",
        "pytest-dependency",
        "pre-commit",
        "seed-isort-config",
        "isort",
        "black",
        "flake8",
        "simple_http_server",
        "testfixtures",
    ],
    entry_points={
        "console_scripts": [
            "train=deepreg.train:main",
            "predict=deepreg.predict:main",
            "gen_tfrecord=deepreg.gen_tfrecord:main",
        ]
    },
)