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),