fix lzx1413#20

add classnum setting

Author: lzx1413

models/RefineSSD_vgg.py

@@ -88,10 +88,10 @@ def __init__(self, size, num_classes, use_refine=False):
                                       nn.Conv2d(256, 12, kernel_size=3, stride=1, padding=1), \
                                       nn.Conv2d(256, 12, kernel_size=3, stride=1, padding=1), \
                                       ])
-        self.odm_conf = nn.ModuleList([nn.Conv2d(256, 63, kernel_size=3, stride=1, padding=1), \
-                                       nn.Conv2d(256, 63, kernel_size=3, stride=1, padding=1), \
-                                       nn.Conv2d(256, 63, kernel_size=3, stride=1, padding=1), \
-                                       nn.Conv2d(256, 63, kernel_size=3, stride=1, padding=1), \
+        self.odm_conf = nn.ModuleList([nn.Conv2d(256, 3*num_classes, kernel_size=3, stride=1, padding=1), \
+                                       nn.Conv2d(256, 3*num_classes, kernel_size=3, stride=1, padding=1), \
+                                       nn.Conv2d(256, 3*num_classes, kernel_size=3, stride=1, padding=1), \
+                                       nn.Conv2d(256, 3*num_classes, kernel_size=3, stride=1, padding=1), \
                                        ])
         self.trans_layers = nn.ModuleList([nn.Sequential(nn.Conv2d(512, 256, kernel_size=3, stride=1, padding=1),
                                                          nn.ReLU(inplace=True),