tf-models-nightly 2.14.0.dev20231016__py2.py3-none-any.whl → 2.14.0.dev20231018__py2.py3-none-any.whl

official/projects/yolo/serving/export_module_factory.py

@@ -20,13 +20,13 @@ import tensorflow as tf
 
 from official.core import config_definitions as cfg
 from official.core import export_base
-from official.projects.yolo.configs.yolo import YoloTask
-from official.projects.yolo.configs.yolov7 import YoloV7Task
+from official.projects.yolo.configs import darknet_classification
+from official.projects.yolo.configs import yolo
+from official.projects.yolo.configs import yolov7
 from official.projects.yolo.modeling import factory as yolo_factory
 from official.projects.yolo.modeling.backbones import darknet  # pylint: disable=unused-import
 from official.projects.yolo.modeling.decoders import yolo_decoder  # pylint: disable=unused-import
 from official.projects.yolo.serving import model_fn as yolo_model_fn
-from official.vision import configs
 from official.vision.dataloaders import classification_input
 from official.vision.modeling import factory
 from official.vision.serving import export_utils
@@ -164,12 +164,12 @@ def create_yolo_export_module(
       input_type, batch_size, input_image_size, num_channels, input_name)
   input_specs = tf.keras.layers.InputSpec(shape=[batch_size] +
                                           input_image_size + [num_channels])
-  if isinstance(params.task, YoloTask):
+  if isinstance(params.task, yolo.YoloTask):
     model, _ = yolo_factory.build_yolo(
         input_specs=input_specs,
         model_config=params.task.model,
         l2_regularization=None)
-  elif isinstance(params.task, YoloV7Task):
+  elif isinstance(params.task, yolov7.YoloV7Task):
     model = yolo_factory.build_yolov7(
         input_specs=input_specs,
         model_config=params.task.model,
@@ -248,13 +248,13 @@ def get_export_module(params: cfg.ExperimentConfig,
                       input_name: Optional[str] = None) -> ExportModule:
   """Factory for export modules."""
   if isinstance(params.task,
-                configs.image_classification.ImageClassificationTask):
+                darknet_classification.ImageClassificationTask):
     export_module = create_classification_export_module(params, input_type,
                                                         batch_size,
                                                         input_image_size,
                                                         num_channels,
                                                         input_name)
-  elif isinstance(params.task, (YoloTask, YoloV7Task)):
+  elif isinstance(params.task, (yolo.YoloTask, yolov7.YoloV7Task)):
     export_module = create_yolo_export_module(params, input_type, batch_size,
                                               input_image_size, num_channels,
                                               input_name)

official/vision/modeling/layers/detection_generator.py

@@ -1128,7 +1128,7 @@ class MultilevelDetectionGenerator(tf.keras.layers.Layer):
       nms_v3_refinements: Optional[int] = None,
       return_decoded: Optional[bool] = None,
       use_class_agnostic_nms: Optional[bool] = None,
