ZF/insightface
1
0
Fork 0
mirror of https://github.com/deepinsight/insightface.git synced 2026-05-14 12:17:55 +00:00
Code Issues Packages Projects Releases Wiki Activity

refine repo structure

Browse Source
This commit is contained in:
nttstar
2020-11-06 13:59:21 +08:00
parent 9fc3cc9c0b
commit b774d6a1b7
309 changed files with 24974 additions and 34253 deletions
Download Patch File Download Diff File Expand all files Collapse all files

72
recognition/partial_fc/pytorch/IJB/plot_ax.py
View File

@@ -21,34 +21,54 @@ root = '/train/trainset/1'
#retina = '{}/glint-face/IJB/result/retina1.0/{}_result/cosface.npy'.format(root, target)
#glint_retinaface_fp16 = '{}/glint-face/IJB/result/glint_retinaface_fp16/{}_result/cosface.npy'.format(root, target)
retina_fp16_10_percents = '{}/glint-face/IJB/result/glint_retinaface_fp16_0.1/{}_result/arcface.npy'.format(root, target)
retina_fp32_10_percents = '{}/glint-face/IJB/result/retina_0.1_fp32/{}_result/cosface.npy'.format(root, target)
retina_fp16 = '{}/glint-face/IJB/result/glint_retinaface_fp16/{}_result/cosface.npy'.format(root, target)
celeb360k_final = '{}/glint-face/IJB/result/celeb360kfinal1.0/{}_result/cosface.npy'.format(root, target)
celeb360k_final_10_percents = '{}/glint-face/IJB/result/celeb360kfinal0.1/{}_result/cosface.npy'.format(root, target)
retina_4GPU = '{}/glint-face/IJB/result/anxiang_ms1m_retina/{}_result/cosface.npy'.format(root, target)
retina_4GPU_scale2 = '{}/glint-face/IJB/result/anxiang_retina_largelr/{}_result/cosface.npy'.format(root, target)
emore_percents_10 = '{}/glint-face/IJB/result/emore0.1/{}_result/cosface.npy'.format(root, target)
emore_percents_40 = '{}/glint-face/IJB/result/emore0.4/{}_result/cosface.npy'.format(root, target)
emore_percents_80 = '{}/glint-face/IJB/result/emore0.8/{}_result/cosface.npy'.format(root, target)
retina_fp16_10_percents = '{}/glint-face/IJB/result/glint_retinaface_fp16_0.1/{}_result/arcface.npy'.format(
root, target)
retina_fp32_10_percents = '{}/glint-face/IJB/result/retina_0.1_fp32/{}_result/cosface.npy'.format(
root, target)
retina_fp16 = '{}/glint-face/IJB/result/glint_retinaface_fp16/{}_result/cosface.npy'.format(
root, target)
celeb360k_final = '{}/glint-face/IJB/result/celeb360kfinal1.0/{}_result/cosface.npy'.format(
root, target)
celeb360k_final_10_percents = '{}/glint-face/IJB/result/celeb360kfinal0.1/{}_result/cosface.npy'.format(
root, target)
retina_4GPU = '{}/glint-face/IJB/result/anxiang_ms1m_retina/{}_result/cosface.npy'.format(
root, target)
retina_4GPU_scale2 = '{}/glint-face/IJB/result/anxiang_retina_largelr/{}_result/cosface.npy'.format(
root, target)
emore_percents_10 = '{}/glint-face/IJB/result/emore0.1/{}_result/cosface.npy'.format(
root, target)
emore_percents_40 = '{}/glint-face/IJB/result/emore0.4/{}_result/cosface.npy'.format(
root, target)
emore_percents_80 = '{}/glint-face/IJB/result/emore0.8/{}_result/cosface.npy'.format(
root, target)
#emore_percents_10 = '{}/glint-face/IJB/result/emore0.1/{}_result/cosface.npy'.format(root, target)
#emore_percents_10 = '{}/glint-face/IJB/result/emore_cosface_0.1_margin0.45/{}_result/cosface.npy'.format(root, target)
emore = '{}/glint-face/IJB/result/emore1.0/{}_result/cosface.npy'.format(root, target)
emore = '{}/glint-face/IJB/result/emore1.0/{}_result/cosface.npy'.format(
root, target)
#celeb360k_0_1 = '{}/glint-face/IJB/result/celeb360k_0.1/{}_result/cosface.npy'.format(root, target)
#celeb360k_1_0_1 = '{}/glint-face/IJB/result/celeb360k/{}_result/cosface.npy'.format(root, target)
save_path = '{}/glint-face/IJB'.format(root)
image_path = '{}/face/IJB_release/{}'.format(root, target)
methods = ['retina_fp16', 'retina_fp16_0.1', 'retina_fp32_0.1', 'celeb360k_final', 'celeb360k_final_10_percents']
methods = [
'retina_fp16', 'retina_fp16_0.1', 'retina_fp32_0.1', 'celeb360k_final',
'celeb360k_final_10_percents'
]
methods = ['retina_4GPU', 'retina_4GPU_scale2']
methods = ['emore', 'emore_percents_10']
methods = ['emore', 'emore_percents_10', 'emore_percents_40', 'emore_percents_80']
files = [retina_fp16, retina_fp16_10_percents, retina_fp32_10_percents, celeb360k_final, celeb360k_final_10_percents]
methods = [
'emore', 'emore_percents_10', 'emore_percents_40', 'emore_percents_80'
]
files = [
retina_fp16, retina_fp16_10_percents, retina_fp32_10_percents,
celeb360k_final, celeb360k_final_10_percents
]
#files = [retina_4GPU, retina_4GPU_scale2]
#files = [emore, emore_percents_10]
files = [emore, emore_percents_10, emore_percents_40, emore_percents_80]
def read_template_pair_list(path):
pairs = pd.read_csv(path, sep=' ', header=None).values
# print(pairs.shape)
@@ -60,8 +80,8 @@ def read_template_pair_list(path):
p1, p2, label = read_template_pair_list(
os.path.join('%s/meta' % image_path, '%s_template_pair_label.txt' % target.lower()))
os.path.join('%s/meta' % image_path,
'%s_template_pair_label.txt' % target.lower()))
scores = []
for file in files:
@@ -69,9 +89,10 @@ for file in files:
methods = np.array(methods)
scores = dict(zip(methods, scores))
colours = dict(zip(methods, sample_colours_from_colourmap(methods.shape[0], 'Set2')))
colours = dict(
zip(methods, sample_colours_from_colourmap(methods.shape[0], 'Set2')))
# x_labels = [1/(10**x) for x in np.linspace(6, 0, 6)]
x_labels = [10 ** -6, 10 ** -5, 10 ** -4, 10 ** -3, 10 ** -2, 10 ** -1]
x_labels = [10**-6, 10**-5, 10**-4, 10**-3, 10**-2, 10**-1]
tpr_fpr_table = PrettyTable(['Methods'] + [str(x) for x in x_labels])
fig = plt.figure()
for method in methods:
@@ -79,17 +100,22 @@ for method in methods:
roc_auc = auc(fpr, tpr)
fpr = np.flipud(fpr)
tpr = np.flipud(tpr) # select largest tpr at same fpr
plt.plot(fpr, tpr, color=colours[method], lw=1,
# label=('[%s (AUC = %0.4f %%)]' % (method.split('-')[-1], roc_auc * 100))
label = method)
plt.plot(
fpr,
tpr,
color=colours[method],
lw=1,
# label=('[%s (AUC = %0.4f %%)]' % (method.split('-')[-1], roc_auc * 100))
label=method)
tpr_fpr_row = []
tpr_fpr_row.append("%s-%s" % (method, target))
for fpr_iter in np.arange(len(x_labels)):
_, min_index = min(list(zip(abs(fpr - x_labels[fpr_iter]), range(len(fpr)))))
_, min_index = min(
list(zip(abs(fpr - x_labels[fpr_iter]), range(len(fpr)))))
# tpr_fpr_row.append('%.4f' % tpr[min_index])
tpr_fpr_row.append('%.2f' % (tpr[min_index] * 100))
tpr_fpr_table.add_row(tpr_fpr_row)
plt.xlim([10 ** -6, 0.1])
plt.xlim([10**-6, 0.1])
plt.ylim([0.30, 1.0])
plt.grid(linestyle='--', linewidth=1)
plt.xticks(x_labels)