python

#!/usr/bin/python

# -*- coding: UTF-8 -*-

  

# face_detect.py

  

# Face Detection using OpenCV. Based on sample code from:

# http://python.pastebin.com/m76db1d6b

  

# Usage: python face_detect.py 

  

import sys, os

from opencv.cv import *

from opencv.highgui import *

from PIL import Image, ImageDraw

from math import sqrt

  

def detectObjects(image):

    """Converts an image to grayscale and prints the locations of any faces found"""

    grayscale = cvCreateImage(cvSize(image.width, image.height), 8, 1)

    cvCvtColor(image, grayscale, CV_BGR2GRAY)

  

    storage = cvCreateMemStorage(0)

    cvClearMemStorage(storage)

    cvEqualizeHist(grayscale, grayscale)

  

    cascade = cvLoadHaarClassifierCascade(

        \'/usr/share/opencv/haarcascades/haarcascade_frontalface_default.xml\',

        cvSize(1,1))

    faces = cvHaarDetectObjects(grayscale, cascade, storage, 1.1, 2,

        CV_HAAR_DO_CANNY_PRUNING, cvSize(20,20))

  

    result = []

    for f in faces:

        result.append((f.x, f.y, f.x+f.width, f.y+f.height))

  

    return result

  

def grayscale(r, g, b):

    return int(r * .3 + g * .59 + b * .11)

  

def process(infile, outfile):

  

    image = cvLoadImage(infile);

    if image:

        faces = detectObjects(image)

  

    im = Image.open(infile)

  

    if faces:

        draw = ImageDraw.Draw(im)

        for f in faces:

            draw.rectangle(f, outline=(255, 0, 255))

  

        im.save(outfile, "JPEG", quality=100)

    else:

        print "Error: cannot detect faces on %s" % infile

  

if __name__ == "__main__":

    process(\'input.jpg\', \'output.jpg\')