tf-models-nightly 2.19.0.dev20241120__py2.py3-none-any.whl → 2.19.0.dev20241121__py2.py3-none-any.whl

official/vision/modeling/heads/instance_heads.py

@@ -173,7 +173,7 @@ class DetectionHead(tf_keras.layers.Layer):
 
     super(DetectionHead, self).build(input_shape)
 
-  def call(self, inputs: tf.Tensor, training: bool = None):
+  def call(self, inputs: tf.Tensor, training: bool = None):  # pytype: disable=annotation-type-mismatch
     """Forward pass of box and class branches for the Mask-RCNN model.
 
     Args:
@@ -379,7 +379,7 @@ class MaskHead(tf_keras.layers.Layer):
 
     super(MaskHead, self).build(input_shape)
 
-  def call(self, inputs: List[tf.Tensor], training: bool = None):
+  def call(self, inputs: List[tf.Tensor], training: bool = None):  # pytype: disable=annotation-type-mismatch
     """Forward pass of mask branch for the Mask-RCNN model.
 
     Args:

official/vision/modeling/heads/segmentation_heads.py

@@ -167,7 +167,7 @@ class MaskScoring(tf_keras.Model):
 
     super(MaskScoring, self).build(input_shape)
 
-  def call(self, inputs: tf.Tensor, training: bool = None):  # pytype: disable=signature-mismatch  # overriding-parameter-count-checks
+  def call(self, inputs: tf.Tensor, training: bool = None):  # pytype: disable=annotation-type-mismatch,signature-mismatch
     """Forward pass mask scoring head.
 
     Args:

official/vision/modeling/layers/roi_aligner.py

@@ -38,10 +38,12 @@ class MultilevelROIAligner(tf_keras.layers.Layer):
     }
     super(MultilevelROIAligner, self).__init__(**kwargs)
 
-  def call(self,
-           features: Mapping[str, tf.Tensor],
-           boxes: tf.Tensor,
-           training: bool = None):
+  def call(
+      self,  # pytype: disable=annotation-type-mismatch
+      features: Mapping[str, tf.Tensor],
+      boxes: tf.Tensor,
+      training: bool = None,
+  ):
     """Generates ROIs.
 
     Args:

official/vision/modeling/retinanet_model.py

@@ -81,12 +81,14 @@ class RetinaNetModel(tf_keras.Model):
     self._detection_generator = detection_generator
     self._anchor_boxes = anchor_boxes
 
