From 3f54493f0edda31b4f425309db63307c3be90b18 Mon Sep 17 00:00:00 2001 From: adamhrv Date: Fri, 28 Jun 2019 03:24:30 +0200 Subject: msc working --- megapixels/commands/msc/summarize.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) (limited to 'megapixels') diff --git a/megapixels/commands/msc/summarize.py b/megapixels/commands/msc/summarize.py index 8df32c45..0d6bc9e1 100644 --- a/megapixels/commands/msc/summarize.py +++ b/megapixels/commands/msc/summarize.py @@ -8,6 +8,9 @@ from app.utils.logger_utils import Logger log = Logger.getLogger() +#dataset_names = ['helen', 'megaface', 'adience', 'pipa', 'lfpw', 'brainwash', 'msceleb', 'duke_mtmc', 'uccs'] +dataset_names = ['who_goes_there', 'megaface', 'ibm_dif'] + @click.command() @click.option('-i', '--input', 'opt_fp_in', required=True) @click.option('-o', '--output', 'opt_fp_out', required=True) @@ -29,7 +32,6 @@ def cli(ctx, opt_fp_in, opt_fp_out): log = Logger.getLogger() - dataset_names = ['helen', 'megaface', 'adience', 'pipa', 'lfpw', 'brainwash', 'msceleb', 'duke_mtmc', 'uccs'] df = pd.DataFrame() fp_out = opt_fp_out.replace('.csv', '_citations.csv') -- cgit v1.2.3-70-g09d2