MediaPipe是Google开源的计算机视觉处理框架,基于TensorFlow来训练模型。图像分割模块提供人像分割、头发分割、多类分割。本文主要探索如何实现人像分割,当然在人像分割基础上,我们可以做背景替换、背景模糊。
目录
一、配置参数与模型
1、配置参数
2、分割模型
2.1 人像分割模型
2.2 头发分割模型
2.3 多类分割模型
二、工程配置
三、初始化工作
1、初始化人像分割
2、初始化摄像头
四、人像分割
1、运行人像分割
2、绘制人像分割
五、分割效果
一、配置参数与模型
1、配置参数
图像分割的参数包括:运行模式、输出类别掩码、输出置信度掩码、标签语言、结果回调,具体如下表所示:
参数 描述 取值范围 默认值
running_mode
IMAGE:单个图像
VIDEO:视频帧
LIVE_STREAM:实时流
{IMAGE,VIDEO,
LIVE_STREAM}
IMAGE
output_category_mask 输出的类别掩码 Boolean false
output_confidence_mask 输出的置信度掩码 Boolean true
di