from setuptools import setup, find_packages

setup(
    name='PytorchTemplate',
    version='0.5.0',
    description='Annotation tool for Medical Image',
    author='Qi Chang',
    author_email='tommy.qichang@gmail.com',
    license='GPL',
    home_page='',
    packages=find_packages(),
    include_package_data=True,
    install_requires=[
        'h5py>=2,<3',
        'pydicom>=1,<2',
        'SimpleITK>=1,<2',
        'numpy>=1,<2',
        'requests>=2,<3',
        'scipy>=1,<2',
        'torch==1.3.1',
        'pytest==5.3.1',
        'pytest-html>=1.22.1',
        'pytest-cov==2.8.1',
        'pytest-mock==1.13.0',
        'nibabel==2.5.1',
        'torchvision==0.4.2',
        'pandas==0.23.0',
        'matplotlib>=3.0.0',
        'scikit-image>=0.15.0',
        'plotly==4.4.1',
        'jupyter==1.0.0',
        'pylama==7.7.1',
        'tqdm==4.40.2',
        'parse==1.12.1',
        'stringcase==1.2.0',
        'pyunpack==0.1.2',
        'patool==1.12',
        'tensorboard>=1.14.0',
        'scikit-learn==0.22',
        'imbalanced-learn==0.6.1',
        'opencv-python==4.1.2.30'
    ]
)