-
Notifications
You must be signed in to change notification settings - Fork 2
/
setup.py
37 lines (36 loc) · 1.26 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
from setuptools import setup, find_namespace_packages
setup(name='docr',
packages=find_namespace_packages(include=["docr", "docr.*"]),
version='0.0.1',
description='Domain specific Convolution and high frequency Reconstruction (DoCR) '
'based UDA for medical image segmentation.',
author='Shishuai Hu',
author_email='sshu@mail.nwpu.edu.cn',
license='Apache License Version 2.0, January 2004',
install_requires=[
"torch>=1.6.0a",
"tqdm",
"setuptools==59.5.0",
"dicom2nifti",
"scikit-image>=0.14",
"medpy",
"scipy",
"batchgenerators>=0.21",
"numpy",
"sklearn",
"SimpleITK",
"pandas",
"requests",
"hiddenlayer", "graphviz", "IPython",
"nibabel", 'tifffile',
"tensorboard"
],
entry_points={
'console_scripts': [
'DoCR_train = docr.training.run_training:main',
'DoCR_test = docr.inference.run_inference:main',
],
},
keywords=['deep learning', 'image segmentation', 'medical image analysis',
'medical image segmentation', 'unsupervised domain adaptation']
)