Missed fix for batch size 1

Author: ruotianluo

lib/nets/network.py

@@ -240,7 +240,7 @@ def _region_proposal(self, net_conv):
     rpn_cls_score = self.rpn_cls_score_net(rpn) # batch * (num_anchors * 2) * h * w
 
     # change it so that the score has 2 as its channel size
-    rpn_cls_score_reshape = rpn_cls_score.view(self._batch_size, 2, -1, rpn_cls_score.size()[-1]) # batch * 2 * (num_anchors*h) * w
+    rpn_cls_score_reshape = rpn_cls_score.view(1, 2, -1, rpn_cls_score.size()[-1]) # batch * 2 * (num_anchors*h) * w
     rpn_cls_prob_reshape = F.softmax(rpn_cls_score_reshape)
 
     # Move channel to the last dimenstion, to fit the input of python functions