-
Notifications
You must be signed in to change notification settings - Fork 8
/
setup.py
28 lines (27 loc) · 904 Bytes
/
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
from setuptools import setup, find_packages
setup(name='speaksee',
version='0.0.1',
description='PyTorch utilities and models for Visual-Semantic tasks',
url='http://github.com/aimagelab/speaksee',
author='Lorenzo Baraldi, Marcella Cornia',
author_email='lorenzo.baraldi@unimore.it',
packages=find_packages(exclude=('test', )),
package_data={'speaksee': ['evaluation/stanford-corenlp-3.4.1.jar']},
include_package_data=True,
install_requires=[
'torch>=0.4.0',
'torchvision',
'numpy',
'h5py',
'tqdm',
'requests',
'pycocotools',
'matplotlib',
'pathos',
],
classifiers=[
"Programming Language :: Python :: 3",
"License :: OSI Approved :: BSD License",
"Operating System :: OS Independent",
],
zip_safe=False)