From 9ce408aa5be9bf6008bdb04f3efb680a1e5250f9 Mon Sep 17 00:00:00 2001 From: Yu Xiang Date: Thu, 16 Jun 2016 16:05:18 -0700 Subject: [PATCH] tmp --- fast-rcnn/experiments/cfgs/pascal_rpn_cls_vgg16.yml | 2 +- fast-rcnn/experiments/scripts/pascal2007_vgg16_rpn_cls.sh | 2 +- fast-rcnn/lib/fast_rcnn/test.py | 4 ++-- 3 files changed, 4 insertions(+), 4 deletions(-) diff --git a/fast-rcnn/experiments/cfgs/pascal_rpn_cls_vgg16.yml b/fast-rcnn/experiments/cfgs/pascal_rpn_cls_vgg16.yml index 805bf1b..e964884 100644 --- 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 index 756328d..e039abe 100755 --- 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 index 44ab834..6d06b85 100644 --- 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()