Python和OpenCV簡單的人臉檢測程式
阿新 • • 發佈:2019-02-06
最近想要搞人臉識別的東西,所以先弄了個簡單的人臉檢測的程式,網上找的許多都無法使用,不知道是不是Opencv版本的關係,或者是和Python結合後產生的問題,感覺很多API都是錯的,這裡用的是opencv2.3.1和python2.7。Python需要安裝numpy,和PIL(不裝應該也行,我這裡使用到了),廢話不多說上程式碼。
參考了:http://creatingwithcode.com/howto/face-detection-in-static-images-with-python/
程式碼不難懂,不理解的地方去搜下相關的API即可。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')
結果如下: