人臉識別初探(Python)

1.利用opencv庫

sudo apt-get install libopencv-*
sudo apt-get install python-opencv
sudo apt-get install python-numpy

2 .Python實現

import os
import os
from PIL import Image,ImageDraw
import cv

def detect_object(image):
    grayscale = cv.CreateImage((image.width,image.height),8,1)#創建空的灰度值圖片
    cv.CvtColor(image,grayscale,cv.CV_BGR2GRAY)
    cascade=cv.Load("/usr/share/opencv/haarcascades/haarcascade_frontalface_alt_tree.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):
    image = cv.LoadImage(infile)
    if image:
       faces = detect_object(image)
    im = Image.open(infile)
    path = os.path.abspath(infile)
    save_path = os.path.splitext(path)[0]+"_face"
    try:
        os.mkdir(save_path)
    except:
        pass
    if faces:
        draw = ImageDraw.Draw(im)
        count=0
        for f in faces:
             count+=1
             draw.rectangle(f,outline=(255,0,0))
             a=im.crop(f)
             file_name=os.path.join(save_path,str(count)+".jpg")
             a.save(file_name)
        drow_save_path = os.path.join(save_path,"out.jpg")
        im.save(drow_save_path,"JPEG",quality=80)
    else:
        print "Error: cannot detect faces on %s" % infile
if __name__ == "__main__":
     process("test3.jpg")
  1. 效果對比
    這裏寫圖片描述
    這裏寫圖片描述
    4.資料
http://www.cnblogs.com/ma6174/archive/2013/03/31/2991315.html
http://www.iteye.com/topic/463668
http://blog.sina.com.cn/s/blog_5562b0440102uw7g.html
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章