add efficientdet-lite3x and update image_size=640 for efficientdet-li…

…te4. PiperOrigin-RevId: 369328400
tensorflow · Apr 19, 2021 · 04d5972 · 04d5972
1 parent 87973cc
commit 04d5972
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/tensorflow_examples/lite/model_maker/third_party/efficientdet/hparams_config.py b/tensorflow_examples/lite/model_maker/third_party/efficientdet/hparams_config.py
@@ -439,11 +439,22 @@ def default_detection_configs():
             box_class_repeats=4,
             **lite_common_param,
         ),
+    'efficientdet-lite3x':
+        dict(
+            name='efficientdet-lite3x',
+            backbone_name='efficientnet-lite3',
+            image_size=640,
+            fpn_num_filters=200,
+            fpn_cell_repeats=6,
+            box_class_repeats=4,
+            anchor_scale=3.0,
+            **lite_common_param,
+        ),
     'efficientdet-lite4':
         dict(
             name='efficientdet-lite4',
             backbone_name='efficientnet-lite4',
-            image_size=512,
+            image_size=640,
             fpn_num_filters=224,
             fpn_cell_repeats=7,
             box_class_repeats=4,