Modification of multibox-The operation is applied on Conv4-3 (#70)

amdegroot · Feb 15, 2018 · afebb30 · afebb30
1 parent 4701d83
commit afebb30
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ssd.py b/ssd.py
@@ -165,7 +165,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)]