summaryrefslogtreecommitdiff
path: root/eval.py
diff options
context:
space:
mode:
authorJordan Gong <jordan.gong@protonmail.com>2021-02-19 22:39:49 +0800
committerJordan Gong <jordan.gong@protonmail.com>2021-02-19 22:39:49 +0800
commitd12dd6b04a4e7c2b1ee43ab6f36f25d0c35ca364 (patch)
tree71b5209ce4b5cfb1d09b89fe133028bbfa481dc9 /eval.py
parent4aa9044122878a8e2b887a8b170c036983431559 (diff)
New branch with auto-encoder only
Diffstat (limited to 'eval.py')
-rw-r--r--eval.py30
1 files changed, 0 insertions, 30 deletions
diff --git a/eval.py b/eval.py
deleted file mode 100644
index c0505f8..0000000
--- a/eval.py
+++ /dev/null
@@ -1,30 +0,0 @@
-import numpy as np
-
-from config import config
-from models import Model
-from utils.dataset import ClipConditions
-from utils.misc import set_visible_cuda
-
-set_visible_cuda(config['system'])
-model = Model(config['system'], config['model'], config['hyperparameter'])
-
-dataset_selectors = {
- 'nm': {'conditions': ClipConditions({r'nm-0\d'})},
- 'bg': {'conditions': ClipConditions({r'nm-0\d', r'bg-0\d'})},
- 'cl': {'conditions': ClipConditions({r'nm-0\d', r'cl-0\d'})},
-}
-
-accuracy = model.predict_all(config['model']['total_iters'], config['dataset'],
- dataset_selectors, config['dataloader'])
-rank = 5
-np.set_printoptions(formatter={'float': '{:5.2f}'.format})
-for n in range(rank):
- print(f'===Rank-{n + 1} Accuracy===')
- for (condition, accuracy_c) in accuracy.items():
- acc_excl_identical_view = accuracy_c[:, :, n].fill_diagonal_(0)
- num_gallery_views = (acc_excl_identical_view != 0).sum(0)
- acc_each_angle = acc_excl_identical_view.sum(0) / num_gallery_views
- print('{0}: {1} mean: {2:5.2f}'.format(
- condition, acc_each_angle.cpu().numpy() * 100,
- acc_each_angle.mean() * 100)
- )