tmp

tanshen · Jun 16, 2016 · 9ce408a · 9ce408a
1 parent f6d37d0
commit 9ce408a
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/fast-rcnn/experiments/cfgs/pascal_rpn_cls_vgg16.yml b/fast-rcnn/experiments/cfgs/pascal_rpn_cls_vgg16.yml
@@ -21,7 +21,7 @@ TEST:
   NUM_PER_OCTAVE: 19
   ASPECTS: !!python/tuple [3.0, 2.0, 1.5, 1.0, 0.75, 0.5, 0.25, 0.125]
   NMS: 0.5
-  ROI_THRESHOLD: 0.005
+  ROI_THRESHOLD: 0.3
   ROI_NUM: 2000
   BBOX_REG: True
   SUBCLS: False
diff --git a/fast-rcnn/experiments/scripts/pascal2007_vgg16_rpn_cls.sh b/fast-rcnn/experiments/scripts/pascal2007_vgg16_rpn_cls.sh
@@ -18,7 +18,7 @@ echo Logging output to "$LOG"
 
 time ./tools/test_net.py --gpu $1 \
   --def models/VGG16/pascal2007/test_rpn_cls.prototxt \
-  --net output/pascal2007/voc_2007_trainval/vgg16_fast_rcnn_rpn_cls_pascal2007_iter_10000.caffemodel \
+  --net output/pascal2007/voc_2007_trainval/vgg16_fast_rcnn_rpn_cls_pascal2007_iter_5000.caffemodel \
   --imdb voc_2007_test \
   --cfg experiments/cfgs/pascal_rpn_cls_vgg16.yml
 

diff --git a/fast-rcnn/lib/fast_rcnn/test.py b/fast-rcnn/lib/fast_rcnn/test.py
@@ -448,7 +448,7 @@ def im_detect_proposal(net, im, boxes_grid, num_classes, num_subclasses, subclas
     print scores.shape
 
     # draw boxes
-    if 0:
+    if 1:
         # print scores, pred_boxes.shape
         import matplotlib.pyplot as plt
         plt.imshow(im)
@@ -651,7 +651,7 @@ def test_net(net, imdb):
                     .astype(np.float32, copy=False)
             count = count + len(cls_scores)
 
-            if 1:
+            if 0:
                 keep = nms(all_boxes[j][i], cfg.TEST.NMS)
                 vis_detections(im, imdb.classes[j], all_boxes[j][i][keep, :])
         _t['misc'].toc()