def test_rpn(image_set, year, root_path, devkit_path, prefix, epoch, ctx, vis=False): # load symbol sym = get_vgg_rpn_test() # load testing data voc, roidb = load_gt_roidb(image_set, year, root_path, devkit_path) test_data = ROIIter(roidb, batch_size=1, shuffle=False, mode='test') # load model args, auxs, _ = load_param(prefix, epoch, convert=True, ctx=ctx) # start testing detector = Detector(sym, ctx, args, auxs) imdb_boxes = generate_detections(detector, test_data, voc, vis=vis) voc.evaluate_recall(roidb, candidate_boxes=imdb_boxes)
def test_rpn(image_set, year, root_path, devkit_path, trained, epoch, ctx): from rcnn.rpn.generate import Detector, generate_detections # load symbol sym = get_vgg_rpn_test() # load testing data voc, roidb = load_gt_roidb(image_set, year, root_path, devkit_path) test_data = ROIIter(roidb, batch_size=1, shuffle=False, mode="test") # load trained args, auxs = load_param(trained, epoch, convert=True, ctx=ctx[0]) # start testing detector = Detector(sym, ctx[0], args, auxs) imdb_boxes = generate_detections(detector, test_data, voc, vis=False) voc.evaluate_recall(roidb, candidate_boxes=imdb_boxes)
def test_rpn(image_set, year, root_path, devkit_path, trained, epoch, ctx): from rcnn.rpn.generate import Detector, generate_detections # load symbol sym = get_vgg_rpn_test() # load testing data voc, roidb = load_gt_roidb(image_set, year, root_path, devkit_path) test_data = ROIIter(roidb, batch_size=1, shuffle=False, mode='test') # load trained args, auxs = load_param(trained, epoch, convert=True, ctx=ctx[0]) # start testing detector = Detector(sym, ctx[0], args, auxs) imdb_boxes = generate_detections(detector, test_data, voc, vis=False) voc.evaluate_recall(roidb, candidate_boxes=imdb_boxes)
def test_rpn(image_set, year, root_path, devkit_path, prefix, epoch, ctx, vis): # set config config.TEST.HAS_RPN = True config.TEST.RPN_PRE_NMS_TOP_N = -1 config.TEST.RPN_POST_NMS_TOP_N = 2000 # load symbol sym = get_vgg_rpn_test() # load testing data voc, roidb = load_gt_roidb(image_set, year, root_path, devkit_path) test_data = ROIIter(roidb, batch_size=1, shuffle=False, mode='test') # load model args, auxs = load_param(prefix, epoch, convert=True, ctx=ctx) # start testing detector = Detector(sym, ctx, args, auxs) imdb_boxes = generate_detections(detector, test_data, voc, vis=vis) voc.evaluate_recall(roidb, candidate_boxes=imdb_boxes)