CodeJam/faceMatch2.py at master · kelzha/CodeJam · GitHub

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
65
66
67
68
69
70
71
72
73
74
75
import tools2 as eigenTools
import os
from PIL import Image
from glob import glob
import numpy as np
from random import shuffle


#hi phil
file_list = glob('./training_dataset_cropped2/*.png')
shuffle(file_list)
# open image
im = Image.open(file_list[0]).convert("L")
# get original dimensions
H,W = np.shape(im)
# print 'shape=',(H,W)

im_number = len(file_list)
# fill array with rows as image
# and columns as pixels
print im_number

datasetSize = 100;
arr = np.zeros([datasetSize,H*W])
idArray = np.zeros([datasetSize,1])
testArray = np.zeros([im_number -datasetSize,H*W])
testId = np.zeros([im_number -datasetSize,1])


for i in range(len(arr)):
	filepath,filename = os.path.split(file_list[i])
	filtername,exts = os.path.splitext(filename)


	im = Image.open(file_list[i]).convert("L")
	imArray = np.asarray(im)


	idArray[i] = int(filtername.split('_')[0])
	arr[i,:] = np.reshape(np.asarray(im),[1,H*W])
for i in range(len(testArray)):
	offset = len(arr)
	filepath,filename = os.path.split(file_list[i+offset])
	filtername,exts = os.path.splitext(filename)
	im = Image.open(file_list[i+offset]).convert("L")
	imArray = np.asarray(im)

	testArray[i] = np.reshape(np.asarray(im),[1,H*W])
	testId[i] =int(filtername.split('_')[0])
	# print int(filtername.split('_')[0])


print len(arr)
facial = eigenTools.EigenFacial(arr,file_list[:len(arr)])
scores = facial.train()

# new_score = facial.get_score(arr[5],facial.mean_image)
# print facial.recognize() + ' ' +
# assert(np.max(new_score - scores[0]) < 10**-10)
# assert(facial.recognize(arr[0]) == 0)
numTests = 0
incorrect = 0
for i in range(len(testArray)):
	# new_score = facial.get_score(testFace,facial.mean_image)

	if idArray[facial.recognize(testArray[i])] !=testId[i]:
		print file_list[facial.recognize(testArray[i])]
		incorrect +=1
	print [idArray[facial.recognize(testArray[i])],testId[i] ]
	# print idArray[facial.recognize(testArray[i])]
	# print facial.recognize(testArray[i])
	numTests  += 1
print float(incorrect)/float(numTests)