Skip to content

GitLab

Commit 03a6d6c8 authored by Vivek Rathod's avatar Vivek Rathod Committed by TF Object Detection Team
Browse files

Pin to tf-models-official 2.2.0 since 2.3.0 pulls TF 2.3 which is incompatible with numpy 1.19.1 

PiperOrigin-RevId: 324352004
parent 007a619a
Show whitespace changes
Inline Side-by-side
Showing with 17 additions and 5 deletions
research/object_detection/packages/tf2/setup.py
View file @ 03a6d6c8
...@@ -6,11 +6,23 @@ from setuptools import setup ...@@ -6,11 +6,23 @@ from setuptools import setup
# Note: adding apache-beam to required packages causes conflict with # Note: adding apache-beam to required packages causes conflict with
# tf-models-offical requirements. These packages request for incompatible # tf-models-offical requirements. These packages request for incompatible
# oauth2client package. # oauth2client package.
REQUIRED_PACKAGES = ['avro-python3==1.8.1', 'apache-beam', REQUIRED_PACKAGES = [
'pillow', 'lxml', # Required for apache-beam with PY3
'matplotlib', 'Cython', 'contextlib2', 'avro-python3==1.8.1',
'tf-slim', 'six', 'pycocotools', 'scipy', 'pandas', 'apache-beam',
'tf-models-official'] 'pillow',
'lxml',
'matplotlib',
'Cython',
'contextlib2',
'tf-slim',
'six',
'pycocotools',
'scipy',
'pandas',
# Required to avoid Numpy 1.19.1 conflict with TF 2.3
'tf-models-official==2.2.2'
]
setup( setup(
name='object_detection', name='object_detection',
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment