Skip to content

Commit

Permalink
Set wait_interval to 0 when calling model_lib_v2.eval_continuously.
Browse files Browse the repository at this point in the history
Added setup_2.4.0.py.
  • Loading branch information
lizlooney committed Jul 11, 2021
1 parent 5f23689 commit e60fea1
Show file tree
Hide file tree
Showing 2 changed files with 48 additions and 1 deletion.
2 changes: 1 addition & 1 deletion research/object_detection/model_main_tf2.py
Original file line number Diff line number Diff line change
Expand Up @@ -87,7 +87,7 @@ def main(unused_argv):
sample_1_of_n_eval_on_train_examples=(
FLAGS.sample_1_of_n_eval_on_train_examples),
checkpoint_dir=FLAGS.checkpoint_dir,
wait_interval=300, timeout=FLAGS.eval_timeout)
wait_interval=0, timeout=FLAGS.eval_timeout)
else:
if FLAGS.use_tpu:
# TPU is automatically inferred if tpu_name is None and
Expand Down
47 changes: 47 additions & 0 deletions research/object_detection/packages/tf2/setup_2.4.0.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,47 @@
"""Setup script for object_detection with TF2.0."""
import os
from setuptools import find_packages
from setuptools import setup

# Note: adding apache-beam to required packages causes conflict with
# tf-models-offical requirements. These packages request for incompatible
# oauth2client package.
REQUIRED_PACKAGES = [
# Required for apache-beam with PY3
'avro-python3',
'apache-beam',
'pillow',
'lxml',
'matplotlib',
'Cython',
'contextlib2',
'tf-slim',
'six',
'pycocotools',
'lvis',
'scipy',
'pandas',
'tf-models-official==2.4.0',
'tensorboard==2.4.0',
'tensorflow==2.4.0',
'tensorflow-estimator==2.4.0'
]

setup(
name='object_detection',
version='0.1',
install_requires=REQUIRED_PACKAGES,
include_package_data=True,
packages=(
[p for p in find_packages() if p.startswith('object_detection')] +
find_packages(where=os.path.join('.', 'slim'))),
package_dir={
'datasets': os.path.join('slim', 'datasets'),
'nets': os.path.join('slim', 'nets'),
'preprocessing': os.path.join('slim', 'preprocessing'),
'deployment': os.path.join('slim', 'deployment'),
'scripts': os.path.join('slim', 'scripts'),
},
description='Tensorflow Object Detection Library',
python_requires='>3.6',
)

0 comments on commit e60fea1

Please sign in to comment.