class Training:
def __init__(self, nose_cascade = None, face_cascade = None, eye_cascade = None, sample = None):
#if nose_cascade or face_cascade is None: warnings.warn('Some of the cascades are still None')
self.nose = Image_X(nose_cascade)
self.face = Image_X(face_cascade)
self.eye = Image_X(eye_cascade)
val = self.GetDimensions(sample)
self.dimensions = val['dimensions']
self.dist = val['distance']
def alignImage(self, PATH, image):
try:
for filename, image in self.eye.alignEyes(image).iteritems():
self.save(PATH, **{filename : image})
return image
except:
try:
self.IdError(image)
except:
pass
def GetDimensions(self, filename):
img = cv2.imread(filename)
val = self.GetCentre(filename)
centre = val['centre']
#return (up, down, left, right)
try:
return {'dimensions' : (img.shape[1] - centre[1], centre[1], centre[0], img.shape[0] - centre[0]),
'distance' : val['dist']}
except:
return None
def GetCentre(self, filename):
img = cv2.imread(filename)
eye = self.eye.ShiTomasiCornerDetect(filename)
try:
return {'centre' : ((eye[0][0]+eye[1][0])/2., eye[0][1]), 'dist' : abs(eye[0][0]-eye[1][0])}
except:
return None
def IdError(self, image):
'''Delete the image'''
try:
os.remove(image)
except:
pass
def Resize(self, filename):
try:
dist = self.GetCentre(filename)['dist']
except:
self.IdError(filename)
return
scaleFactor = self.dist/float(dist)
return cv2.resize(cv2.imread(filename), (0,0), fx = scaleFactor, fy = scaleFactor)
def faceIdentify(self, image):
for filename, list_of_parameters in self.face.identify(image).iteritems():
if len(list_of_parameters) is not 1:
#self.IdError(image)
return None
for (x, y, w, h) in list_of_parameters:
return (w, h)
def alignFace(self, image):
'''
x -> left
y -> down
w -> right + left
h -> up + down
'''
#image = self.alignedImage(image)
try:
centre = self.GetCentre(image)['centre']
dim = self.dimensions
except:
self.IdError(image)
try:
return (centre[0] - dim[2],
centre[1] - dim[1],
dim[2] + dim[3],
dim[0] + dim[1])
except:
return None
def crop(self, image):
x = self.alignFace(image)
try:
return self.face.getImg(image)[x[1]:x[1]+x[3] , x[0]:x[0]+x[2]]
except:
self.IdError(image)
return None
def save(self, PATH, **kwargs):
os.chdir(PATH)
for filename, image in kwargs.iteritems():
if image is not None:
print "Saved {0} image".format(filename)
cv2.imwrite(filename , image)
else:
print "Not saved {0}".format(filename)
return