Merge pull request #10338 from srihari-humbarwadi:readme

PiperOrigin-RevId: 413033276

official/vision/detection/utils/object_detection/preprocessor.py → official/vision/utils/object_detection/preprocessor.py

@@ -39,11 +39,9 @@ we pass it to the functions. At the end of the preprocess we expand the image
 back to rank 4.
 """
 
-import tensorflow as tf
-
 import numpy as np
-
-from official.vision.detection.utils.object_detection import box_list
+import tensorflow as tf
+from official.vision.utils.object_detection import box_list
 
 
 def _flip_boxes_left_right(boxes):

official/vision/detection/utils/object_detection/target_assigner.py → official/vision/utils/object_detection/target_assigner.py

@@ -33,8 +33,8 @@ images must be handled externally.
 
 import tensorflow as tf
 
-from official.vision.detection.utils.object_detection import box_list
-from official.vision.detection.utils.object_detection import shape_utils
+from official.vision.utils.object_detection import box_list
+from official.vision.utils.object_detection import shape_utils
 
 KEYPOINTS_FIELD_NAME = 'keypoints'
 

official/vision/detection/utils/object_detection/visualization_utils.py → official/vision/utils/object_detection/visualization_utils.py

@@ -34,8 +34,8 @@ import PIL.ImageFont as ImageFont
 import six
 import tensorflow as tf
 
-from official.vision.detection.utils import box_utils
-from official.vision.detection.utils.object_detection import shape_utils
+from official.vision.beta.ops import box_ops
+from official.vision.utils.object_detection import shape_utils
 
 _TITLE_LEFT_MARGIN = 10
 _TITLE_TOP_MARGIN = 10
@@ -107,8 +107,8 @@ def visualize_images_with_bounding_boxes(images, box_outputs, step,
   image_shape = tf.shape(images[0])
   image_height = tf.cast(image_shape[0], tf.float32)
   image_width = tf.cast(image_shape[1], tf.float32)
-  normalized_boxes = box_utils.normalize_boxes(box_outputs,
-                                               [image_height, image_width])
+  normalized_boxes = box_ops.normalize_boxes(box_outputs,
+                                             [image_height, image_width])
 
   bounding_box_color = tf.constant([[1.0, 1.0, 0.0, 1.0]])
   image_summary = tf.image.draw_bounding_boxes(

orbit/standard_runner.py

@@ -265,7 +265,10 @@ class StandardEvaluator(runner.AbstractEvaluator, metaclass=abc.ABCMeta):
 
     Args:
       eval_dataset: A `tf.nest`-compatible structure of `tf.data.Dataset` or
-        `DistributedDataset`.
+        `DistributedDataset`. On TPUs, if users want to exaust the dataset
+        without specifying number of eval steps, it is recommended to set
+        `drop_remainder=False` when batching the dataset, so the infrastructure
+        can handle the last partial batch properly.
       options: An `orbit.StandardEvaluatorOptions` instance.
     """
     options = options or StandardEvaluatorOptions()

official/vision/keras_cv/LICENSE → tensorflow_models/LICENSE

-Copyright 2020 The TensorFlow Authors. All rights reserved.
+Copyright 2015 The TensorFlow Authors.  All rights reserved.
 
                                  Apache License
                            Version 2.0, January 2004
@@ -200,4 +200,4 @@ Copyright 2020 The TensorFlow Authors. All rights reserved.
    distributed under the License is distributed on an "AS IS" BASIS,
    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
    See the License for the specific language governing permissions and
-   limitations under the License.
+   limitations under the License.
\ No newline at end of file

tensorflow_models/__init__.py

+# Copyright 2021 The TensorFlow Authors. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""TensorFlow Models Libraries."""
+# pylint: disable=wildcard-import
+from tensorflow_models import nlp
+from tensorflow_models import vision
+
+from official import core
+from official.modeling import hyperparams
+from official.modeling import optimization
+from official.modeling import tf_utils

tensorflow_models/nlp/__init__.py

+# Copyright 2021 The TensorFlow Authors. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""TensorFlow Models NLP Libraries."""
+from official.nlp import tasks
+from official.nlp.modeling import *

tensorflow_models/tensorflow_models_test.py

+# Copyright 2021 The TensorFlow Authors. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""Tests for tensorflow_models imports."""
+
+import tensorflow as tf
+import tensorflow_models as tfm
+
+
+class TensorflowModelsTest(tf.test.TestCase):
+
+  def testVisionImport(self):
+    _ = tfm.vision.layers.SqueezeExcitation(
+        in_filters=8, out_filters=4, se_ratio=1)
+    _ = tfm.vision.configs.image_classification.Losses()
+
+  def testNLPImport(self):
+    _ = tfm.nlp.layers.TransformerEncoderBlock(
+        num_attention_heads=2, inner_dim=10, inner_activation='relu')
+    _ = tfm.nlp.tasks.TaggingTask(params=tfm.nlp.tasks.TaggingConfig())
+
+  def testCommonImports(self):
+    _ = tfm.hyperparams.Config()
+    _ = tfm.optimization.LinearWarmup(
+        after_warmup_lr_sched=0.0, warmup_steps=10, warmup_learning_rate=0.1)
+
+
+if __name__ == '__main__':
+  tf.test.main()

tensorflow_models/vision/__init__.py

+# Copyright 2021 The TensorFlow Authors. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""TensorFlow Models Vision Libraries."""
+from official.vision.beta import configs
+from official.vision.beta.modeling import *