视频人脸检测小脚本

用opencv for python 检测人脸的小脚本代码

import cv2

def generate():
	face_cascade = cv2.CascadeClassifier("./cascades/haarcascade_frontalface_alt2.xml")
	eye_cascade = cv2.CascadeClassifier("./cascades/haarcascade_eye.xml")

	camera = cv2.VideoCapture("screencast.avi", )
	count = 0

	while True:
		ret ,frame = camera.read()
	#	gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
	#	cv2.imshow("test",frame)cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
	#	cv2.imwrite("test.jpg",frame)
	#	cv2.waitKey(0)
	#	print(ret)
		if  ret == False:
			break
		gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
	#	cv2.imshow("test",frame)
	#	cv2.imwrite("test.jpg",frame)
	#	cv2.waitKey(0)
	#	gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)

		faces = face_cascade.detectMultiScale(gray, 1.3, 5)

		for (x,y,w,h) in faces:
			img = cv2.rectangle(frame,(x,y),(x+w,y+h),(255,0,0),2)

			f = cv2.resize(gray[y:y+h, x:x+w],(200,200))

			cv2.imwrite("./data/jm1/%s.jpg" %str(count),f)

			count += 1

	#	cv2.imshow("camera", frame)
	#	if cv2.waitKey(1000/12) & 0xff == ord("q"):
	#		break


	camera.release()
	cv2.destroyAllWindows()

if __name__ == "__main__":
	generate()

注释段代码基本是因为开始代码出了点问题debug用的可以不看，最后发现是没有加下面这段代码的锅：

		if  ret == False:
			break

没有跳出循环的结果是最后videocapture（）.read()返回一个False而frame应该是None，空的frame不能被灰度转换代码识别，故而出现以下异常提示：

cv2.error: OpenCV(3.4.2) c:\projects\opencv-python\opencv\modules\imgproc\src\color.hpp:253: error: (-215:Assertion failed) VScn::contains(scn) && VDcn::contains(dcn) && VDepth::contains(depth) in function 'cv::CvtHelper<struct cv::Set<3,4,-1>,struct cv::Set<1,-1,-1>,struct cv::Set<0,2,5>,2>::CvtHelper'

因为是执行时异常而不是出现在编译的时候，所以原因显得并不直观（python只是告诉你内置异常代码），花了我近半小时找问题……故而想做个记录，有出现这种执行时错误就不要纠结是不是拼写啊、缩进啊什么的问题啦，专心看看你的代码逻辑是不是合理吧~~~

另这里使用haarcascade_frontalface_alt2.xml而不是通常介绍的haarcascade_frontalface_default或者haarcascade_frontalface_alt是因为本人用一个很久前的垃圾摄像头截了段视频“screencast.avi”（怎样操纵摄像头截视频代码我有空再写）视频清晰度低得发指，实践发现无论是default或者alt都无法很好检测视频中本人帅气的脸~~只有alt2完美完成此项任务，至于它们之间的区别，请翻看最新版opencv的ducumentation，我有空也会写一些~~