summaryrefslogtreecommitdiff
path: root/megapixels
diff options
context:
space:
mode:
authoradamhrv <adam@ahprojects.com>2019-06-28 03:24:30 +0200
committeradamhrv <adam@ahprojects.com>2019-06-28 03:24:30 +0200
commit3f54493f0edda31b4f425309db63307c3be90b18 (patch)
tree39b4ac58db7ea1902ae20abbe2b7841afaa5d051 /megapixels
parent8d4aa1c5b24eb8770633cd586eb85617b32bb5c5 (diff)
msc working
Diffstat (limited to 'megapixels')
-rw-r--r--megapixels/commands/msc/summarize.py4
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')