From 9a42975803e2346d30b53904d95c015b02cc4f7e Mon Sep 17 00:00:00 2001 From: iramirsina Date: Wed, 25 Oct 2017 17:47:27 -0400 Subject: [PATCH] Modification of multibox-The operation is applied on Conv4-3 --- ssd.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ssd.py b/ssd.py index cd570064c..ef92e91a0 100644 --- a/ssd.py +++ b/ssd.py @@ -163,7 +163,7 @@ def add_extras(cfg, i, batch_norm=False): def multibox(vgg, extra_layers, cfg, num_classes): loc_layers = [] conf_layers = [] - vgg_source = [24, -2] + vgg_source = [21, -2] for k, v in enumerate(vgg_source): loc_layers += [nn.Conv2d(vgg[v].out_channels, cfg[k] * 4, kernel_size=3, padding=1)]