tf-models-nightly 2.17.0.dev20240617__py2.py3-none-any.whl → 2.20.0.dev20251205__py2.py3-none-any.whl

official/vision/modeling/video_classification_model_test.py

@@ -1,4 +1,4 @@
-# Copyright 2024 The TensorFlow Authors. All Rights Reserved.
+# Copyright 2025 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.
@@ -14,7 +14,6 @@
 
 """Tests for video classification network."""
 
-# Import libraries
 from absl.testing import parameterized
 import numpy as np
 import tensorflow as tf, tf_keras

official/vision/ops/__init__.py

@@ -1,4 +1,4 @@
-# Copyright 2024 The TensorFlow Authors. All Rights Reserved.
+# Copyright 2025 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.

official/vision/ops/anchor.py

@@ -1,4 +1,4 @@
-# Copyright 2024 The TensorFlow Authors. All Rights Reserved.
+# Copyright 2025 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.
@@ -18,8 +18,6 @@ import collections
 import math
 from typing import Dict, Optional, Tuple
 
-# Import libraries
-
 import tensorflow as tf, tf_keras
 
 from official.vision.ops import box_matcher

official/vision/ops/anchor_generator.py

@@ -1,4 +1,4 @@
-# Copyright 2024 The TensorFlow Authors. All Rights Reserved.
+# Copyright 2025 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.

official/vision/ops/anchor_generator_test.py

@@ -1,4 +1,4 @@
-# Copyright 2024 The TensorFlow Authors. All Rights Reserved.
+# Copyright 2025 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.

official/vision/ops/anchor_test.py

@@ -1,4 +1,4 @@
-# Copyright 2024 The TensorFlow Authors. All Rights Reserved.
+# Copyright 2025 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.
@@ -14,7 +14,6 @@
 
 """Tests for anchor.py."""
 
-# Import libraries
 from absl.testing import parameterized
 import numpy as np
 import tensorflow as tf, tf_keras

official/vision/ops/augment.py

@@ -1,4 +1,4 @@
-# Copyright 2024 The TensorFlow Authors. All Rights Reserved.
+# Copyright 2025 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.
@@ -844,19 +844,7 @@ def color(image: tf.Tensor, factor: float) -> tf.Tensor:
 
 def contrast(image: tf.Tensor, factor: float) -> tf.Tensor:
   """Equivalent of PIL Contrast."""
-  degenerate = tf.image.rgb_to_grayscale(image)
-  # Cast before calling tf.histogram.
-  degenerate = tf.cast(degenerate, tf.int32)
-
-  # Compute the grayscale histogram, then compute the mean pixel value,
-  # and create a constant image size of that value.  Use that as the
-  # blending degenerate target of the original image.
-  hist = tf.histogram_fixed_width(degenerate, [0, 255], nbins=256)
-  mean = tf.reduce_sum(tf.cast(hist, tf.float32)) / 256.0
-  degenerate = tf.ones_like(degenerate, dtype=tf.float32) * mean
-  degenerate = tf.clip_by_value(degenerate, 0.0, 255.0)
-  degenerate = tf.image.grayscale_to_rgb(tf.cast(degenerate, tf.uint8))
-  return blend(degenerate, image, factor)
+  return tf.image.adjust_contrast(image, factor)
 
 
 def brightness(image: tf.Tensor, factor: float) -> tf.Tensor:
@@ -2709,8 +2697,8 @@ class MixupAndCutmix:
 
   @staticmethod
   def _sample_from_beta(alpha, beta, shape):
