Google出品Mediapipe人像分割，可任意更换图片与视频背景

最新推荐文章于 2024-10-09 14:46:48 发布

人工智能研究所

最新推荐文章于 2024-10-09 14:46:48 发布

阅读量1k

点赞数 1

分类专栏： mediapipe对象检测算法文章标签： opencv 计算机视觉人工智能 mediapipe

本文链接：https://blog.csdn.net/weixin_44782294/article/details/131621315

版权

mediapipe对象检测算法专栏收录该内容

23 篇文章 16 订阅 ¥49.90 ¥99.00

订阅专栏

超级会员免费看

本文介绍了Google的Mediapipe人像分割解决方案，用于实时视频聊天中的人像背景替换。模型基于MobileNetV3，提供通用和景观两种模型。通过代码示例展示了如何使用Mediapipe进行图片和视频的人像分割，以及如何实现背景替换功能。

摘要由CSDN通过智能技术生成

在一些视频聊天软件中，我们可以看到很多人的背景特别的漂亮，甚至我们都怀疑是不是真实地来到了某个地方，本期介绍一下Mediapipe人像分割（RVM人像分割）

MediaPipe Selfie Segmentation 对场景中的突出人物进行分割。它可以在智能手机和笔记本电脑上实时运行。

模型

在这个解决方案中，MediaPipe 提供了两种模型：一般模型和景观模型。两种模型都基于MobileNetV3，并进行了修改以提高效率。通用模型在 256x256x3 (HWC) 张量上运行，并输出代表分割掩码的 256x256x1 张量。景观模型类似于通用模型，但在 144x256x3 (HWC) 张量上运行。它比一般模型具有更少的 FLOP，因此运行速度更快。请注意，再将输入图像输入 ML 模型之前，MediaPipe Selfie Segmentation 会自动将输入图像的大小调整为所需的张量维度。

代码实现

import cv2
import mediapipe as mp
import numpy as np
mp_drawing = mp.solutions.drawing_utils
mp_selfie_segmentation = mp.solutions.selfie_segmentation
# 图片人物抠图:
IMAGE_FILES = []
BG_COLOR = (0, 255, 0) # 背景颜色也可以使用其他的照片，要求与原