Internal change

PiperOrigin-RevId: 405689756

official/projects/edgetpu/vision/configs/mobilenet_edgetpu_config.py

@@ -193,7 +193,12 @@ def mobilenet_edgetpu() -> cfg.ExperimentConfig:
 
 # Registration for MobileNet-EdgeTPU-V1 models.
 # We use 'depth_multiplier' to scale the models.
-# E.g. dm1p25 implies depth multiplier of 1.25x
+# E.g. dm0p75 implies depth multiplier of 0.75x
+@exp_factory.register_config_factory('mobilenet_edgetpu_dm0p75')
+def mobilenet_edgetpu_dm0p75() -> cfg.ExperimentConfig:
+  return mobilenet_edgetpu_base_experiment_config('mobilenet_edgetpu_dm0p75')
+
+
 @exp_factory.register_config_factory('mobilenet_edgetpu_dm1p25')
 def mobilenet_edgetpu_dm1p25() -> cfg.ExperimentConfig:
   return mobilenet_edgetpu_base_experiment_config('mobilenet_edgetpu_dm1p25')

official/projects/edgetpu/vision/modeling/backbones/mobilenet_edgetpu.py

@@ -41,6 +41,7 @@ MOBILENET_EDGETPU_V2_CONFIGS = frozenset([
 # MobileNet-EdgeTPU-V1 configs.
 MOBILENET_EDGETPU_CONFIGS = frozenset([
     'mobilenet_edgetpu',
+    'mobilenet_edgetpu_dm0p75',
     'mobilenet_edgetpu_dm1p25',
     'mobilenet_edgetpu_dm1p5',
     'mobilenet_edgetpu_dm1p75',

official/projects/edgetpu/vision/modeling/mobilenet_edgetpu_v1_model.py

@@ -27,6 +27,7 @@ ModelConfig = mobilenet_edgetpu_v1_model_blocks.ModelConfig
 MODEL_CONFIGS = {
     # (width, depth, resolution, dropout)
     'mobilenet_edgetpu': ModelConfig.from_args(1.0, 1.0, 224, 0.1),
+    'mobilenet_edgetpu_dm0p75': ModelConfig.from_args(0.75, 1.0, 224, 0.1),
     'mobilenet_edgetpu_dm1p25': ModelConfig.from_args(1.25, 1.0, 224, 0.1),
     'mobilenet_edgetpu_dm1p5': ModelConfig.from_args(1.5, 1.0, 224, 0.1),
     'mobilenet_edgetpu_dm1p75': ModelConfig.from_args(1.75, 1.0, 224, 0.1)

official/projects/edgetpu/vision/tasks/image_classification_test.py

@@ -34,6 +34,7 @@ class ImageClassificationTaskTest(tf.test.TestCase, parameterized.TestCase):
                             ('mobilenet_edgetpu_v2_m'),
                             ('mobilenet_edgetpu_v2_l'),
                             ('mobilenet_edgetpu'),
+                            ('mobilenet_edgetpu_dm0p75'),
                             ('mobilenet_edgetpu_dm1p25'),
                             ('mobilenet_edgetpu_dm1p5'),
                             ('mobilenet_edgetpu_dm1p75'))