Skip to content
Snippets Groups Projects
Commit 0c76d950 authored by Manuel Günther's avatar Manuel Günther
Browse files

Added --parallel command line option to the bin/verify.py script (and all derived scripts).

parent e45cd27c
No related branches found
No related tags found
No related merge requests found
...@@ -165,7 +165,7 @@ def train_enroller(algorithm, extractor, force = False): ...@@ -165,7 +165,7 @@ def train_enroller(algorithm, extractor, force = False):
train_features = read_features(train_files, reader, True) train_features = read_features(train_files, reader, True)
# perform training # perform training
logger.info("- Enrollment: training enroller '%s' using %d identities: ", fs.enroller_file, len(train_features)) logger.info("- Enrollment: training enroller '%s' using %d identities", fs.enroller_file, len(train_features))
algorithm.train_enroller(train_features, fs.enroller_file) algorithm.train_enroller(train_features, fs.enroller_file)
......
...@@ -131,6 +131,8 @@ def command_line_parser(description=__doc__, exclude_resources_from=[]): ...@@ -131,6 +131,8 @@ def command_line_parser(description=__doc__, exclude_resources_from=[]):
help = 'Performs score calibration after the scores are computed.') help = 'Performs score calibration after the scores are computed.')
flag_group.add_argument('-z', '--zt-norm', action='store_true', flag_group.add_argument('-z', '--zt-norm', action='store_true',
help = 'Enable the computation of ZT norms') help = 'Enable the computation of ZT norms')
flag_group.add_argument('-r', '--parallel', type=int,
help = 'This flag is a shortcut for running the commands on the local machine with the given amount of parallel threads; equivalent to --grid bob.bio.base.grid.Grid("local", number_of_parallel_threads=X) --run-local-scheduler --stop-on-failure.')
return { return {
'main' : parser, 'main' : parser,
...@@ -191,6 +193,11 @@ def initialize(parsers, command_line_parameters = None, skips = []): ...@@ -191,6 +193,11 @@ def initialize(parsers, command_line_parameters = None, skips = []):
if skip not in args.execute_only: if skip not in args.execute_only:
exec("args.skip_%s = True" % (skip.replace("-", "_"))) exec("args.skip_%s = True" % (skip.replace("-", "_")))
if args.parallel is not None:
args.grid = ['bob.bio.base.grid.Grid("local", number_of_parallel_processes = %d)' % args.parallel]
args.run_local_scheduler = True
args.stop_on_failure = True
# logging # logging
bob.core.log.set_verbosity_level(logger, args.verbose) bob.core.log.set_verbosity_level(logger, args.verbose)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment