Beispiel #1
0
def load_rpn_roidb(image_set, year, root_path, devkit_path, flip=False):
    voc = PascalVOC(image_set, year, root_path, devkit_path)
    gt_roidb = voc.gt_roidb()
    rpn_roidb = voc.rpn_roidb(gt_roidb)
    if flip:
        rpn_roidb = voc.append_flipped_images(rpn_roidb)
    prepare_roidb(voc, rpn_roidb)
    means, stds = add_bbox_regression_targets(rpn_roidb)
    return voc, rpn_roidb, means, stds
Beispiel #2
0
def load_test_rpn_roidb(image_set, year, root_path, devkit_path):
    voc = PascalVOC(image_set, year, root_path, devkit_path)
    gt_roidb = voc.gt_roidb()
    rpn_roidb = voc.rpn_roidb(gt_roidb)
    prepare_roidb(voc, rpn_roidb)
    return voc, rpn_roidb