Skip to content
Snippets Groups Projects

Using the proper verify script depending on system

Merged Tiago de Freitas Pereira requested to merge fix-119 into master
@@ -8,13 +8,12 @@ A script to run biometric recognition baselines
from .. import load_resource
from .verify import main as verify
import os
from ..baseline import get_available_databases, search_preprocessor
from bob.extension.scripts.click_helper import verbosity_option
import click
import tempfile
import logging
import os
logger = logging.getLogger("bob.bio.base")
@@ -23,9 +22,11 @@ logger = logging.getLogger("bob.bio.base")
'allow_extra_args': True})
@click.argument('baseline', required=True)
@click.argument('database', required=True)
@click.option('--verify', default='verify', show_default=True,
type=click.Choice(('verify', 'gmm', 'isv', 'ivector')))
@verbosity_option()
@click.pass_context
def baseline(ctx, baseline, database, **kwargs):
def baseline(ctx, baseline, database, verify, **kwargs):
"""Run a biometric recognition baseline.
\b
@@ -61,7 +62,7 @@ def baseline(ctx, baseline, database, **kwargs):
preprocessor = loaded_baseline.preprocessors[db]
# this is the default sub-directory that is used
if "-T" in ctx.args or "--temp-directory" in ctx.args:
if "-T" in ctx.args or "--temp-directory" in ctx.args:
sub_directory = os.path.join(database, baseline)
else:
sub_directory = baseline
@@ -89,12 +90,20 @@ verbose = {verbose}
verbose=ctx.meta['verbosity'],
)
if verify == "verify":
from .verify import main as main_verify
elif verify == "gmm":
from bob.bio.gmm.script.verify_gmm import main as main_verify
elif verify == "isv":
from bob.bio.gmm.script.verify_isv import main as main_verify
elif verify == "ivector":
from bob.bio.gmm.script.verify_ivector import main as main_verify
with tempfile.NamedTemporaryFile(mode='w+t', prefix='{}_'.format(baseline),
suffix='.py', delete=False, dir='.') as f:
f.write(config)
f.flush()
f.seek(0)
verify([f.name] + ctx.args)
main_verify([f.name] + ctx.args)
click.echo("You may want to delete `{}' after the experiments are "
"finished running.".format(f.name))
"finished running.".format(f.name))
Loading