人体分割#

在视频制作中，例如天气预报，我们经常需要从摄像机画面中剪辑出人物图像，并将其叠加到其他画面上。这通常通过 色度键 技术实现，即在一个统一颜色的背景前拍摄人物，然后移除背景。在本实验中，我们将训练一个神经网络模型来剪辑出人物轮廓。

我们将使用来自 Kaggle 的 Segmentation Full Body MADS Dataset。请从 Kaggle 手动下载数据集，并解压到当前目录中。

In [2]:

dataset_path = 'segmentation_full_body_mads_dataset_1192_img'

import os
import matplotlib.pyplot as plt

让我们看看数据集中的图像是什么样子的：

In [3]:

img_path = os.path.join(dataset_path,'images')
mask_path = os.path.join(dataset_path,'masks')

fnames = os.listdir(img_path)

def load_image(img_name):
    img = plt.imread(os.path.join(img_path,img_name))
    mask = plt.imread(os.path.join(mask_path,img_name))
    return img,mask

In [4]:

img, mask = load_image(fnames[5])

fig,ax = plt.subplots(1,2,figsize=(10,5))
ax[0].imshow(img)
ax[1].imshow(mask)
ax[0].axis('off')
ax[1].axis('off')

(-0.5, 511.5, 383.5, -0.5)

In [ ]:

免责声明：
本文档使用AI翻译服务Co-op Translator进行翻译。尽管我们努力确保翻译的准确性，但请注意，自动翻译可能包含错误或不准确之处。原始语言的文档应被视为权威来源。对于关键信息，建议使用专业人工翻译。我们对因使用此翻译而产生的任何误解或误读不承担责任。