setup.py

# -*- coding: utf-8 -*-
# -----------------------------------------------------------------------------
# Copyright (c) Edgar Andrés Margffoy-Tuay, Emilio Botero and Juan Camilo Pérez
#
# Licensed under the terms of the MIT License
# (see LICENSE.txt for details)
# -----------------------------------------------------------------------------
"""Setup script for DMN."""

# Standard library imports
import os
import ast
import shutil
import os.path as osp

# Third party imports
from setuptools import setup, find_packages

PACKAGE = 'dmn_pytorch'
HERE = os.path.abspath(os.path.dirname(__file__))


def get_version(module=PACKAGE):
    """Get version."""
    with open(os.path.join(HERE, module, '__init__.py'), 'r') as f:
        data = f.read()
    lines = data.split('\n')
    for line in lines:
        if line.startswith('VERSION_INFO'):
            version_tuple = ast.literal_eval(line.split('=')[-1].strip())
            version = '.'.join(map(str, version_tuple))
            break
    return version


def get_description():
    """Get long description."""
    with open(os.path.join(HERE, 'README.md'), 'r') as f:
        data = f.read()
    return data

REQUIREMENTS = []

setup(
    name=PACKAGE,
    version=get_version(),
    keywords=['Computer Vision', 'Segmentation', 'NLP'],
    url='https://github.com/andfoy/query-objseg',
    license='MIT',
    author='Edgar Andrés Margffoy Tuay',
    author_email='andfoy@gmail.com',
    description='Semantic Segmentation based on Natural Language Queries',
    long_description=get_description(),
    packages=find_packages(exclude=['contrib', 'docs', 'tests*']),
    install_requires=REQUIREMENTS,
    include_package_data=True,
    classifiers=[
        'Development Status :: 4 - Beta',
        'Intended Audience :: Developers',
        'Intended Audience :: Science/Research',
        'License :: OSI Approved :: MIT License',
        'Operating System :: POSIX :: Linux',
        'Programming Language :: Python :: 3.5',
        'Programming Language :: Python :: 3.6'])