tf-models-nightly 2.17.0.dev20240430__py2.py3-none-any.whl → 2.17.0.dev20240502__py2.py3-none-any.whl

official/projects/qat/vision/modeling/factory.py

@@ -190,10 +190,11 @@ def build_qat_retinanet(
               head.get_config()))
 
   optimized_model = retinanet_model.RetinaNetModel(
-      optimized_backbone,
-      decoder,
-      head,
-      model.detection_generator,
+      backbone=optimized_backbone,
+      decoder=decoder,
+      head=head,
+      detection_generator=model.detection_generator,
+      anchor_boxes=model.anchor_boxes,
       min_level=model_config.min_level,
       max_level=model_config.max_level,
       num_scales=model_config.anchor.num_scales,

official/vision/modeling/retinanet_model.py

@@ -239,6 +239,10 @@ class RetinaNetModel(tf_keras.Model):
   def detection_generator(self) -> tf_keras.layers.Layer:
     return self._detection_generator
 
+  @property
+  def anchor_boxes(self) -> Mapping[str, tf.Tensor] | None:
+    return self._anchor_boxes
+
   def get_config(self) -> Mapping[str, Any]:
     return self._config_dict
 

{tf_models_nightly-2.17.0.dev20240430.dist-info → tf_models_nightly-2.17.0.dev20240502.dist-info}/METADATA

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

{tf_models_nightly-2.17.0.dev20240430.dist-info → tf_models_nightly-2.17.0.dev20240502.dist-info}/RECORD

@@ -634,7 +634,7 @@ official/projects/qat/vision/configs/retinanet_test.py,sha256=D8zoO3pEs38bsmzWBw
 official/projects/qat/vision/configs/semantic_segmentation.py,sha256=YbbhULSzKaV-d9Nf-ASQbq0hztcT4cddQpAjT9PN4aE,2280
 official/projects/qat/vision/configs/semantic_segmentation_test.py,sha256=FeitPnHGENtfX_6WfO5Ak7C6VT1bPqtodtMaNuOqKVA,2031
 official/projects/qat/vision/modeling/__init__.py,sha256=TP0l_DQigubnbzCmmy09vW0-YdJs_bUrx1kmsVaVidg,757
-official/projects/qat/vision/modeling/factory.py,sha256=IdKFni8nln55CTWSaSYH0ori2i68yt39wfDAauTw89M,11054
+official/projects/qat/vision/modeling/factory.py,sha256=M5tsN9aKL_mM4F6pRRDT3xAb9o9gu6NGbIcepOwCniw,11135
 official/projects/qat/vision/modeling/factory_test.py,sha256=ysBk5RDu-ZBUOnBJ12r5K5ze-V17ZVrNN6h-_uscCio,10155
 official/projects/qat/vision/modeling/segmentation_model.py,sha256=SW80u-HarvnG-I18lfSPFtldtCgeiQH4thSOLUDAA98,3087
 official/projects/qat/vision/modeling/heads/__init__.py,sha256=7JQmDV0CEiffKPyDQpgE1nzrliNjyPKYQt8FvZkrIV4,743
@@ -1036,7 +1036,7 @@ official/vision/modeling/factory_3d.py,sha256=P9V783V0JQZLesvSdHAGQ6fLrXHq6XnJdv
 official/vision/modeling/factory_test.py,sha256=64QsxFiXaruNHczE_85cxFmxFTrHFjG3Ol5HfI9OeX8,6719
 official/vision/modeling/maskrcnn_model.py,sha256=03QTWfRGb5wacFVo1SHOSLJ4hLc9Eq8VfEXJntylAEc,20774
 official/vision/modeling/maskrcnn_model_test.py,sha256=_KqwN8VcwM8oUBSGwJjVaPC0JeuN0H_qf5GPDYNrvJQ,14844
-official/vision/modeling/retinanet_model.py,sha256=vF2IUvHCpZ9cEQdO4MsWRbh3bt1sY1TZk2UryfV4h30,10236
+official/vision/modeling/retinanet_model.py,sha256=binKQmYkVqTtSXdeyObSWiwIPTEutfMUuF05MOgG8xQ,10339
 official/vision/modeling/retinanet_model_test.py,sha256=m9cBm8arAb0sNx5QsWRu_ildzdUDXVQHDlvjAb9x0Fk,11164
 official/vision/modeling/segmentation_model.py,sha256=BrH3w0yZ60uwyEGP08iYt0rAWVlJjyUkRr0eZN3yl4s,3438
 official/vision/modeling/segmentation_model_test.py,sha256=fimmpt0pFMcX214Vuf0C4NcRdWLLB9F2IJeiuTFUO54,2817
@@ -1206,9 +1206,9 @@ tensorflow_models/tensorflow_models_test.py,sha256=nc6A9K53OGqF25xN5St8EiWvdVbda
 tensorflow_models/nlp/__init__.py,sha256=4tA5Pf4qaFwT-fIFOpX7x7FHJpnyJT-5UgOeFYTyMlc,807
 tensorflow_models/uplift/__init__.py,sha256=mqfa55gweOdpKoaQyid4A_4u7xw__FcQeSIF0k_pYmI,999
 tensorflow_models/vision/__init__.py,sha256=zBorY_v5xva1uI-qxhZO3Qh-Dii-Suq6wEYh6hKHDfc,833
-tf_models_nightly-2.17.0.dev20240430.dist-info/AUTHORS,sha256=1dG3fXVu9jlo7bul8xuix5F5vOnczMk7_yWn4y70uw0,337
-tf_models_nightly-2.17.0.dev20240430.dist-info/LICENSE,sha256=WxeBS_DejPZQabxtfMOM_xn8qoZNJDQjrT7z2wG1I4U,11512
-tf_models_nightly-2.17.0.dev20240430.dist-info/METADATA,sha256=oI--15-Ni4wbhuXXzJa3wLz17ffmPLNKaB0gq9G42XU,1432
-tf_models_nightly-2.17.0.dev20240430.dist-info/WHEEL,sha256=kGT74LWyRUZrL4VgLh6_g12IeVl_9u9ZVhadrgXZUEY,110
-tf_models_nightly-2.17.0.dev20240430.dist-info/top_level.txt,sha256=gum2FfO5R4cvjl2-QtP-S1aNmsvIZaFFT6VFzU0f4-g,33
-tf_models_nightly-2.17.0.dev20240430.dist-info/RECORD,,
+tf_models_nightly-2.17.0.dev20240502.dist-info/AUTHORS,sha256=1dG3fXVu9jlo7bul8xuix5F5vOnczMk7_yWn4y70uw0,337
+tf_models_nightly-2.17.0.dev20240502.dist-info/LICENSE,sha256=WxeBS_DejPZQabxtfMOM_xn8qoZNJDQjrT7z2wG1I4U,11512
+tf_models_nightly-2.17.0.dev20240502.dist-info/METADATA,sha256=ahKFJH3rVJXm81Pxd1fNMR40ttHgEil0IJNEufn4ipQ,1432
+tf_models_nightly-2.17.0.dev20240502.dist-info/WHEEL,sha256=kGT74LWyRUZrL4VgLh6_g12IeVl_9u9ZVhadrgXZUEY,110
+tf_models_nightly-2.17.0.dev20240502.dist-info/top_level.txt,sha256=gum2FfO5R4cvjl2-QtP-S1aNmsvIZaFFT6VFzU0f4-g,33
+tf_models_nightly-2.17.0.dev20240502.dist-info/RECORD,,