python

关注公众号 jb51net

关闭
首页 > 脚本专栏 > python > opencv背景替换加载

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

作者:*

本文主要介绍了opencv背景替换加载的实现示例,文中通过示例代码介绍的非常详细,具有一定的参考价值,感兴趣的小伙伴们可以参考一下

看到有个例子,使用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()

目前有个缺陷就是,录屏和摄像头都在一起,比较难看,后面想利用一个独立摄像头,和独立的视频来源,这样的效果会更好,当然还会有其它更好的应用场景。

到此这篇关于opencv背景替换加载的实现示例的文章就介绍到这了,更多相关opencv背景替换加载内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!

您可能感兴趣的文章:
阅读全文