Skip to content

Commit afebb30

Browse files
astorfiamdegroot
authored andcommitted
Modification of multibox-The operation is applied on Conv4-3 (amdegroot#70)
1 parent 4701d83 commit afebb30

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

ssd.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -165,7 +165,7 @@ def add_extras(cfg, i, batch_norm=False):
165165
def multibox(vgg, extra_layers, cfg, num_classes):
166166
loc_layers = []
167167
conf_layers = []
168-
vgg_source = [24, -2]
168+
vgg_source = [21, -2]
169169
for k, v in enumerate(vgg_source):
170170
loc_layers += [nn.Conv2d(vgg[v].out_channels,
171171
cfg[k] * 4, kernel_size=3, padding=1)]

0 commit comments

Comments
 (0)