-  def call(self,
-           images: Union[tf.Tensor, Sequence[tf.Tensor]],
-           image_shape: Optional[tf.Tensor] = None,
-           anchor_boxes: Mapping[str, tf.Tensor] | None = None,
-           output_intermediate_features: bool = False,
-           training: bool = None) -> Mapping[str, tf.Tensor]:
+  def call(
+      self,  # pytype: disable=annotation-type-mismatch
+      images: Union[tf.Tensor, Sequence[tf.Tensor]],
+      image_shape: Optional[tf.Tensor] = None,
+      anchor_boxes: Mapping[str, tf.Tensor] | None = None,
+      output_intermediate_features: bool = False,
+      training: bool = None,
+  ) -> Mapping[str, tf.Tensor]:
     """Forward pass of the RetinaNet model.
 
     Args:

official/vision/modeling/segmentation_model.py

@@ -58,8 +58,9 @@ class SegmentationModel(tf_keras.Model):
     self.head = head
     self.mask_scoring_head = mask_scoring_head
 
-  def call(self, inputs: tf.Tensor, training: bool = None  # pytype: disable=signature-mismatch  # overriding-parameter-count-checks
-           ) -> Dict[str, tf.Tensor]:
+  def call(
+      self, inputs: tf.Tensor, training: bool = None  # pytype: disable=annotation-type-mismatch,signature-mismatch
+  ) -> Dict[str, tf.Tensor]:
     backbone_features = self.backbone(inputs)
 
     if self.decoder:

{tf_models_nightly-2.19.0.dev20241120.dist-info → tf_models_nightly-2.19.0.dev20241121.dist-info}/METADATA

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

{tf_models_nightly-2.19.0.dev20241120.dist-info → tf_models_nightly-2.19.0.dev20241121.dist-info}/RECORD

@@ -1050,9 +1050,9 @@ official/vision/modeling/factory_3d.py,sha256=2pUdT2mC1tg20qqSHUtitGIvLKW866R2Ss
 official/vision/modeling/factory_test.py,sha256=ojUEYRMdtkKq75IcesdtOlrPwTxfJUgrS12BEsHps5w,6700
 official/vision/modeling/maskrcnn_model.py,sha256=03QTWfRGb5wacFVo1SHOSLJ4hLc9Eq8VfEXJntylAEc,20774
 official/vision/modeling/maskrcnn_model_test.py,sha256=UuNX3_BApjXXvuCNIMtbjkS-ZOtq8mORpOVNkG4glNU,14825
-official/vision/modeling/retinanet_model.py,sha256=SxgcAEqquogFBONsNPajiInj-SqezpqfIi2K0QBZCTk,10320
+official/vision/modeling/retinanet_model.py,sha256=d2rqliULA4fh_EO5c3YDWtZxEpadzcTpQg1M5YhJAU0,10350
 official/vision/modeling/retinanet_model_test.py,sha256=KOHqE1wRKD_488XBnPy1qTdCkkgCpo7BYAoA67iPwNQ,11145
-official/vision/modeling/segmentation_model.py,sha256=aZd6g4P_cHJDLWS7h7cNC3YsdAuyEpRGxZMZJzsqAx4,3419
+official/vision/modeling/segmentation_model.py,sha256=WgWI-KF4s655O7rITJzDsqQXrwhgaq6yUAxrIvD_C9E,3405
 official/vision/modeling/segmentation_model_test.py,sha256=fimmpt0pFMcX214Vuf0C4NcRdWLLB9F2IJeiuTFUO54,2817
 official/vision/modeling/video_classification_model.py,sha256=RKrT22D5yW435S9AyYlgOFCIi0knbmZFvgwh_5JubPI,4713
 official/vision/modeling/video_classification_model_test.py,sha256=V6dZUHSdV-2R5l0dAnAr7ArFdYIHFm6mbQgtt_7GbbY,3252
@@ -1094,9 +1094,9 @@ official/vision/modeling/decoders/nasfpn_test.py,sha256=vHyfoJ5XjIEsN66-Clvr6Vvc
 official/vision/modeling/heads/__init__.py,sha256=u3-HLjhhOzNx5Yhshw1j6-pgQAIHl4lAJWvoht8YDCk,1088
 official/vision/modeling/heads/dense_prediction_heads.py,sha256=t-vbxQBCJwTg41WQZZsBXVWUMffaiZCVJNTGq6Ey2mk,27822
 official/vision/modeling/heads/dense_prediction_heads_test.py,sha256=VKjvGzcWWGBmaeqlFEWWnAjQnwvxPBgcmP7N31OmRIU,9060
-official/vision/modeling/heads/instance_heads.py,sha256=6JBhSjg6fPhqZauIHGE9C3uEM7hc4FqAzdpb60sCW9U,17810
+official/vision/modeling/heads/instance_heads.py,sha256=AOXouP8w8f3Jv_2S8fOgv0MrsuUJ2Lob1_SJOkGWNUs,17898
 official/vision/modeling/heads/instance_heads_test.py,sha256=k6lqCAFOEnW-rcSHwlqUE6-I72tpKg2ldFJn5nE8ai4,4188
-official/vision/modeling/heads/segmentation_heads.py,sha256=Rt0mvv1sPNPWHyncVs2qJ1hE9Wwuvw3Fu6uWjOUKroI,20184
+official/vision/modeling/heads/segmentation_heads.py,sha256=E3wWfpazWC6VZ1u0iHyCByXaJM8IhJs9AbDT5mswo-w,20172
 official/vision/modeling/heads/segmentation_heads_test.py,sha256=XgpA_AewJ8ivS8m1q0LNk0XtVKG3nqy7lRqgPW3ZNFA,3793
 official/vision/modeling/layers/__init__.py,sha256=_RjeNeStjst4yPYwYUBZ1G9dnmx4Lz5X110N5H1f6uU,2597
 official/vision/modeling/layers/box_sampler.py,sha256=UakVZOA9EVY00RzgTFbuTsUvuzFZ2feJWvn5f6U0m6c,3411
@@ -1113,7 +1113,7 @@ official/vision/modeling/layers/nn_blocks_3d_test.py,sha256=6JC8SgZW1EIgll2MqiRF
 official/vision/modeling/layers/nn_blocks_test.py,sha256=4cQhJvIq7jxp5ldfjPCf7CDcU3_7Syk2-kzBg2u7KPs,40401
 official/vision/modeling/layers/nn_layers.py,sha256=_Ftdcbl04zbEXxkaB_61Lq8jXvNmRkS3XHhjkot-mlw,52982
 official/vision/modeling/layers/nn_layers_test.py,sha256=EnQrMEAuPFe0P15KbiSqwnqEfM-UxcVOnPM-U6v0mF0,13319
-official/vision/modeling/layers/roi_aligner.py,sha256=UnCqzMHrGGNSIw1YQPljzCipQYTJtFi4h9-3Wl9oQrk,2543
+official/vision/modeling/layers/roi_aligner.py,sha256=ZEtAJnqxSla2xVQS0JfOze51-dOxFUgyQDFiYXWhUpc,2583
 official/vision/modeling/layers/roi_aligner_test.py,sha256=zOxhUJqbbtheKMFrBy0RfUCUDzTDYe61klvW6K24QjQ,1285
 official/vision/modeling/layers/roi_generator.py,sha256=ZL-m5IObg8JsXJ_DZ0QQusv2UBzHNiGXlTZZU1rJorU,14418
 official/vision/modeling/layers/roi_sampler.py,sha256=uajmniFsZrmAR_-ojva_dVm9xZU3wze5EqCryP9gUoQ,10006
@@ -1222,9 +1222,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.19.0.dev20241120.dist-info/AUTHORS,sha256=1dG3fXVu9jlo7bul8xuix5F5vOnczMk7_yWn4y70uw0,337
-tf_models_nightly-2.19.0.dev20241120.dist-info/LICENSE,sha256=WxeBS_DejPZQabxtfMOM_xn8qoZNJDQjrT7z2wG1I4U,11512
-tf_models_nightly-2.19.0.dev20241120.dist-info/METADATA,sha256=LTsB52E_KTsU0by7YUbRPJnnOxR1qvtRxZdlQAGYd0M,1432
-tf_models_nightly-2.19.0.dev20241120.dist-info/WHEEL,sha256=kGT74LWyRUZrL4VgLh6_g12IeVl_9u9ZVhadrgXZUEY,110
-tf_models_nightly-2.19.0.dev20241120.dist-info/top_level.txt,sha256=gum2FfO5R4cvjl2-QtP-S1aNmsvIZaFFT6VFzU0f4-g,33
-tf_models_nightly-2.19.0.dev20241120.dist-info/RECORD,,
+tf_models_nightly-2.19.0.dev20241121.dist-info/AUTHORS,sha256=1dG3fXVu9jlo7bul8xuix5F5vOnczMk7_yWn4y70uw0,337
+tf_models_nightly-2.19.0.dev20241121.dist-info/LICENSE,sha256=WxeBS_DejPZQabxtfMOM_xn8qoZNJDQjrT7z2wG1I4U,11512
+tf_models_nightly-2.19.0.dev20241121.dist-info/METADATA,sha256=kkf1-eZIurwCImjbs_KaAfa0TbLAcn7CAVXxtS__NT0,1432
+tf_models_nightly-2.19.0.dev20241121.dist-info/WHEEL,sha256=kGT74LWyRUZrL4VgLh6_g12IeVl_9u9ZVhadrgXZUEY,110
+tf_models_nightly-2.19.0.dev20241121.dist-info/top_level.txt,sha256=gum2FfO5R4cvjl2-QtP-S1aNmsvIZaFFT6VFzU0f4-g,33
+tf_models_nightly-2.19.0.dev20241121.dist-info/RECORD,,