diff --git a/requirements.txt b/requirements.txt index 7939681..902fef5 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,5 +1,4 @@ numpy>=1.14.5 torch>=0.4.0 scikit-image>=0.14.0 -SimpleITK>=2.0.2 --e git+https://github.com/MIC-DKFZ/batchgenerators#egg=batchgenerators +SimpleITK==2.0.2 diff --git a/setup.py b/setup.py index 8e67f25..69fc4c9 100755 --- a/setup.py +++ b/setup.py @@ -1,5 +1,7 @@ from setuptools import setup, find_packages +requirements= open('requirements.txt').read().split('\n') + setup(name='HD_BET', version='1.0', description='Tool for brain extraction', @@ -9,12 +11,7 @@ author_email='f.isensee@dkfz.de', license='Apache 2.0', zip_safe=False, - install_requires=[ - 'numpy', - 'torch>=0.4.1', - 'scikit-image', - 'SimpleITK' - ], + install_requires=requirements, scripts=['HD_BET/hd-bet'], packages=find_packages(include=['HD_BET']), classifiers=[