图像处理 使用opencv给视频添加水印

课后作业练习
1）读取视频；
2）设置每秒读取多少张图片，对图片加水印
3）将处理过的图片合成视频

# -*- coding:utf-8 -*-
import cv2

cap = cv2.VideoCapture('zy.mp4')
cap.isOpened()
frame_count = 1
success = True
i = 0
frame = cap.read()
# 设置每秒读取多少张图片
fps = 25
imgInfo = frame[1].shape
print(imgInfo)
# 获取图片宽高度信息
size = (imgInfo[1], imgInfo[0])
fourcc = cv2.VideoWriter_fourcc(*"mp4v")
videoWrite = cv2.VideoWriter('output.mp4', fourcc , fps, size)
while success:
    success, frame = cap.read()
    print('Read a new frame: ', success)
    if not success:
        videoWrite.release()
        print("finish")
        break
    else:

        logo = cv2.imread('img/logo.jpg')
        w1, h1, c1 = frame.shape
        w2, h2, c2 = logo.shape
        roi = frame[0:w2, h1-h2:h1]
        # 灰度化
        gray_logo = cv2.cvtColor(logo, cv2.COLOR_BGR2GRAY)
        # 黑化