setup.py

from setuptools import setup, find_packages

setup(name='deepexplain',
      version='0.3',
      description='Perturbation and gradient-based methods for Deep Network interpretability',
      url='https://github.com/marcoancona/DeepExplain',
      author='Marco Ancona (ETH Zurich)',
      author_email='marco.ancona@inf.ethz.ch',
      license='MIT',
      packages=find_packages(),
      install_requires=[
            'scipy',
            'matplotlib',
            'scikit-image'
      ],
      extras_require={
            "tf": ["tensorflow>=1.0.0"],
            "tf_gpu": ["tensorflow-gpu>=1.0.0"],
      },
      zip_safe=False,
      test_suite='nose.collector',
      tests_require=['nose'],
      )