setup.py

import setuptools
from setuptools import setup

install_deps = ['numpy', 'scipy', 'natsort', 'tabulate',
                'tifffile', 'tqdm', 'numba',
                'torch>=1.6', 'scikit-image', 'tensorflow', 'tensorboardX',
                'opencv-python-headless', 'pyqtgraph==0.11.0rc0', 'pyqt5',]

try:
    import torch
    a = torch.ones(2, 3)
    version = int(torch.__version__[2])
    if version >= 6:
        install_deps.remove('torch')
except:
    pass

with open("README.md", "r", encoding='utf-8') as fh:
    long_description = fh.read()

setup(
    name="scellseg",
    version='0.1.7',
    license="BSD",
    author="Dejin Xun & Deheng Chen",
    author_email="xundejin@zju.edu.cn",
    description="a specializable cell segmentation tool",
    long_description=long_description,
    long_description_content_type="text/markdown",
    url="https://github.com/cellimnet/scellseg-publish",
    packages=setuptools.find_packages(),
    install_requires = install_deps,
    include_package_data=True,
    classifiers=(
        "Programming Language :: Python :: 3",
        "License :: OSI Approved :: BSD License",
        "Operating System :: OS Independent",
    ),
     entry_points = {
        'console_scripts': [
          'scellseg = scellseg.__main__:main']
     }
)