tf-models-nightly 2.18.0.dev20240801__py2.py3-none-any.whl → 2.18.0.dev20240803__py2.py3-none-any.whl

official/projects/panoptic/configs/panoptic_deeplab.py

@@ -14,11 +14,10 @@
 
 """Panoptic Deeplab configuration definition."""
 import dataclasses
+import math
 import os
 from typing import List, Optional, Union
 
-import numpy as np
-
 from official.core import config_definitions as cfg
 from official.core import exp_factory
 from official.modeling import hyperparams
@@ -220,7 +219,7 @@ def panoptic_deeplab_resnet_coco() -> cfg.ExperimentConfig:
   aspp_dilation_rates = [6, 12, 18]
   multigrid = [1, 2, 4]
   stem_type = 'v1'
-  level = int(np.math.log2(output_stride))
+  level = int(math.log2(output_stride))
 
   config = cfg.ExperimentConfig(
       runtime=cfg.RuntimeConfig(
@@ -385,7 +384,7 @@ def panoptic_deeplab_mobilenetv3_large_coco() -> cfg.ExperimentConfig:
   input_size = [640, 640, 3]
   output_stride = 16
   aspp_dilation_rates = [6, 12, 18]
-  level = int(np.math.log2(output_stride))
+  level = int(math.log2(output_stride))
 
   config = cfg.ExperimentConfig(
       runtime=cfg.RuntimeConfig(
@@ -547,7 +546,7 @@ def panoptic_deeplab_mobilenetv3_small_coco() -> cfg.ExperimentConfig:
   input_size = [640, 640, 3]
   output_stride = 16
   aspp_dilation_rates = [6, 12, 18]
-  level = int(np.math.log2(output_stride))
+  level = int(math.log2(output_stride))
 
   config = cfg.ExperimentConfig(
       runtime=cfg.RuntimeConfig(

official/vision/configs/semantic_segmentation.py

@@ -14,11 +14,10 @@
 
 """Semantic segmentation configuration definition."""
 import dataclasses
+import math
 import os
 from typing import List, Optional, Sequence, Union
 
-import numpy as np
-
 from official.core import config_definitions as cfg
 from official.core import exp_factory
 from official.modeling import hyperparams
@@ -233,7 +232,7 @@ def seg_deeplabv3_pascal() -> cfg.ExperimentConfig:
   aspp_dilation_rates = [12, 24, 36]  # [6, 12, 18] if output_stride = 16
   multigrid = [1, 2, 4]
   stem_type = 'v1'
-  level = int(np.math.log2(output_stride))
+  level = int(math.log2(output_stride))
   config = cfg.ExperimentConfig(
       task=SemanticSegmentationTask(
           model=SemanticSegmentationModel(
@@ -325,7 +324,7 @@ def seg_deeplabv3plus_pascal() -> cfg.ExperimentConfig:
   aspp_dilation_rates = [6, 12, 18]
   multigrid = [1, 2, 4]
   stem_type = 'v1'
-  level = int(np.math.log2(output_stride))
+  level = int(math.log2(output_stride))
   config = cfg.ExperimentConfig(
       task=SemanticSegmentationTask(
           model=SemanticSegmentationModel(
@@ -492,7 +491,7 @@ def mnv2_deeplabv3_pascal() -> cfg.ExperimentConfig:
   steps_per_epoch = PASCAL_TRAIN_EXAMPLES // train_batch_size
   output_stride = 16
   aspp_dilation_rates = []
-  level = int(np.math.log2(output_stride))
+  level = int(math.log2(output_stride))
   pool_kernel_size = []
 
   config = cfg.ExperimentConfig(
@@ -593,7 +592,7 @@ def seg_deeplabv3plus_cityscapes() -> cfg.ExperimentConfig:
   aspp_dilation_rates = [6, 12, 18]
   multigrid = [1, 2, 4]
   stem_type = 'v1'
-  level = int(np.math.log2(output_stride))
+  level = int(math.log2(output_stride))
   config = cfg.ExperimentConfig(
       task=SemanticSegmentationTask(
           model=SemanticSegmentationModel(
@@ -694,7 +693,7 @@ def mnv2_deeplabv3_cityscapes() -> cfg.ExperimentConfig:
   aspp_dilation_rates = []
   pool_kernel_size = [512, 1024]
 
-  level = int(np.math.log2(output_stride))
+  level = int(math.log2(output_stride))
   config = cfg.ExperimentConfig(
       task=SemanticSegmentationTask(
           model=SemanticSegmentationModel(

official/vision/modeling/backbones/resnet_deeplab.py

@@ -14,10 +14,11 @@
 
 """Contains definitions of Residual Networks with Deeplab modifications."""
 
-from typing import Callable, Optional, Tuple, List
+import math
+from typing import Callable, List, Optional, Tuple
 
-import numpy as np
 import tensorflow as tf, tf_keras
+
 from official.modeling import hyperparams
 from official.modeling import tf_utils
 from official.vision.modeling.backbones import factory
@@ -238,7 +239,7 @@ class DilatedResNet(tf_keras.Model):
     else:
       x = layers.MaxPool2D(pool_size=3, strides=2, padding='same')(x)
 
-    normal_resnet_stage = int(np.math.log2(self._output_stride)) - 2
+    normal_resnet_stage = int(math.log2(self._output_stride)) - 2
 
     endpoints = {}
     for i in range(normal_resnet_stage + 1):

official/vision/modeling/backbones/resnet_deeplab_test.py

@@ -14,7 +14,10 @@
 
 """Tests for resnet_deeplab models."""
 
+import math
+
 # Import libraries
+
 from absl.testing import parameterized
 import numpy as np
 import tensorflow as tf, tf_keras
@@ -46,10 +49,15 @@ class ResNetTest(parameterized.TestCase, tf.test.TestCase):
     inputs = tf_keras.Input(shape=(input_size, input_size, 3), batch_size=1)
     endpoints = network(inputs)
     print(endpoints)
-    self.assertAllEqual([
-        1, input_size / output_stride, input_size / output_stride,
-        512 * endpoint_filter_scale
-    ], endpoints[str(int(np.math.log2(output_stride)))].shape.as_list())
+    self.assertAllEqual(
+        [
+            1,
+            input_size / output_stride,
+            input_size / output_stride,
+            512 * endpoint_filter_scale,
+        ],
+        endpoints[str(int(math.log2(output_stride)))].shape.as_list(),
+    )
 
   @parameterized.parameters(
       ('v0', None, 0.0, False, False),
@@ -82,10 +90,15 @@ class ResNetTest(parameterized.TestCase, tf.test.TestCase):
     inputs = tf_keras.Input(shape=(input_size, input_size, 3), batch_size=1)
     endpoints = network(inputs)
     print(endpoints)
-    self.assertAllEqual([
-        1, input_size / output_stride, input_size / output_stride,
-        512 * endpoint_filter_scale
-    ], endpoints[str(int(np.math.log2(output_stride)))].shape.as_list())
+    self.assertAllEqual(
+        [
+            1,
+            input_size / output_stride,
+            input_size / output_stride,
+            512 * endpoint_filter_scale,
+        ],
+        endpoints[str(int(math.log2(output_stride)))].shape.as_list(),
+    )
 
   @combinations.generate(
       combinations.combine(

{tf_models_nightly-2.18.0.dev20240801.dist-info → tf_models_nightly-2.18.0.dev20240803.dist-info}/METADATA

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

{tf_models_nightly-2.18.0.dev20240801.dist-info → tf_models_nightly-2.18.0.dev20240803.dist-info}/RECORD

@@ -595,7 +595,7 @@ official/projects/nhnet/utils.py,sha256=9pULza5Cjh31lAtZiRmKlwvzsUkkYWUL4OecYygX
 official/projects/panoptic/__init__.py,sha256=7oiypy0N82PDw9aSdcJBLVoGTd_oRSUOdvuJhMv4leQ,609
 official/projects/panoptic/train.py,sha256=rlBGgqJohRQaBUFC_iR-5g2gFNVVBOJMeBcTtbPX-8E,1275
 official/projects/panoptic/configs/__init__.py,sha256=7oiypy0N82PDw9aSdcJBLVoGTd_oRSUOdvuJhMv4leQ,609
-official/projects/panoptic/configs/panoptic_deeplab.py,sha256=afujxe-majo7Wy2nRV8uHKnhF_SmvJwAlBRlgapilkM,26059
+official/projects/panoptic/configs/panoptic_deeplab.py,sha256=iXTd3Fp_xaMkX4z747CNq28b83iEPNr4j_-JM9f6Xhg,26042
 official/projects/panoptic/configs/panoptic_maskrcnn.py,sha256=cyWDAqVfLTHa_qPwKybehx_ZkHRW9pC_2e5VCgPIjKQ,11613
 official/projects/panoptic/tasks/__init__.py,sha256=7oiypy0N82PDw9aSdcJBLVoGTd_oRSUOdvuJhMv4leQ,609
 official/projects/panoptic/tasks/panoptic_deeplab.py,sha256=nMrW7fvpNrNb-5O9jrtps4vL_gaa3wWg2c77LQRo9KU,14723
@@ -971,7 +971,7 @@ official/vision/configs/maskrcnn.py,sha256=yL8kggxXaCTIpSkcozAV2UudO7UqVcEh1_-rM
 official/vision/configs/maskrcnn_test.py,sha256=Wfkbz30h2qxPcpuu6CEpQsf8I_2df6y10-4bRLsWlj8,1733
 official/vision/configs/retinanet.py,sha256=oCKinkh4IyPslmI1pakwi6dVziwjkZ2cIcpSoGRjqnM,17806
 official/vision/configs/retinanet_test.py,sha256=ffS3QufQMLF8FZhKNmi7Yr1RDTnIyZ1XKQ9agr2EyW8,1699
-official/vision/configs/semantic_segmentation.py,sha256=4ZAyLWKcFYReyrEWBc5b7wld3mMcuH0RcaRe_4J2RrA,30831
+official/vision/configs/semantic_segmentation.py,sha256=xJbCWl46hTlUkw_S5UeKctdjuAZPTqE6UtSqQaLezYc,30808
 official/vision/configs/semantic_segmentation_test.py,sha256=va-ZG6CtBKcs0NicZe6WmJvHxPxxih7nB0orNtrRiEA,1867
 official/vision/configs/video_classification.py,sha256=DDiBq9tCfk3x-FNprWvWc5hLaVAjGLTWgGvXCpElbw4,14549
 official/vision/configs/video_classification_test.py,sha256=I1HSamxRQ3-f4-YHIeUChnT5CtHCxFQdiL0zy6RRUXU,1879
@@ -1058,8 +1058,8 @@ official/vision/modeling/backbones/mobilenet_test.py,sha256=7cl5eerD5j5UqHL8SLmp
 official/vision/modeling/backbones/resnet.py,sha256=dnYkdlYUzChGLOrQnUbwb9YJ7BDiFwgnLptks7kFb7k,16384
 official/vision/modeling/backbones/resnet_3d.py,sha256=Cq1lrlRqIg9ss_ud1iM_axW9lsTVtGYe3iA4DL9Orzk,18657
 official/vision/modeling/backbones/resnet_3d_test.py,sha256=hhCkW28UXc2peKHGgFl0MnYexFV8qTwEUkMPZ26a_MA,3799
-official/vision/modeling/backbones/resnet_deeplab.py,sha256=RCwLTEGwe4XGPGM5-ELfqdHVJRW35-whoFk6U_PK8Sc,15840
-official/vision/modeling/backbones/resnet_deeplab_test.py,sha256=JXklR7mTi7wWVAPsu48wW5IXzJBLw862uzlKPGdMdps,5520
+official/vision/modeling/backbones/resnet_deeplab.py,sha256=eHl44lBbvEi231to2-NiLkMiQweax8dGbkELQdC8FQs,15831
+official/vision/modeling/backbones/resnet_deeplab_test.py,sha256=7bpxr1j0e8mQ5VTMAAQ0d76OVUX2NYKfDL1PlGinhtg,5648
 official/vision/modeling/backbones/resnet_test.py,sha256=rjIFkLsbsUqobectT96jqMygOyCWWih0l9sZbR-Wi9I,5555
 official/vision/modeling/backbones/resnet_unet.py,sha256=2B4VIX2jPqT3aYpH69VSip6zqSeq4WcMFp0_cvEDFxM,21303
 official/vision/modeling/backbones/resnet_unet_test.py,sha256=SPwHtodwfnLnGeEcgK-PZYt_xHDoVe98ULfiKeuI13Q,2615
@@ -1212,9 +1212,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.18.0.dev20240801.dist-info/AUTHORS,sha256=1dG3fXVu9jlo7bul8xuix5F5vOnczMk7_yWn4y70uw0,337
-tf_models_nightly-2.18.0.dev20240801.dist-info/LICENSE,sha256=WxeBS_DejPZQabxtfMOM_xn8qoZNJDQjrT7z2wG1I4U,11512
-tf_models_nightly-2.18.0.dev20240801.dist-info/METADATA,sha256=W-AHm2glrBX8XvFSByp2dUnwnJtDp1zyOc2sBYDUWvk,1432
-tf_models_nightly-2.18.0.dev20240801.dist-info/WHEEL,sha256=kGT74LWyRUZrL4VgLh6_g12IeVl_9u9ZVhadrgXZUEY,110
-tf_models_nightly-2.18.0.dev20240801.dist-info/top_level.txt,sha256=gum2FfO5R4cvjl2-QtP-S1aNmsvIZaFFT6VFzU0f4-g,33
-tf_models_nightly-2.18.0.dev20240801.dist-info/RECORD,,
+tf_models_nightly-2.18.0.dev20240803.dist-info/AUTHORS,sha256=1dG3fXVu9jlo7bul8xuix5F5vOnczMk7_yWn4y70uw0,337
+tf_models_nightly-2.18.0.dev20240803.dist-info/LICENSE,sha256=WxeBS_DejPZQabxtfMOM_xn8qoZNJDQjrT7z2wG1I4U,11512
+tf_models_nightly-2.18.0.dev20240803.dist-info/METADATA,sha256=F3guzvtjxT-14NlHuPYJuPjzAky-wKNuPcr6Tpc9KNk,1432
+tf_models_nightly-2.18.0.dev20240803.dist-info/WHEEL,sha256=kGT74LWyRUZrL4VgLh6_g12IeVl_9u9ZVhadrgXZUEY,110
+tf_models_nightly-2.18.0.dev20240803.dist-info/top_level.txt,sha256=gum2FfO5R4cvjl2-QtP-S1aNmsvIZaFFT6VFzU0f4-g,33
+tf_models_nightly-2.18.0.dev20240803.dist-info/RECORD,,