xt-cvdata/setup.py at master · XtractTech/xt-cvdata · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
import setuptools, os

PACKAGE_NAME = 'xt-cvdata'
VERSION = '0.8.0'
AUTHOR = 'Xtract AI'
EMAIL = 'info@xtract.ai'
DESCRIPTION = 'Utilities for building and working with computer vision datasets'
GITHUB_URL = 'https://github.com/XtractTech/xt-cvdata'

parent_dir = os.path.dirname(os.path.realpath(__file__))

with open(f'{parent_dir}/README.md', 'r') as f:
    long_description = f.read()

setuptools.setup(
    name=PACKAGE_NAME,
    version=VERSION,
    author=AUTHOR,
    author_email=EMAIL,
    description=DESCRIPTION,
    long_description=long_description,
    long_description_content_type='text/markdown',
    url=GITHUB_URL,
    packages=[
        'xt_cvdata',
        'xt_cvdata.apis',
        'xt_cvdata.datasets',
        'xt_cvdata.transforms'
    ],
    provides=['xt_cvdata'],
    classifiers=[
        "Programming Language :: Python :: 3",
        "Operating System :: OS Independent",
    ],
    install_requires=[
        'numpy',
        'pandas',
        'torch',
        'torchvision',
        'h5py',
        'pillow',
        'matplotlib',
        'tqdm',
        'pycocotools',
        'albumentations'
    ],
)