-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
27 lines (25 loc) · 900 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
from setuptools import find_packages
from setuptools import setup
REQUIRED_PACKAGES = ['Keras==2.1.6',
'h5py==2.7.1',
'Pillow>=2.0.0',
'google-apitools==0.5.23',
'google-cloud-storage>=1.10',
'opencv-python>=3.4.2',
# 'tensorflow==1.9',
# 'tensorflow-gpu==1.12',
'gitpython==2.1.11',
'pypng==0.0.18',
'imgaug==0.2.6',
'parsy==1.2.0',
'google-cloud-logging==1.8.0']
#'sklearn==0.19.1'
#]
setup(
name='trainer',
version='0.3',
install_requires=REQUIRED_PACKAGES,
packages=find_packages(exclude=("backgrounds",)),
include_package_data=True,
description='Keras trainer application'
)