Esempio n. 1
0
def main():
    args = parse_args()
    ctx = mx.gpu(args.gpu)
    print(args)
    test_rcnn(args.network, args.dataset, args.root_path, args.subset,
              args.split, ctx, args.prefix, args.epoch, args.vis, args.shuffle,
              args.has_rpn, args.proposal, args.thresh)
Esempio n. 2
0
def main():
    args = parse_args()
    logger.info('Called with argument: %s' % args)
    ctx = mx.gpu(args.gpu)
    test_rcnn(args.network, args.dataset, args.image_set, args.root_path, args.dataset_path,
              ctx, args.prefix, args.epoch,
              args.vis, args.shuffle, args.has_rpn, args.proposal, args.thresh)
Esempio n. 3
0
def main():
    args = parse_args()
    ctx = mx.gpu(args.gpu)
    print(args)
    test_rcnn(args.network, args.dataset, args.image_set, args.root_path, args.dataset_path,
              ctx, args.prefix, args.epoch,
              args.vis, args.shuffle, args.has_rpn, args.proposal, args.thresh)
Esempio n. 4
0
def main():
    args = parse_args()
    ctx = mx.gpu(args.gpu)
    print(args)
    test_rcnn(args.network, args.dataset, args.image_set, args.root_path,
              args.dataset_path, ctx, args.prefix, args.epoch, args.vis,
              args.shuffle, args.has_rpn, args.proposal, args.thresh,
              args.use_global_context, args.use_roi_align, args.use_box_voting,
              args.detailed_analysis)
Esempio n. 5
0
def main():
    args = parse_args()
    ctx = mx.gpu(args.gpu)
    print(args)
    cc = CrayonClient(hostname='10.132.90.242')
    if args.exp_name is None:
        args.exp_name = datetime.now().strftime('frcnnEval_%m-%d')
    try:
        exp = cc.create_experiment(args.exp_name)
    except:
        exp = cc.open_experiment(args.exp_name)

    for x in args.epoch.split(","):
        mAp = test_rcnn(args.network, args.dataset, args.image_set,
                        args.root_path, args.dataset_path, ctx, args.prefix,
                        int(x), args.vis, args.shuffle, args.has_rpn,
                        args.proposal, args.thresh, args.use_global_context,
                        args.use_roi_align)
        exp.add_scalar_value('mAp', mAp)
    return
Esempio n. 6
0
import mxnet as mx

from rcnn.tools.test_rcnn import parse_args
from rcnn.tools.test_rcnn import test_rcnn

if __name__ == '__main__':
    args = parse_args()
    ctx = mx.gpu(args.gpu)
    test_rcnn(args, ctx, args.prefix, args.epoch,
              vis=args.vis, shuffle=args.shuffle, has_rpn=args.has_rpn, proposal=args.proposal)