1. 程式人生 > >Python和OpenCV簡單的人臉檢測程式

Python和OpenCV簡單的人臉檢測程式

    最近想要搞人臉識別的東西,所以先弄了個簡單的人臉檢測的程式,網上找的許多都無法使用,不知道是不是Opencv版本的關係,或者是和Python結合後產生的問題,感覺很多API都是錯的,這裡用的是opencv2.3.1和python2.7。Python需要安裝numpy,和PIL(不裝應該也行,我這裡使用到了),廢話不多說上程式碼。

參考了:http://creatingwithcode.com/howto/face-detection-in-static-images-with-python/

from PIL import Image, ImageDraw
from math import sqrt
import cv
import os
import sys

def detect_object(image):

    grayscale = cv.CreateImage((image.width, image.height), 8, 1)
    cv.CvtColor(image, grayscale, cv.CV_BGR2GRAY)

    cascade = cv.Load("D:\\opencv\\data\\haarcascades\\haarcascade_frontalface_default.xml")
    rect = cv.HaarDetectObjects(grayscale, cascade, cv.CreateMemStorage(), 1.1, 2,
        cv.CV_HAAR_DO_CANNY_PRUNING, (20,20))
    
    result = []
    for r in rect:
        result.append((r[0][0], r[0][1], r[0][0]+r[0][2], r[0][1]+r[0][3]))

    return result

def process(infile, outfile):
    image = cv.LoadImage(infile);
    if image:
        faces = detect_object(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')
程式碼不難懂,不理解的地方去搜下相關的API即可。

結果如下: