Internal change

PiperOrigin-RevId: 320988288

research/object_detection/README.md

@@ -54,21 +54,16 @@ Note: The models we provide in [TF2 Zoo](g3doc/tf2_detection_zoo.md) and
 [TF1 Zoo](g3doc/tf1_detection_zoo.md) are specific to the TensorFlow major
 version and are not interoperable.
 
-Please select one of the links below for TensorFlow version-specific
+Please select one of the two links below for TensorFlow version specific
 documentation of the Object Detection API:
 
 <!-- mdlint off(WHITESPACE_LINE_LENGTH) -->
-### Tensorflow 2.x
-  *   <a href='g3doc/tf2.md'>
-        Object Detection API TensorFlow 2</a><br>
-  *   <a href='g3doc/tf2_detection_zoo.md'>
-        TensorFlow 2 Model Zoo</a><br>
-
-### Tensorflow 1.x
-  *   <a href='g3doc/tf1.md'>
-        Object Detection API TensorFlow 1</a><br>
-  *   <a href='g3doc/tf1_detection_zoo.md'>
-        TensorFlow 1 Model Zoo</a><br>
+
+
+| [![Object Detection API TensorFlow 2](https://img.shields.io/badge/Object%20Detection%20API-TensorFlow%202-orange)](g3doc/tf2.md) | [![TensorFlow 2 Model Zoo](https://img.shields.io/badge/Model%20Zoo-TensorFlow%202-Orange)](g3doc/tf2_detection_zoo.md) |
+|---|---|
+| [![Object Detection API TensorFlow 1](https://img.shields.io/badge/Object%20Detection%20API-TensorFlow%201-orange)](g3doc/tf1.md) | [![TensorFlow 1 Model Zoo](https://img.shields.io/badge/Model%20Zoo-TensorFlow%201-Orange)](g3doc/tf1_detection_zoo.md) |
+
 <!-- mdlint on -->
 
 ## Whats New

research/object_detection/g3doc/tf1.md

@@ -73,8 +73,6 @@ the [Model Zoo](tf1_detection_zoo.md).
       Supported object detection evaluation protocols</a><br>
 *   <a href='tpu_compatibility.md'>
       TPU compatible detection pipelines</a><br>
-*   <a href='tf1_training_and_evaluation.md'>
-      Training and evaluation guide (CPU, GPU, or TPU)</a><br>
 
 ## Extras:
 

research/object_detection/g3doc/tf2.md

@@ -80,5 +80,3 @@ We provide a large collection of models that are trained on COCO 2017 in the
       Supported object detection evaluation protocols</a><br>
 *   <a href='tpu_compatibility.md'>
       TPU compatible detection pipelines</a><br>
-*   <a href='tf2_training_and_evaluation.md'>
-      Training and evaluation guide (CPU, GPU, or TPU)</a><br>
\ No newline at end of file

research/object_detection/model_lib_v2.py

@@ -413,7 +413,7 @@ def train_loop(
     train_steps=None,
     use_tpu=False,
     save_final_config=False,
-    checkpoint_every_n=1000,
+    checkpoint_every_n=5000,
     checkpoint_max_to_keep=7,
     **kwargs):
   """Trains a model using eager + functions.
@@ -854,6 +854,7 @@ def eval_continuously(
     checkpoint_dir=None,
     wait_interval=180,
     timeout=3600,
+    eval_index=None,
     **kwargs):
   """Run continuous evaluation of a detection model eagerly.
 
@@ -883,6 +884,8 @@ def eval_continuously(
       new checkpoint.
     timeout: The maximum number of seconds to wait for a checkpoint. Execution
       will terminate if no new checkpoints are found after these many seconds.
+    eval_index: int, optional If give, only evaluate the dataset at the given
+      index.
 
     **kwargs: Additional keyword arguments for configuration override.
   """
@@ -936,6 +939,11 @@ def eval_continuously(
         model=detection_model)
     eval_inputs.append((eval_input_config.name, next_eval_input))
 
+  if eval_index is not None:
+    eval_inputs = [eval_inputs[eval_index]]
+    tf.logging.info('eval_index selected - {}'.format(
+        eval_inputs))
+
   global_step = tf.compat.v2.Variable(
       0, trainable=False, dtype=tf.compat.v2.dtypes.int64)