-      box_coder_weights: list[float] | None = None,
+      box_coder_weights: Optional[List[float]] = None,
       **kwargs,
   ):
     """Initializes a multi-level detection generator.

official/vision/tasks/maskrcnn.py

@@ -575,6 +575,10 @@ class MaskRCNNTask(base_task.Task):
     else:
       instance_metrics = self.instance_box_perclass_metrics
       prefix = ''
+    if instance_metrics is None:
+      raise ValueError(
+          'No instance metrics defined when use_masks is %s' % use_masks
+      )
     result = instance_metrics.result()
     iou_thresholds = instance_metrics.get_config()['iou_thresholds']
 

{tf_models_nightly-2.14.0.dev20231016.dist-info → tf_models_nightly-2.14.0.dev20231018.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: tf-models-nightly
-Version: 2.14.0.dev20231016
+Version: 2.14.0.dev20231018
 Summary: TensorFlow Official Models
 Home-page: https://github.com/tensorflow/models
 Author: Google Inc.

{tf_models_nightly-2.14.0.dev20231016.dist-info → tf_models_nightly-2.14.0.dev20231018.dist-info}/RECORD

@@ -781,7 +781,7 @@ official/projects/yolo/optimization/configs/__init__.py,sha256=1ToRMjre4mErL4Ek4
 official/projects/yolo/optimization/configs/optimization_config.py,sha256=kl0F8qjqtgNbeuDH4mFxwna0Unm6bl49a6dfE4Z_s8A,2084
 official/projects/yolo/optimization/configs/optimizer_config.py,sha256=0YZgwyEQKfEwP44x9wzZ7N01I0bdawqEOgWmezora0Y,2283
 official/projects/yolo/serving/__init__.py,sha256=1ToRMjre4mErL4Ek4_dMVxMjXNPossNXggV8fqbISao,609
-official/projects/yolo/serving/export_module_factory.py,sha256=tRtJbbaLxV1BLm3OTTkyzkkyRrGxOTJdcg4gCuInW6k,9910
+official/projects/yolo/serving/export_module_factory.py,sha256=71jA0N-vox78I9585MSQsQjBGPMi8i9RnFnILZVP-7E,9938
 official/projects/yolo/serving/export_saved_model.py,sha256=wFjN6wAJ-ckNEtSTwMY6Kj4jDz7CSAytOqFy1AVCt7w,4584
 official/projects/yolo/serving/export_tflite.py,sha256=Pt1rkKu9TxVYB8HtLadJQ_buWzNXqWL8Zp0TFDPUXzw,902
 official/projects/yolo/serving/model_fn.py,sha256=J1i-UPX1TuPqPaBWu2HyhtVHthq6VRbsnR98Sjrs0iY,3008
@@ -993,7 +993,7 @@ official/vision/modeling/layers/__init__.py,sha256=Yk0zkJVVgwi--tv5-NKIAKHCzpFw9
 official/vision/modeling/layers/box_sampler.py,sha256=hsFoo9lsD57LDUnj0SlIoM9QNg26yqQxvADPtQRVhV8,3401
 official/vision/modeling/layers/deeplab.py,sha256=fe35K3dhYXJ1Ep1CF3B3a1x5Au9X6AfXUZ2TngWZMSI,8561
 official/vision/modeling/layers/deeplab_test.py,sha256=5auLEAy5Ou-CiCZCUcXKchcJ_csAdIThNIJhrHjbKzw,1894
-official/vision/modeling/layers/detection_generator.py,sha256=lUVRgfJilT7CsjK4ktKeInSvYmHmEElOpocct88Gvss,65440
+official/vision/modeling/layers/detection_generator.py,sha256=Zm3snhOKvNZOMAD-0mZqK-XU7oy1tOXGhAxaIcv9Xr0,65443
 official/vision/modeling/layers/detection_generator_test.py,sha256=IXFyozqe8dF9qr26kZioPMiD1WBL96ksYskz0--dgIQ,13567
 official/vision/modeling/layers/edgetpu.py,sha256=AmfMIxT1XCSZo0zXoZO0h-WOdxNYYYZo4lGf_d7DL6g,16727
 official/vision/modeling/layers/edgetpu_test.py,sha256=V3FauCD5xeIjkq2HbvAPQag4FwaekX1nDY5vTHrO3K0,10110
@@ -1057,7 +1057,7 @@ official/vision/serving/video_classification.py,sha256=5SpQapPt4zkKTmBtUheuvKfj6
 official/vision/serving/video_classification_test.py,sha256=bUhsW11ti-yS4rRRflavWUaoFO5nyuSnewTN_OeHz_c,4222
 official/vision/tasks/__init__.py,sha256=pwY_FnD2dwCOuYUHfZoAvnONwDWalX5E1J27tSJgnzA,995
 official/vision/tasks/image_classification.py,sha256=Q7TgcKEhHLP_64jlLbpu42Y8Hzdf3VhOm4Xb5dlEx1o,16689
-official/vision/tasks/maskrcnn.py,sha256=Gc5cu_W48bMz_5d8HL8UUR3uJTL0PMvKm3ZnyO2ALk4,25421
+official/vision/tasks/maskrcnn.py,sha256=_gB6Pc9bgajD_5eXkZNITjM1DFvg2Bwtb050k_Fu1X0,25559
 official/vision/tasks/retinanet.py,sha256=UKiwdO7Id53nqpigW9Hmd-7xGtdEb0il_y5K3Zz3Fkw,18104
 official/vision/tasks/semantic_segmentation.py,sha256=QA6qyxjdxptBRneGhECOtubvGo4ZyG17fFBV6Yz0hlY,14233
 official/vision/tasks/video_classification.py,sha256=E_AQwPvSzMx2MhnQctHKGKKshot-R9lwYLE6HFEux5M,14308
@@ -1112,9 +1112,9 @@ tensorflow_models/__init__.py,sha256=Ciz_YBke6teb6y42QyQTUBDdXJAiV7Qdu1zOoZvYiKw
 tensorflow_models/tensorflow_models_test.py,sha256=Kz2y4V-rtBhZFFfKD2soCq52hviSfJVV1L2ztqS-9oM,1385
 tensorflow_models/nlp/__init__.py,sha256=3dULDpUBpDi9vljpXadq6oJrWH4y6z42Bz2d3hopYZw,807
 tensorflow_models/vision/__init__.py,sha256=4y77XkHaH8qLls3-6ta4tMp3Xj8CLbB0ihH91HsQ9z4,833
-tf_models_nightly-2.14.0.dev20231016.dist-info/AUTHORS,sha256=1dG3fXVu9jlo7bul8xuix5F5vOnczMk7_yWn4y70uw0,337
-tf_models_nightly-2.14.0.dev20231016.dist-info/LICENSE,sha256=WxeBS_DejPZQabxtfMOM_xn8qoZNJDQjrT7z2wG1I4U,11512
-tf_models_nightly-2.14.0.dev20231016.dist-info/METADATA,sha256=1CfYO-hJQHTxqf6823ZzIVbqqaHovnZBnxOGx3mPYVA,1390
-tf_models_nightly-2.14.0.dev20231016.dist-info/WHEEL,sha256=kGT74LWyRUZrL4VgLh6_g12IeVl_9u9ZVhadrgXZUEY,110
-tf_models_nightly-2.14.0.dev20231016.dist-info/top_level.txt,sha256=gum2FfO5R4cvjl2-QtP-S1aNmsvIZaFFT6VFzU0f4-g,33
-tf_models_nightly-2.14.0.dev20231016.dist-info/RECORD,,
+tf_models_nightly-2.14.0.dev20231018.dist-info/AUTHORS,sha256=1dG3fXVu9jlo7bul8xuix5F5vOnczMk7_yWn4y70uw0,337
+tf_models_nightly-2.14.0.dev20231018.dist-info/LICENSE,sha256=WxeBS_DejPZQabxtfMOM_xn8qoZNJDQjrT7z2wG1I4U,11512
+tf_models_nightly-2.14.0.dev20231018.dist-info/METADATA,sha256=2HvdZtJ09pOSJaBsNR08te5g6sQF5QE8zIMkCsjxTP8,1390
+tf_models_nightly-2.14.0.dev20231018.dist-info/WHEEL,sha256=kGT74LWyRUZrL4VgLh6_g12IeVl_9u9ZVhadrgXZUEY,110
+tf_models_nightly-2.14.0.dev20231018.dist-info/top_level.txt,sha256=gum2FfO5R4cvjl2-QtP-S1aNmsvIZaFFT6VFzU0f4-g,33
+tf_models_nightly-2.14.0.dev20231018.dist-info/RECORD,,