opencv学习笔记（3）——视频的处理

读取电脑文件中的视频的方法：

reba1 = cv2.VideoCapture('C:\\Users\\shm\\Desktop\\python_data\\video_data\\reba1.mp4')

显示第一帧

flag,frame = reba1.read()#返回一帧的图片和读取下一帧的结果：T/F
print(flag,frame.shape)
#输出结果：True (960, 544, 3)


cv2.imshow('reba',frame)
cv2.waitKey(0)
cv2.destroyAllWindows()

播放视频

while reba1.isOpened():
    flag,frame = reba1.read()    
    if flag == False:
        break
    cv2.imshow('reba',frame)

    if ord('q') == cv2.waitKey(29):
        break
cv2.destroyAllWindows()

获取视频每秒帧数

reba1.get(propId = cv2.CAP_PROP_FPS)#获取每秒帧数

CAP_PROP_FRAME_COUNT获取总帧数
CAP_PROP_HEIGHT每一帧的高度，WIDTH宽度
FOURCC编码格式

人脸识别：用for循环对每一帧运行一遍呗

#人脸识别最好转换成黑白的再处理，速度更快
detector = cv2.CascadeClassifier('C:\\Users\\shm\\Desktop\\python_code\\opencv_classifier\\haarcascade_frontalface_default.xml')#分类器
while reba1.isOpened():
    flag,frame = reba1.read()
    
    gray = cv2.cvtColor(frame, code = cv2.COLOR_BGR2GRAY)

    face_zone = detector.detectMultiScale(gray,scaleFactor = 1.2,minNeighbors = 10)
    for x,y,w,h in face_zone:
        cv2.rectangle(frame,pt1 = (x,y),pt2 = (x+w,y+h),color = [0,0,255],thickness = 2)#画一个正方形框框
    if flag == False:
        break
    cv2.imshow('reba',frame)
    if ord('q') == cv2.waitKey(29):
        break
cv2.destroyAllWindows()

调取本地摄像头,把上面的路径改成0就行

capture = cv2.VideoCapture(0)
detector = cv2.CascadeClassifier('C:\\Users\\shm\\Desktop\\python_code\\opencv_classifier\\haarcascade_frontalface_default.xml')#分类器
while capture.isOpened():
    flag,frame = capture.read()
    
    gray = cv2.cvtColor(frame, code = 6)

    face_zone = detector.detectMultiScale(gray,scaleFactor = 1.5,minNeighbors = 10)
    for x,y,w,h in face_zone:
        cv2.rectangle(frame,pt1 = (x,y),pt2 = (x+w,y+h),color = [0,0,255],thickness = 2)#画一个正方形框框
    if flag == False:
        break
    cv2.imshow('reba',frame)
    if ord('q') == cv2.waitKey(0):
        break
cv2.destroyAllWindows()
#重复上面提取人脸代码

保存视频

#保存一下上面热巴提取人脸之后的视频
w = int(reba1.get(cv2.CAP_PROP_FRAME_WIDTH))#获取视频帧宽高
h = int(reba1.get(cv2.CAP_PROP_FRAME_HEIGHT))

video_writer = cv2.VideoWriter('C:\\Users\\shm\\Desktop\\python_data\\video_data\\reba2.avi',cv2.VideoWriter_fourcc('M','P','4','2'),24,(w,h))#四个参数，文件名、视频编码、fps、视频帧尺寸
#视频编码：cv2.VideoWriter_fourcc('M','P','4','2')为常用格式记住就行
#注意得保存成avi文件，如果是mp4无法储存

#把热巴提取过人脸的视频存起来
detector = cv2.CascadeClassifier('C:\\Users\\shm\\Desktop\\python_code\\opencv_classifier\\haarcascade_frontalface_default.xml')#分类器
while reba1.isOpened():
    
    flag,frame = reba1.read()
    if flag == False:
        break
    gray = cv2.cvtColor(frame, code = 6)

    face_zone = detector.detectMultiScale(gray,scaleFactor = 1.5,minNeighbors = 10)
    for x,y,w,h in face_zone:
        cv2.rectangle(frame,pt1 = (x,y),pt2 = (x+w,y+h),color = [0,0,255],thickness = 2)#画一个正方形框框
    video_writer.write(frame)
    if flag == False:
        break
    cv2.imshow('reba',frame)
    if ord('q') == cv2.waitKey(29):
        break
cv2.destroyAllWindows()
reba1.release()
video_writer.release()