opencv绿幕换背景

import cv2

# 读取视频文件
cap = cv2.VideoCapture('19.mp4')
background_img = cv2.imread('3.jpg')

# 调整背景图像的大小与视频帧相同
background_img = cv2.resize(background_img, (int(cap.get(cv2.CAP_PROP_FRAME_WIDTH)), int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))))

while cap.isOpened():
    ret, frame = cap.read()
    if not ret:
        break

    # 转换绿幕图像为HSV颜色空间
    hsv = cv2.cvtColor(frame, cv2.COLOR_BGR2HSV)
    # 定义绿色的HSV范围
    lower_green = np.array([35, 50, 50])
    upper_green = np.array([85, 255, 255])
    # 根据阈值将绿幕区域转换为掩码
    mask = cv2.inRange(hsv, lower_green, upper_green)
    # 反转掩码
    mask_inv = cv2.bitwise_not(mask)
    # 从绿幕图像中提取前景
    fg = cv2.bitwise_and(frame, frame, mask=mask_inv)
    # 从背景图像中提取背景
    bg = cv2.bitwise_and(background_img, background_img, mask=mask)
    # 将前景和背景合并
    result = cv2.add(fg, bg)
    cv2.imshow('img',result)
    k = cv2.waitKey(10) 
    if k == 27:
        break

cap.release()
cv2.destroyAllWindows()