File size: 1,794 Bytes
0367344 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 |
#!/usr/bin/env python
# encoding: utf-8
"""
@author: MingDong
@file: eval_lfw_blufr.py
@desc: test lfw accuracy on blufr protocol
"""
import argparse
import scipy.io as sio
def readName(file='pairs.txt'):
name_list = []
f = open(file, 'r')
lines = f.readlines()
for line in lines[1:]:
line_split = line.rstrip().split()
if len(line_split) == 3:
name_list.append(line_split[0])
elif len(line_split) == 4:
name_list.append(line_split[0])
name_list.append(line_split[2])
else:
print('wrong file, please check again')
return list(set(name_list))
def main(args):
blufr_info = sio.loadmat(args.lfw_blufr_file)
#print(blufr_info)
name_list = readName()
image = blufr_info['imageList']
missing_files = []
for i in range(image.shape[0]):
name = image[i][0][0]
index = name.rfind('_')
name = name[0:index]
if name not in name_list:
print(name)
missing_files.append(name)
print('lfw pairs.txt total persons: ', len(name_list))
print('blufr_mat_missing persons: ', len(missing_files))
# Some of the missing file:
# Zdravko_Mucic
# Zelma_Novelo
# Zeng_Qinghong
# Zumrati_Juma
# lfw pairs.txt total persons: 4281
# blufr_mat_missing persons: 1549
if __name__ == '__main__':
parser = argparse.ArgumentParser(description='lfw blufr test')
parser.add_argument('--lfw_blufr_file', type=str, default='./blufr_lfw_config.mat', help='feature dimension')
parser.add_argument('--lfw_pairs.txt', type=str, default='./pairs.txt', help='feature dimension')
parser.add_argument('--gpus', type=str, default='2,3', help='gpu list')
args = parser.parse_args()
main(args)
|