-
Notifications
You must be signed in to change notification settings - Fork 1
/
setup.py
55 lines (50 loc) · 1.3 KB
/
setup.py
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
47
48
49
50
51
52
53
54
55
import os
from setuptools import find_packages, setup
__version__ = "0.1"
if "VERSION" in os.environ:
BUILD_NUMBER = os.environ["VERSION"].rsplit(".", 1)[-1]
else:
BUILD_NUMBER = os.environ.get("BUILD_NUMBER", "dev")
dependencies = [
"click>=7.0",
"numpy>=1.17.0,<2.0",
"torch==1.5.0",
"torchvision==0.6.0",
"opencv-python>=4.1.1.26,<5.0",
"tensorboard==2.3.0",
"scikit-learn>=0.21.0,<1.0",
"scikit-image",
"git-python>=1.0.3",
"gin-config==0.3.0",
"torchsummary>=1.5.1",
"xlsxwriter>=1.2.9",
"nibabel",
"nilearn",
"black",
"seaborn",
]
setup(
name="visualizer",
version="{0}.{1}".format(__version__, BUILD_NUMBER),
description="Visualizing CNNs for Semantic Segmentation of BraTS 2019 Dataset",
author="Deepan Chakravarthi Padmanabhan",
install_requires=dependencies,
packages=find_packages(),
zip_safe=False,
entry_points=dict(
console_scripts=[
"visualizer_train=visualizer.train:train",
"visualizer_evaluate=visualizer.evaluate.evaluate:evaluate",
]
),
data_files=[
(
"visualizer_train_config",
[
"config/train.gin",
"config/evaluate.gin",
],
)
],
python_requires=">=3.6.9,<3.9",
)