前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >使用Python-去除视频背景

使用Python-去除视频背景

原创
作者头像
如此这般那便是极好
发布2023-08-08 09:36:22
4940
发布2023-08-08 09:36:22
举报
文章被收录于专栏:未来码迹未来码迹

目前需求:需要将视频的背景去掉,只保留人像,方便后期合成其他视频

大概步骤:

1、先对视频每一帧进行图像分割

2、对每一帧的图像进行抠图

3.对完成抠图的每一帧图像进行合并

使用之前先导入库

cv2:进行图片分割

PIL:操作图像

os:操作系统目录

rembg :对图像进行抠图移除背景

numpy:图片分割时使用

代码语言:Python
复制
import cv2
from PIL import Image
import os
from rembg import remove
import numpy as np

分割视频为图像的代码

代码语言:python
复制
# 读取视频文件
video_path = 'D:\\vidio\\1.mp4'
cap = cv2.VideoCapture(video_path)
# 获取视频帧率和总帧数
fps = cap.get(cv2.CAP_PROP_FPS)
# 创建保存照片的目录
total_frames = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
if not os.path.exists('frames'):
    os.makedirs('frames')
# 分割视频为照片
for i in range(total_frames):
    ret, frame = cap.read()
    if ret:
        # 每秒25帧
        if i % int(fps/25) == 0:
            # 保存照片
            cv2.imwrite(f'frames/frame_{i}.jpg', frame)
    else:
        break

移除图片背景代码

代码语言:python
复制
# 处理照片
for filename in os.listdir('frames'):
    if filename.endswith('.jpg'):
        # 打开照片
        img = Image.open(f'frames/{filename}')
        # 抠图
        img = remove(img)
        # 保存处理后的照片
        img = cv2.cvtColor(np.array(img), cv2.COLOR_RGBA2BGR)
        # 保存处理后的照片
        cv2.imwrite(f'processed_frames/{filename}', img)

合并图片为视频代码

代码语言:python
复制
# 设置输入图像目录和输出视频文件名
input_dir = "D:\\pythonProject1\\processed_frames"
output_file = "output_video.mp4"

# 获取输入目录中的所有图像文件
image_files = [f for f in os.listdir(input_dir) if f.endswith(".jpg")]

# 从第一张图像获取图像的尺寸
img = cv2.imread(os.path.join(input_dir, image_files[0]))
height, width, _ = img.shape

# 创建用于写入视频的VideoWriter对象
video_writer = cv2.VideoWriter(output_file, cv2.VideoWriter_fourcc(*"mp4v"), 30, (width, height))

# 逐个读取图像并将其写入视频
for image_file in image_files:
    img = cv2.imread(os.path.join(input_dir, image_file))
    video_writer.write(img)

所有代码

代码语言:python
复制
import cv2
from PIL import Image
import os
from rembg import remove
import numpy as np
# 读取视频文件
video_path = 'D:\\vidio\\1.mp4'
cap = cv2.VideoCapture(video_path)
# 获取视频帧率和总帧数
fps = cap.get(cv2.CAP_PROP_FPS)
# 创建保存照片的目录
total_frames = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
if not os.path.exists('frames'):
    os.makedirs('frames')
# 分割视频为照片
for i in range(total_frames):
    ret, frame = cap.read()
    if ret:
        # 每秒25帧
        if i % int(fps/25) == 0:
            # 保存照片
            cv2.imwrite(f'frames/frame_{i}.jpg', frame)
    else:
        break

# 处理照片
for filename in os.listdir('frames'):
    if filename.endswith('.jpg'):
        # 打开照片
        img = Image.open(f'frames/{filename}')
        # 抠图
        img = remove(img)
        # 保存处理后的照片
        img = cv2.cvtColor(np.array(img), cv2.COLOR_RGBA2BGR)
        # 保存处理后的照片
        cv2.imwrite(f'processed_frames/{filename}', img)

# 设置输入图像目录和输出视频文件名
input_dir = "D:\\pythonProject1\\processed_frames"
output_file = "output_video.mp4"

# 获取输入目录中的所有图像文件
image_files = [f for f in os.listdir(input_dir) if f.endswith(".jpg")]

# 从第一张图像获取图像的尺寸
img = cv2.imread(os.path.join(input_dir, image_files[0]))
height, width, _ = img.shape

# 创建用于写入视频的VideoWriter对象
video_writer = cv2.VideoWriter(output_file, cv2.VideoWriter_fourcc(*"mp4v"), 30, (width, height))

# 逐个读取图像并将其写入视频
for image_file in image_files:
    img = cv2.imread(os.path.join(input_dir, image_file))
    video_writer.write(img)

# 释放VideoWriter对象并关闭文件
video_writer.release()
cv2.destroyAllWindows()

原创声明:本文系作者授权腾讯云开发者社区发表,未经许可,不得转载。

如有侵权,请联系 cloudcommunity@tencent.com 删除。

原创声明:本文系作者授权腾讯云开发者社区发表,未经许可,不得转载。

如有侵权,请联系 cloudcommunity@tencent.com 删除。

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档