if __name__ == '__main__':
dataset_location = os.path.join(fpath, '..', '..', 'data')
imscan = ImageScanner()
people = os.listdir(dataset_location)
for person in people:
fullpath = os.path.join(dataset_location, person)
if not os.path.isdir(fullpath):
print "Skipping ", person
continue
print 'Viewing {}'.format(person)
for image_name in os.listdir(fullpath):
img = cv2.imread(os.path.join(fullpath, image_name))
img = cv2.resize(img, (img.shape[1] / 2, img.shape[0] / 2), interpolation=cv2.INTER_CUBIC)
detection = imscan.detect_face(img)
if detection is not None:
x0, y0, x1, y1 = detection
cv2.rectangle(img, (x0, y0), (x1, y1), (0, 255, 0), 2)
cv2.putText(img, "{} at {}".format(person, image_name), (x0, y1 + 25),
cv2.FONT_HERSHEY_SIMPLEX, 0.5, (50, 255, 50))
else:
print image_name, 'no face detected'
cv2.imshow("image", img)
key = cv2.waitKey(0) & 0xff
if key == ord('q'):
sys.exit()
