diff options
| -rw-r--r-- | megapixels/commands/msc/summarize.py | 4 |
1 files changed, 3 insertions, 1 deletions
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') |