-    sample_alpha = tf.random.gamma(shape, 1., beta=alpha)
-    sample_beta = tf.random.gamma(shape, 1., beta=beta)
+    sample_alpha = tf.random.gamma(shape, alpha, beta=1.0)
+    sample_beta = tf.random.gamma(shape, beta, beta=1.0)
     return sample_alpha / (sample_alpha + sample_beta)
 
   def _cutmix(self, images: tf.Tensor,

official/vision/ops/augment_test.py

@@ -1,4 +1,4 @@
-# Copyright 2024 The TensorFlow Authors. All Rights Reserved.
+# Copyright 2025 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.

official/vision/ops/box_matcher.py

@@ -1,4 +1,4 @@
-# Copyright 2024 The TensorFlow Authors. All Rights Reserved.
+# Copyright 2025 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.

official/vision/ops/box_matcher_test.py

@@ -1,4 +1,4 @@
-# Copyright 2024 The TensorFlow Authors. All Rights Reserved.
+# Copyright 2025 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.

official/vision/ops/box_ops.py

@@ -1,4 +1,4 @@
-# Copyright 2024 The TensorFlow Authors. All Rights Reserved.
+# Copyright 2025 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.
@@ -14,7 +14,6 @@
 
 """Box related ops."""
 
-# Import libraries
 import numpy as np
 import tensorflow as tf, tf_keras
 

official/vision/ops/iou_similarity.py

@@ -1,4 +1,4 @@
-# Copyright 2024 The TensorFlow Authors. All Rights Reserved.
+# Copyright 2025 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.

official/vision/ops/iou_similarity_test.py

@@ -1,4 +1,4 @@
-# Copyright 2024 The TensorFlow Authors. All Rights Reserved.
+# Copyright 2025 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.

official/vision/ops/mask_ops.py

@@ -1,4 +1,4 @@
-# Copyright 2024 The TensorFlow Authors. All Rights Reserved.
+# Copyright 2025 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.
@@ -17,8 +17,6 @@
 import math
 from typing import List, Tuple
 
-# Import libraries
-
 import cv2
 import numpy as np
 import tensorflow as tf, tf_keras

official/vision/ops/mask_ops_test.py

@@ -1,4 +1,4 @@
-# Copyright 2024 The TensorFlow Authors. All Rights Reserved.
+# Copyright 2025 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.
@@ -14,7 +14,6 @@
 
 """Tests for mask_ops.py."""
 
-# Import libraries
 import numpy as np
 import tensorflow as tf, tf_keras
 from official.vision.ops import mask_ops

official/vision/ops/nms.py

@@ -1,4 +1,4 @@
-# Copyright 2024 The TensorFlow Authors. All Rights Reserved.
+# Copyright 2025 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.
@@ -14,7 +14,6 @@
 
 """Tensorflow implementation of non max suppression."""
 
-# Import libraries
 import tensorflow as tf, tf_keras
 
 from official.vision.ops import box_ops

official/vision/ops/preprocess_ops.py

@@ -1,4 +1,4 @@
-# Copyright 2024 The TensorFlow Authors. All Rights Reserved.
+# Copyright 2025 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.
@@ -82,13 +82,28 @@ def normalize_image(
 ) -> tf.Tensor:
   """Normalizes the image to zero mean and unit variance.
 
-  If the input image dtype is float, it is expected to either have values in
-  [0, 1) and offset is MEAN_NORM, or have values in [0, 255] and offset is
-  MEAN_RGB.
+  This function normalizes the input image by subtracting the `offset`
+  and dividing by the `scale`.
+
+  **Important Note about Input Types and Normalization:**
+
+  * **Integer Images:** If the input `image` is an integer type (e.g., `uint8`),
+    the provided `offset` and `scale` values should be already **normalized**
+    to the range [0, 1]. This is because the function converts integer images to
+    float32 with values in the range [0, 1] before the normalization happens.
+
+  * **Float Images:** If the input `image` is a float type (e.g., `float32`),
+    the `offset` and `scale` values should be in the **same range** as the
+    image data.
+      - If the image has values in [0, 1], the `offset` and `scale` should
+        also be in [0, 1].
+      - If the image has values in [0, 255], the `offset` and `scale` should
+        also be in [0, 255].
 
   Args:
-    image: A tf.Tensor in either (1) float dtype with values in range [0, 1) or
-      [0, 255], or (2) int type with values in range [0, 255].
+    image: A `tf.Tensor` in either:
+           (1) float dtype with values in range [0, 1) or [0, 255], or
+           (2) int type with values in range [0, 255].
     offset: A tuple of mean values to be subtracted from the image.
     scale: A tuple of normalization factors.
 
@@ -182,7 +197,7 @@ def resize_and_crop_image(
   2. Pad the rescaled image to the padded_size.
 
   Args:
-    image: a `Tensor` of shape [height, width, 3] representing an image.
+    image: a `Tensor` of shape [height, width, c] representing an image.
     desired_size: a `Tensor` or `int` list/tuple of two elements representing
       [height, width] of the desired actual output image size.
     padded_size: a `Tensor` or `int` list/tuple of two elements representing
@@ -201,7 +216,7 @@ def resize_and_crop_image(
       behaviour is to place it at left top corner.
 
   Returns:
-    output_image: `Tensor` of shape [height, width, 3] where [height, width]
+    output_image: `Tensor` of shape [height, width, c] where [height, width]
       equals to `output_size`.
     image_info: a 2D `Tensor` that encodes the information of the image and the
       applied preprocessing. It is in the format of
@@ -780,13 +795,27 @@ def horizontal_flip_masks(masks):
     return masks[:, :, ::-1]
 
 
+def vertical_flip_image(image):
+  """Flips image vertically."""
+  return tf.image.flip_up_down(image)
+
+
+def vertical_flip_masks(masks):
+  """Flips masks vertically. Expects rank-3 input dimensions."""
+  # For masks shape of [h, w, 1].
+  if masks.shape[-1] == 1:
+    return masks[::-1, :, :]
+  else:
+    return masks[:, ::-1, :]
+
+
 def random_horizontal_flip(
     image, normalized_boxes=None, masks=None, seed=1, prob=0.5
 ):
   """Randomly flips input image and bounding boxes and/or masks horizontally.
 
   Expects input tensors without the batch dimension; i.e. for RGB image assume
-  rank-3 input like [h, w, 3], for masks assume either [h, w, 1] or [1, h, w].
+  rank-3 input like [h, w, c], for masks assume either [h, w, 1] or [1, h, w].
 
   Args:
     image: `tf.Tensor`, the image to apply the random flip, [h, w, channels].
@@ -885,7 +914,7 @@ def random_vertical_flip(
     do_flip = tf.less(tf.random.uniform([], seed=seed), prob)
 
     image = tf.cond(
-        do_flip, lambda: tf.image.flip_up_down(image), lambda: image
+        do_flip, lambda: vertical_flip_image(image), lambda: image
     )
 
     if normalized_boxes is not None:
@@ -898,7 +927,7 @@ def random_vertical_flip(
     if masks is not None:
       masks = tf.cond(
           do_flip,
-          lambda: tf.image.flip_up_down(masks[..., None])[..., 0],
+          lambda: vertical_flip_masks(masks),
           lambda: masks,
       )
 

official/vision/ops/preprocess_ops_3d.py

@@ -1,4 +1,4 @@
-# Copyright 2024 The TensorFlow Authors. All Rights Reserved.
+# Copyright 2025 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.
@@ -339,7 +339,8 @@ def resize_smallest(frames: tf.Tensor, min_resize: int) -> tf.Tensor:
 def random_crop_resize(frames: tf.Tensor, output_h: int, output_w: int,
                        num_frames: int, num_channels: int,
                        aspect_ratio: Tuple[float, float],
-                       area_range: Tuple[float, float]) -> tf.Tensor:
+                       area_range: Tuple[float, float],
+                       seed: int = 0) -> tf.Tensor:
   """First crops clip with jittering and then resizes to (output_h, output_w).
 
   Args:
@@ -350,6 +351,7 @@ def random_crop_resize(frames: tf.Tensor, output_h: int, output_w: int,
     num_channels: Number of channels of the clip.
     aspect_ratio: Float tuple with the aspect range for cropping.
     area_range: Float tuple with the area range for cropping.
+    seed: A seed to use for the random sampling.
 
   Returns:
     A Tensor of shape [timesteps, output_h, output_w, channels] of type
@@ -367,7 +369,8 @@ def random_crop_resize(frames: tf.Tensor, output_h: int, output_w: int,
       aspect_ratio_range=aspect_ratio,
       area_range=area_range,
       max_attempts=100,
-      use_image_if_no_bounding_boxes=True)
+      use_image_if_no_bounding_boxes=True,
+      seed=seed)
   bbox_begin, bbox_size, _ = sample_distorted_bbox
   offset_y, offset_x, _ = tf.unstack(bbox_begin)
   target_height, target_width, _ = tf.unstack(bbox_size)

official/vision/ops/preprocess_ops_3d_test.py

@@ -1,4 +1,4 @@
-# Copyright 2024 The TensorFlow Authors. All Rights Reserved.
+# Copyright 2025 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.

official/vision/ops/preprocess_ops_test.py

@@ -1,4 +1,4 @@
-# Copyright 2024 The TensorFlow Authors. All Rights Reserved.
+# Copyright 2025 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.
@@ -16,8 +16,6 @@
 
 import io
 
-# Import libraries
-
 from absl.testing import parameterized
 import numpy as np
 from PIL import Image
@@ -80,11 +78,19 @@ class InputUtilsTest(parameterized.TestCase, tf.test.TestCase):
           aug_scale_max=2.0,
           output_scales=(20 / 100, 10 / 200),
       ),
+      dict(
+          testcase_name='no_jittering_with_4_channels',
+          input_size=(100, 200),
+          desired_size=(20, 10),
+          aug_scale_max=1.0,
+          output_scales=(20 / 100, 10 / 200),
+          channels=4,
+      ),
   )
   def test_resize_and_crop_image_not_keep_aspect_ratio(
-      self, input_size, desired_size, aug_scale_max, output_scales
+      self, input_size, desired_size, aug_scale_max, output_scales, channels=3
   ):
-    image = tf.convert_to_tensor(np.random.rand(*input_size, 3))
+    image = tf.convert_to_tensor(np.random.rand(*input_size, channels))
 
     resized_image, image_info = preprocess_ops.resize_and_crop_image(
         image,
@@ -95,7 +101,7 @@ class InputUtilsTest(parameterized.TestCase, tf.test.TestCase):
     )
     resized_image_shape = tf.shape(resized_image)
 
-    self.assertAllEqual([*desired_size, 3], resized_image_shape.numpy())
+    self.assertAllEqual([*desired_size, channels], resized_image_shape.numpy())
     if aug_scale_max == 1:
       self.assertNDArrayNear(
           [input_size, desired_size, output_scales, [0.0, 0.0]],
@@ -108,7 +114,7 @@ class InputUtilsTest(parameterized.TestCase, tf.test.TestCase):
       (100, 256, 128, 256, 32, 1.0, 1.0, 128, 256),
       (200, 512, 200, 128, 32, 0.25, 0.25, 224, 128),
   )
-  def test_resize_and_crop_image_rectangluar_case(
+  def test_resize_and_crop_image_rectangular_case(
       self,
       input_height,
       input_width,

official/vision/ops/sampling_ops.py

@@ -1,4 +1,4 @@
-# Copyright 2024 The TensorFlow Authors. All Rights Reserved.
+# Copyright 2025 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.
@@ -31,7 +31,6 @@ when number of examples set to True in indicator is less than batch_size.
 This is originally implemented in TensorFlow Object Detection API.
 """
 
-# Import libraries
 import tensorflow as tf, tf_keras
 
 

official/vision/ops/spatial_transform_ops.py

@@ -1,4 +1,4 @@
-# Copyright 2024 The TensorFlow Authors. All Rights Reserved.
+# Copyright 2025 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.

official/vision/ops/target_gather.py

@@ -1,4 +1,4 @@
-# Copyright 2024 The TensorFlow Authors. All Rights Reserved.
+# Copyright 2025 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.

official/vision/ops/target_gather_test.py

@@ -1,4 +1,4 @@
-# Copyright 2024 The TensorFlow Authors. All Rights Reserved.
+# Copyright 2025 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.

official/vision/registry_imports.py

@@ -1,4 +1,4 @@
-# Copyright 2024 The TensorFlow Authors. All Rights Reserved.
+# Copyright 2025 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.

official/vision/serving/__init__.py

@@ -1,4 +1,4 @@
-# Copyright 2024 The TensorFlow Authors. All Rights Reserved.
+# Copyright 2025 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.

official/vision/serving/detection.py

@@ -1,4 +1,4 @@
-# Copyright 2024 The TensorFlow Authors. All Rights Reserved.
+# Copyright 2025 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.
@@ -136,6 +136,14 @@ class DetectionModule(export_base.ExportModule):
 
     return detections_dict
 
+  def _flatten_output(self, feature_map, feature_size=4):
+    flatten_outputs = []
+    for level_output in feature_map.values():
+      flatten_outputs.append(
+          tf.reshape(level_output, (self._batch_size, -1, feature_size))
+      )
+    return tf.concat(flatten_outputs, axis=1)
+
   def preprocess(
       self, images: tf.Tensor
   ) -> Tuple[tf.Tensor, Mapping[str, tf.Tensor], tf.Tensor]:
@@ -271,6 +279,18 @@ class DetectionModule(export_base.ExportModule):
         final_outputs['detection_outer_boxes'] = detections[
             'detection_outer_boxes'
         ]
+    elif (
+        isinstance(self.params.task.model, configs.retinanet.RetinaNet)
+        and not self.params.task.model.detection_generator.decode_boxes
+    ):
+      final_outputs = {
+          'raw_boxes': self._flatten_output(detections['box_outputs'], 4),
+          'raw_scores': tf.sigmoid(
+              self._flatten_output(
+                  detections['cls_outputs'], self.params.task.model.num_classes
+              )
+          ),
+      }
     else:
       # For RetinaNet model, apply export_config.
       if isinstance(self.params.task.model, configs.retinanet.RetinaNet):

official/vision/serving/detection_test.py

@@ -1,4 +1,4 @@
-# Copyright 2024 The TensorFlow Authors. All Rights Reserved.
+# Copyright 2025 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.
@@ -38,6 +38,7 @@ class DetectionExportTest(tf.test.TestCase, parameterized.TestCase):
       normalized_coordinates=False,
       nms_version='batched',
       output_intermediate_features=False,
+      decode_boxes=True,
   ):
     params = exp_factory.get_exp_config(experiment_name)
     params.task.model.outer_boxes_scale = outer_boxes_scale
@@ -48,6 +49,8 @@ class DetectionExportTest(tf.test.TestCase, parameterized.TestCase):
     params.task.model.detection_generator.nms_version = nms_version
     if output_intermediate_features:
       params.task.export_config.output_intermediate_features = True
+    if not decode_boxes:
+      params.task.model.detection_generator.decode_boxes = False
     detection_module = detection.DetectionModule(
         params,
         batch_size=1,
@@ -232,6 +235,41 @@ class DetectionExportTest(tf.test.TestCase, parameterized.TestCase):
         max_values.numpy(), tf.ones_like(max_values).numpy()
     )
 
+  @parameterized.parameters(
+      'retinanet_mobile_coco',
+      'retinanet_spinenet_coco',
+  )
+  def test_export_without_decoding_boxes(
+      self,
+      experiment_name,
+  ):
+    input_type = 'tflite'
+    tmp_dir = self.get_temp_dir()
+    module = self._get_detection_module(
+        experiment_name,
+        input_type=input_type,
+        apply_nms=False,
+        decode_boxes=False,
+    )
+
+    self._export_from_module(module, input_type, tmp_dir)
+
+    imported = tf.saved_model.load(tmp_dir)
+    detection_fn = imported.signatures['serving_default']
+
+    images = self._get_dummy_input(
+        input_type, batch_size=1, image_size=(640, 640)
+    )
+    outputs = detection_fn(tf.constant(images))
+
+    self.assertContainsSubset(
+        {
+            'raw_boxes',
+            'raw_scores',
+        },
+        outputs.keys(),
+    )
+
 
 if __name__ == '__main__':
   tf.test.main()

official/vision/serving/export_base.py

@@ -1,4 +1,4 @@
-# Copyright 2024 The TensorFlow Authors. All Rights Reserved.
+# Copyright 2025 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.

official/vision/serving/export_base_v2.py

@@ -1,4 +1,4 @@
-# Copyright 2024 The TensorFlow Authors. All Rights Reserved.
+# Copyright 2025 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.

official/vision/serving/export_base_v2_test.py

@@ -1,4 +1,4 @@
-# Copyright 2024 The TensorFlow Authors. All Rights Reserved.
+# Copyright 2025 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.

official/vision/serving/export_module_factory.py

@@ -1,4 +1,4 @@
-# Copyright 2024 The TensorFlow Authors. All Rights Reserved.
+# Copyright 2025 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.

official/vision/serving/export_module_factory_test.py

@@ -1,4 +1,4 @@
-# Copyright 2024 The TensorFlow Authors. All Rights Reserved.
+# Copyright 2025 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.

official/vision/serving/export_saved_model.py

@@ -1,4 +1,4 @@
-# Copyright 2024 The TensorFlow Authors. All Rights Reserved.
+# Copyright 2025 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.

official/vision/serving/export_saved_model_lib.py

@@ -1,4 +1,4 @@
-# Copyright 2024 The TensorFlow Authors. All Rights Reserved.
+# Copyright 2025 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.

official/vision/serving/export_saved_model_lib_test.py

@@ -1,4 +1,4 @@
-# Copyright 2024 The TensorFlow Authors. All Rights Reserved.
+# Copyright 2025 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.

official/vision/serving/export_saved_model_lib_v2.py

@@ -1,4 +1,4 @@
-# Copyright 2024 The TensorFlow Authors. All Rights Reserved.
+# Copyright 2025 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.

official/vision/serving/export_tfhub.py

@@ -1,4 +1,4 @@
-# Copyright 2024 The TensorFlow Authors. All Rights Reserved.
+# Copyright 2025 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.
@@ -14,7 +14,6 @@
 
 """A script to export the image classification as a TF-Hub SavedModel."""
 
-# Import libraries
 from absl import app
 from absl import flags
 

official/vision/serving/export_tfhub_lib.py

@@ -1,4 +1,4 @@
-# Copyright 2024 The TensorFlow Authors. All Rights Reserved.
+# Copyright 2025 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.
@@ -15,8 +15,6 @@
 """A script to export a TF-Hub SavedModel."""
 from typing import List, Optional
 
-# Import libraries
-
 import tensorflow as tf, tf_keras
 
 from official.core import config_definitions as cfg