/
setup.py
54 lines (45 loc) · 1.16 KB
/
setup.py
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
48
49
50
51
52
53
54
import os
import io
import re
from setuptools import setup, find_packages
def read(*names, **kwargs):
with io.open(os.path.join(os.path.dirname(__file__), *names),
encoding=kwargs.get("encoding", "utf8")) as fp:
return fp.read()
readme = read('README.rst')
VERSION = '0.1.1'
requirements = [
"torch==1.0.0",
"torchvision==0.2.1",
"tqdm>=4.19.5",
"tensorboardX>=1.1",
"ipython>=6.2.1",
"opencv-python>=3.4.0.12",
"Pillow>=5.0.0",
"numpy>=1.14.0",
"scikit-image>=0.13.1",
"visdom>=0.1.8.5"
"cython==0.29.1",
"pycocotools==2.0.0",
"ConfigArgParse==0.13.0"
]
setup(
# Metadata
name='pysemseg',
version=VERSION,
author='Petko Nikolov',
author_email='py.nikolov@gmail.com',
url='https://github.com/petko-nikolov/pysemseg',
description='Pytorch library for training Deep Learning models for Semantic Segmentation',
long_description=readme,
license='BSD',
# Package info
packages=find_packages(),
zip_safe=True,
install_requires=requirements,
entry_points={
'console_scripts': [
'pysemseg-train=pysemseg.train:main'
]
}
)