def testDetectLandmarks(self):
input_file = os.path.abspath(
'./resources/landmarks/Fayssal_Mekdad_0002.jpg')
fidu_file = input_file.replace('.jpg', '.cfidu')
if os.path.exists(fidu_file):
os.remove(fidu_file)
detect_landmarks(fname=input_file)
fidu_score, yaw_angle, fidu_points = read_fidu(fidu_file)
self.assertEqual(fidu_score, 252)
self.assertEqual(yaw_angle, 0)
img = cv2.imread(input_file)
draw_fidu(img, fidu_points)
cv2.imshow('landmarks', img)
cv2.waitKey()
if os.path.exists(fidu_file):
os.remove(fidu_file)
detect_landmarks(fname=input_file, max_size=160 * 160)
fidu_score, yaw_angle, fidu_points = read_fidu(fidu_file)
print "new fidu score is", fidu_score
self.assertGreater(fidu_score, 200)
self.assertEqual(yaw_angle, 0)
img = cv2.imread(input_file)
draw_fidu(img, fidu_points)
cv2.imshow('landmarks, detected on reduced image', img)
cv2.waitKey()
def testDetectLandmarks(self):
input_file = os.path.abspath("./resources/landmarks/Fayssal_Mekdad_0002.jpg")
fidu_file = input_file.replace(".jpg", ".cfidu")
if os.path.exists(fidu_file):
os.remove(fidu_file)
detect_landmarks(fname=input_file)
fidu_score, yaw_angle, fidu_points = read_fidu(fidu_file)
self.assertEqual(fidu_score, 252)
self.assertEqual(yaw_angle, 0)
img = cv2.imread(input_file)
draw_fidu(img, fidu_points)
cv2.imshow("landmarks", img)
cv2.waitKey()
if os.path.exists(fidu_file):
os.remove(fidu_file)
detect_landmarks(fname=input_file, max_size=160 * 160)
fidu_score, yaw_angle, fidu_points = read_fidu(fidu_file)
print "new fidu score is", fidu_score
self.assertGreater(fidu_score, 200)
self.assertEqual(yaw_angle, 0)
img = cv2.imread(input_file)
draw_fidu(img, fidu_points)
cv2.imshow("landmarks, detected on reduced image", img)
cv2.waitKey()
def align_faces(self,
input_images,
output_path,
fidu_max_size=None,
fidu_min_size=None,
is_align=True,
is_draw_fidu=False,
delete_no_fidu=False):
'''
input_images - can be either a folder (all *.jpgs in it) or a list of filenames
, fidu_max_size = None, fidu_min_size = None):
'''
if type(input_images) == type(''):
input_images = glob.glob(os.path.join(input_images, '*.jpg'))
for input_image in input_images:
detect_landmarks(fname=os.path.abspath(input_image),
max_size=fidu_max_size,
min_size=fidu_min_size,
fidu_exec_dir=self.fidu_exec_dir)
fidu_file = input_image.rsplit('.', 1)[0] + '.cfidu'
fidu_score, yaw_angle, fidu_points = read_fidu(fidu_file)
if not (fidu_score is not None and yaw_angle in self.valid_angles):
# skip face
if delete_no_fidu:
try:
os.remove(fidu_file)
os.remove(input_image)
except:
continue
continue
if is_align:
# save the aligned image
sub_img = cv2.imread(input_image)
_, base_fname = os.path.split(input_image)
aligned_img, R = self.aligner.align(sub_img, fidu_points)
aligned_img_file = os.path.join(
output_path,
base_fname.rsplit('.', 1)[0] + '.aligned.png')
cv2.imwrite(aligned_img_file, aligned_img)
# save a copy of the aligned image, with the landmarks drawn
if is_draw_fidu:
aligned_img_file = os.path.join(
output_path,
base_fname.rsplit('.', 1)[0] +
'.aligned.withpoints.png')
fidu_points_in_aligned = unwarp_fidu(
orig_fidu_points=fidu_points, unwarp_mat=R)
draw_fidu(aligned_img,
fidu_points_in_aligned,
radius=9,
color=(255, 0, 0),
thickness=3)
cv2.imwrite(aligned_img_file, aligned_img)
def align_faces(self, input_images, output_path, fidu_max_size = None, fidu_min_size = None, is_align = True, is_draw_fidu = False, delete_no_fidu = False):
'''
input_images - can be either a folder (all *.jpgs in it) or a list of filenames
, fidu_max_size = None, fidu_min_size = None):
'''
if type(input_images) == type(''):
input_images = glob.glob(os.path.join(input_images, '*.jpg'))
for input_image in input_images:
detect_landmarks(fname = os.path.abspath(input_image),
max_size = fidu_max_size,
min_size = fidu_min_size,
fidu_exec_dir = self.fidu_exec_dir)
fidu_file = input_image.rsplit('.',1)[0] + '.cfidu'
fidu_score, yaw_angle, fidu_points = read_fidu(fidu_file)
if not (fidu_score is not None and yaw_angle in self.valid_angles):
# skip face
if delete_no_fidu:
try:
os.remove(fidu_file)
os.remove(input_image)
except:
continue
continue
if is_align:
# save the aligned image
sub_img = cv2.imread(input_image)
_, base_fname = os.path.split(input_image)
aligned_img, R = self.aligner.align(sub_img, fidu_points)
aligned_img_file = os.path.join(output_path, base_fname.rsplit('.',1)[0] + '.aligned.png')
cv2.imwrite(aligned_img_file, aligned_img)
# save a copy of the aligned image, with the landmarks drawn
if is_draw_fidu:
aligned_img_file = os.path.join(output_path, base_fname.rsplit('.',1)[0] + '.aligned.withpoints.png')
fidu_points_in_aligned = unwarp_fidu(orig_fidu_points = fidu_points, unwarp_mat = R)
draw_fidu(aligned_img, fidu_points_in_aligned, radius = 9, color = (255,0,0), thickness = 3)
cv2.imwrite(aligned_img_file, aligned_img)
def testAffineAlign(self):
aligner = AffineAligner(fidu_model_file = '../resources/model_ang_0.txt')
img_files = ['./resources/affine_align/Meryl_Streep_0013.jpg', './resources/affine_align/Fayssal_Mekdad_0002.jpg']
for img_file in img_files:
img_file = os.path.abspath(img_file)
detect_landmarks(fname = img_file)
fidu_file = img_file.replace('.jpg','.cfidu')
img = cv2.imread(img_file)
score, yaw_angle, fidu_points = read_fidu(fidu_file)
aligned_img, R = aligner.align(img, fidu_points)
aligned_img_cpy = aligned_img.copy()
draw_fidu(img, fidu_points)
fidu_points_in_aligned = unwarp_fidu(orig_fidu_points = fidu_points, unwarp_mat = R)
draw_fidu(aligned_img_cpy, fidu_points_in_aligned, radius = 9, color = (255,0,0), thickness = 3)
cv2.imshow('padded_face with landmarks', img)
cv2.imshow('aligned_face', cv2.resize(aligned_img, (320,320), interpolation = cv2.INTER_CUBIC))
cv2.imshow('aligned_face with landmarks', cv2.resize(aligned_img_cpy, (320,320), interpolation = cv2.INTER_CUBIC))
cv2.waitKey()
def testAffineAlign(self):
aligner = AffineAligner(fidu_model_file='../resources/model_ang_0.txt')
img_files = [
'./resources/affine_align/Meryl_Streep_0013.jpg',
'./resources/affine_align/Fayssal_Mekdad_0002.jpg'
]
for img_file in img_files:
img_file = os.path.abspath(img_file)
detect_landmarks(fname=img_file)
fidu_file = img_file.replace('.jpg', '.cfidu')
img = cv2.imread(img_file)
score, yaw_angle, fidu_points = read_fidu(fidu_file)
aligned_img, R = aligner.align(img, fidu_points)
aligned_img_cpy = aligned_img.copy()
draw_fidu(img, fidu_points)
fidu_points_in_aligned = unwarp_fidu(orig_fidu_points=fidu_points,
unwarp_mat=R)
draw_fidu(aligned_img_cpy,
fidu_points_in_aligned,
radius=9,
color=(255, 0, 0),
thickness=3)
cv2.imshow('padded_face with landmarks', img)
cv2.imshow(
'aligned_face',
cv2.resize(aligned_img, (320, 320),
interpolation=cv2.INTER_CUBIC))
cv2.imshow(
'aligned_face with landmarks',
cv2.resize(aligned_img_cpy, (320, 320),
interpolation=cv2.INTER_CUBIC))
cv2.waitKey()