parser.add_argument('--num_images', type=int, default=0, help='number of images to plot') parser.add_argument('--image_dir', type=str, help='folder to save sampled images') parser.add_argument('--score_threshold', type=float, help='threshold for predicted scores.') parser.add_argument('--output', type=str, help='file to save detected boxes.') args = parser.parse_args(gen_be=False) if args.model_file is None: parser.print_usage() exit('You need to specify model file to evaluate.') if args.ssd_config: args.ssd_config = { k: v for k, v in [ss.split(':') for ss in args.ssd_config] } config = json.load(open(args.ssd_config['val']), object_pairs_hook=OrderedDict) if args.batch_size == 0: args.batch_size = config["batch_size"] # setup backend
Simple example of using the dataloader with pre-generated augmentation data """ arg_defaults = {'batch_size': 0} parser = NeonArgparser(__doc__, default_overrides=arg_defaults) parser.add_argument('--ssd_config', action='append', required=True, help='ssd json file path') parser.add_argument('--height', type=int, help='image height') parser.add_argument('--width', type=int, help='image width') parser.add_argument('--num_images', type=int, default=0, help='number of images to plot') parser.add_argument('--image_dir', type=str, help='folder to save sampled images') parser.add_argument('--score_threshold', type=float, help='threshold for predicted scores.') parser.add_argument('--output', type=str, help='file to save detected boxes.') args = parser.parse_args(gen_be=False) if args.model_file is None: parser.print_usage() exit('You need to specify model file to evaluate.') if args.ssd_config: args.ssd_config = {k: v for k, v in [ss.split(':') for ss in args.ssd_config]} config = json.load(open(args.ssd_config['val']), object_pairs_hook=OrderedDict) if args.batch_size == 0: args.batch_size = config["batch_size"] # setup backend be = gen_backend(backend=args.backend, batch_size=args.batch_size, device_id=args.device_id, compat_mode='caffe', rng_seed=1, deterministic_update=True, deterministic=True) be.enable_winograd = 0