From 599c3834bd28c8a4abea465ba41cedb18c7e42e7 Mon Sep 17 00:00:00 2001 From: Jia Guo Date: Fri, 8 Dec 2017 22:24:29 +0800 Subject: [PATCH] remove cfp_ff, do ver on flip only --- src/eval/verification.py | 7 +++++-- src/train_softmax.py | 2 +- 2 files changed, 6 insertions(+), 3 deletions(-) diff --git a/src/eval/verification.py b/src/eval/verification.py index 75ef282..b3eb841 100644 --- a/src/eval/verification.py +++ b/src/eval/verification.py @@ -235,8 +235,11 @@ def test(data_set, mx_model, batch_size): embeddings = embeddings_list[0].copy() embeddings = sklearn.preprocessing.normalize(embeddings) - _, _, accuracy, val, val_std, far = evaluate(embeddings, issame_list, nrof_folds=10) - acc1, std1 = np.mean(accuracy), np.std(accuracy) + acc1 = 0.0 + std1 = 0.0 + #_, _, accuracy, val, val_std, far = evaluate(embeddings, issame_list, nrof_folds=10) + #acc1, std1 = np.mean(accuracy), np.std(accuracy) + #print('Validation rate: %2.5f+-%2.5f @ FAR=%2.5f' % (val, val_std, far)) #embeddings = np.concatenate(embeddings_list, axis=1) embeddings = embeddings_list[0] + embeddings_list[1] diff --git a/src/train_softmax.py b/src/train_softmax.py index 0b56870..fbcc839 100644 --- a/src/train_softmax.py +++ b/src/train_softmax.py @@ -416,7 +416,7 @@ def train_net(args): ver_list = [] ver_name_list = [] - for name in ['lfw','cfp_ff','cfp_fp','agedb_30']: + for name in ['lfw','cfp_fp','agedb_30']: path = os.path.join(args.data_dir,name+".bin") if os.path.exists(path): data_set = verification.load_bin(path, image_size)