python教程分享python人物视频背景替换实现虚拟空间穿梭

pip install mediapipe  

import cv2  import mediapipe as mp  import numpy as np  mp_drawing = mp.solutions.drawing_utils  mp_selfie_segmentation = mp.solutions.selfie_segmentation  bg_color = (0, 0, 225) # red  cap = cv2.videocapture('111.mp4')  with mp_selfie_segmentation.selfiesegmentation(      model_selection=0) as selfie_segmentation:    bg_image = none    while cap.isopened():      success, image = cap.read()      if not success:        break      image = cv2.cvtcolor(cv2.flip(image, 1), cv2.color_bgr2rgb)      image.flags.writeable = false      results = selfie_segmentation.process(image)      image.flags.writeable = true      image = cv2.cvtcolor(image, cv2.color_rgb2bgr)      cOndition= np.stack(        (results.segmentation_mask,) * 3, axis=-1) > 0.8      if bg_image is none:        bg_image = np.zeros(image.shape, dtype=np.uint8)        bg_image[:] = bg_color        print(bg_image.shape)      output_image = np.where(condition, image, bg_image)      cv2.imshow('mediapipe selfie segmentation', output_image)      if cv2.waitkey(5) & 0xff == 27:        break  cap.release()  

bg_image = cv2.imread("bbb.jpg")