Internal change

PiperOrigin-RevId: 453736287

official/vision/modeling/backbones/resnet_deeplab.py

@@ -43,6 +43,18 @@ RESNET_SPECS = {
         ('bottleneck', 256, 23),
         ('bottleneck', 512, 3),
     ],
+    152: [
+        ('bottleneck', 64, 3),
+        ('bottleneck', 128, 8),
+        ('bottleneck', 256, 36),
+        ('bottleneck', 512, 3),
+    ],
+    200: [
+        ('bottleneck', 64, 3),
+        ('bottleneck', 128, 24),
+        ('bottleneck', 256, 36),
+        ('bottleneck', 512, 3),
+    ],
 }
 
 

official/vision/modeling/backbones/resnet_deeplab_test.py

@@ -29,8 +29,12 @@ class ResNetTest(parameterized.TestCase, tf.test.TestCase):
   @parameterized.parameters(
       (128, 50, 4, 8),
       (128, 101, 4, 8),
+      (128, 152, 4, 8),
+      (128, 200, 4, 8),
       (128, 50, 4, 16),
       (128, 101, 4, 16),
+      (128, 152, 4, 16),
+      (128, 200, 4, 16),
   )
   def test_network_creation(self, input_size, model_id,
                             endpoint_filter_scale, output_stride):