FaceDetect.py

FaceDetect.py

from PIL import Image, ImageDraw
import photos
from objc_util import *
CIImage, CIDetector = map(ObjCClass, ['CIImage', 'CIDetector'])

#imagebuffer = photos.capture_image()
imagebuffer = photos.pick_image()
imagebuffer.save('.temp.jpg')
data = NSData.dataWithContentsOfFile_('.temp.jpg')
ci_img = CIImage.imageWithData_(data)

opt = {'CIDetectorAccuracy': 'CIDetectorAccuracyHigh'}
d = CIDetector.detectorOfType_context_options_('CIDetectorTypeFace', None, opt)
face = d.featuresInImage_(ci_img)
#print(face)
if face.count() > 0:
	f = face.firstObject()
	b = f.bounds()
	w=b.size.width
	h=b.size.height

	drawbuffer = ImageDraw.Draw(imagebuffer)
	drawbuffer.rectangle((b.origin.x,imagebuffer.size[1]-(b.origin.y+h), b.origin.x+w,imagebuffer.size[1]-b.origin.y), outline='white')
	del drawbuffer
else:
	print('No face detected.')
	
imagebuffer.show()