original pvanet FC

CharlesShang · Mar 6, 2017 · ba99717 · ba99717
1 parent 5bf2fc9
commit ba99717
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/lib/networks/PVAnet_train.py b/lib/networks/PVAnet_train.py
@@ -131,12 +131,12 @@ def setup(self):
         #  .concat(axis=3, name='convf'))
 
         (self.feed('convf_2', 'roi-data')
-         .roi_pool(5, 5, 1.0 / 16, name='roi_pooling')
-         .fc(1024, name='fc6', relu=False)
-         .bn_scale_combo(c_in = 1024, name='fc6', relu=True)
+         .roi_pool(6, 6, 1.0 / 16, name='roi_pooling')
+         .fc(4096, name='fc6', relu=False)
+         .bn_scale_combo(c_in = 4096, name='fc6', relu=True)
          .dropout(0.5, name='fc6/drop6')
-         .fc(1024, name='fc7', relu=False)
-         .bn_scale_combo(c_in=1024, name='fc7', relu=True)
+         .fc(4096, name='fc7', relu=False)
+         .bn_scale_combo(c_in=4096, name='fc7', relu=True)
          .dropout(0.5, name='fc7/drop7')
          .fc(n_classes, relu=False, name='cls_score')
          .softmax(name='cls_prob'))