Small fixes of layer numbers for MobileNetV3

qfgaohao · Apr 5, 2020 · eaa555e · eaa555e
1 parent 5e5a756
commit eaa555e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/vision/ssd/mobilenetv3_ssd_lite.py b/vision/ssd/mobilenetv3_ssd_lite.py
@@ -24,7 +24,7 @@ def SeperableConv2d(in_channels, out_channels, kernel_size=1, stride=1, padding=
 def create_mobilenetv3_large_ssd_lite(num_classes, width_mult=1.0, use_batch_norm=True, onnx_compatible=False, is_test=False):
     base_net = MobileNetV3_Large().features
 
-    source_layer_indexes = [ 16, 20 ]
+    source_layer_indexes = [ 15, 21 ]
     extras = ModuleList([
         Block(3, 960, 256, 512, hswish(), None, stride=2),
         Block(3, 512, 128, 256, hswish(), None, stride=2),
@@ -58,7 +58,7 @@ def create_mobilenetv3_large_ssd_lite(num_classes, width_mult=1.0, use_batch_nor
 def create_mobilenetv3_small_ssd_lite(num_classes, width_mult=1.0, use_batch_norm=True, onnx_compatible=False, is_test=False):
     base_net = MobileNetV3_Small().features
 
-    source_layer_indexes = [ 11, 16 ]
+    source_layer_indexes = [ 11, 17 ]
     extras = ModuleList([
         Block(3, 576, 256, 512, hswish(), None, stride=2),
         Block(3, 512, 128, 256, hswish(), None, stride=2),