/
setup.py
40 lines (34 loc) · 1.19 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
from __future__ import print_function
from setuptools import find_packages, setup
def get_long_description():
with open('README.md') as f:
long_description = f.read()
try:
import github2pypi
return github2pypi.replace_url(
slug='IanTaehoonYoo/semantic-segmentation-pytorch', content=long_description
)
except Exception:
return long_description
setup(name="seg_torch",
version="0.1.7",
description="Semantic Segmentation with Pytorch",
long_description=get_long_description(),
long_description_content_type='text/markdown',
author="Ian Yoo",
author_email='thyoostar@gmail.com',
platforms=["any"],
license="MIT",
url="https://github.com/IanTaehoonYoo/semantic-segmentation-pytorch",
packages=find_packages(exclude=["segmentation/test/dataset"]),
install_requires=[
"torch>=1.5.0",
"torchvision>=0.5.0",
"tensorboardX>=2.0"
"opencv-python",
"tqdm"],
extras_require={
"tensorflow": ["tensorflow"], #this is to provide backbone models.
},
classifiers=['License :: OSI Approved :: MIT License']
)