forked from google/automl
-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
41 lines (39 loc) · 1.48 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
import setuptools
setuptools.setup(
name = 'efficientdet',
version = '0.0.0',
description = '',
url = 'https://github.com/google/automl',
author = 'Google',
author_email = '',
maintainer = '',
maintainer_email = '',
packages = setuptools.find_packages(),
install_requires = [
'lxml>=4.6.1',
'absl-py>=0.7.1',
'matplotlib>=3.0.3',
'numpy>=1.16.4',
'Pillow>=6.0.0',
'PyYAML>=5.1',
'six>=1.12.0',
'tensorflow>=2.3.0',
'tensorflow-addons>=0.11.2',
'neural-structured-learning>=1.3.1',
'tensorflow-model-optimization>=0.5',
'Cython>=0.29.13',
'pycocotools'
],
entry_points = {
'console_scripts': [
'efficientdet-tf-model-inspect=efficientdet.model_inspect:launcher',
'efficientdet-tf-main=efficientdet.main:launcher',
'efficientdet-keras-train=efficientdet.keras.train:launcher',
'efficientdet-keras-inspector=efficientdet.keras.inspector:launcher',
'efficientdet-keras-eval=efficientdet.keras.eval:launcher',
'efficientdet-create-coco-tfrecord=efficientdet.dataset.create_coco_tfrecord:launcher',
'efficientdet-create-pascal-tfrecord=efficientdet.dataset.create_pascal_tfrecord:launcher',
'efficientdet-inspect-tfrecord=efficientdet.dataset.inspect_tfrecords:launcher'
],
}
)