From eaa555ea84317450b67476295a93aa9a6cf6f874 Mon Sep 17 00:00:00 2001 From: Alexey Chernov <4ernov@gmail.com> Date: Sun, 5 Apr 2020 23:27:23 +0300 Subject: [PATCH] Small fixes of layer numbers for MobileNetV3 --- vision/ssd/mobilenetv3_ssd_lite.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/vision/ssd/mobilenetv3_ssd_lite.py b/vision/ssd/mobilenetv3_ssd_lite.py index 89239fbf..470f386a 100644 --- 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),