change threshold for testing

Author: Xinlei Chen

lib/model/test.py

@@ -135,7 +135,7 @@ def apply_nms(all_boxes, thresh):
       nms_boxes[cls_ind][im_ind] = dets[keep, :].copy()
   return nms_boxes
 
-def test_net(sess, net, imdb, weights_filename, max_per_image=100, thresh=0.05):
+def test_net(sess, net, imdb, weights_filename, max_per_image=100, thresh=0.):
   np.random.seed(cfg.RNG_SEED)
   """Test a Fast R-CNN network on an image database."""
   num_images = len(imdb.image_index)

lib/nets/network.py

@@ -281,7 +281,8 @@ def _add_losses(self, sigma_rpn=3.0):
       self._losses['rpn_loss_box'] = rpn_loss_box
 
       loss = cross_entropy + loss_box + rpn_cross_entropy + rpn_loss_box
-      self._losses['total_loss'] = loss
+      regularization_loss = tf.add_n(tf.losses.get_regularization_losses(), 'regu')
+      self._losses['total_loss'] = loss + regularization_loss
 
       self._event_summaries.update(self._losses)