opencv背景替换加载的实现示例

# -*- coding: UTF-8 -*-
import cv2
import mediapipe
from PIL import ImageGrab
import numpy as np
import cvzone
from cvzone.SelfiSegmentationModule import SelfiSegmentation

#######################
#输入图像参数
cap = cv2.VideoCapture(0)
cap.set(3, 640)
cap.set(4, 480)

segmentor = SelfiSegmentation()
#######################
#定义的变量

######################
#获取背景视频
def get_bg_video():
    img = ImageGrab.grab(bbox=(100, 100, 1280, 960)) #bbox specifies specific region (bbox= x,y,width,height)
    img_np = np.array(img)
    img_np = cv2.resize(img_np, (640, 480)) #保障图像信息的尺寸一致

    return img_np

if __name__ == "__main__":
    while True:
        is_succ, img = cap.read() #获取摄像头图像
        video = get_bg_video()
        img_out = segmentor.removeBG(img, video, threshold=0.9) #去除背景
        cv2.imshow('replace_bg', img_out)

        key = cv2.waitKey(1)
        if key == ord('q'):
            break
cap.release()
cv2.destroyAllWindows()