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
