import cv2
import numpy as np
import os
import random
from PIL import Image
import matplotlib.pyplot as plt
# https://blog.csdn.net/qq_44886601/article/details/135764589?spm=1001.2014.3001.5502
# 将mask绘制在原图
def draw_image(im, ms, brg, opacity):
image_mask = im.copy()
contours, _ = cv2.findContours(ms, cv2.RETR_TREE, cv2.CHAIN_APPROX_NONE) # 查找轮廓
image_mask = cv2.drawContours(image_mask, contours, -1, (0, 255, 0), 2) # 绘制边界
image_mask = cv2.fillPoly(image_mask, contours, color=brg) # 填充
img_bgr = cv2.addWeighted(im, opacity, image_mask, 1 - opacity, 0)
return im, ms, img_bgr[:, :, ::-1]
def main(imagePath, labelPath, bgr, opacity):
image = np.array(Image.open(imagePath).convert('RGB'))
label = np.array(Image.open(labelPath).convert('L'))
a, b, c = draw_image(image, label, bgr, opacity)
plt.figure(figsize=(12, 8))
for index, i in enumerate((a, b, c)):
plt.subplot(1, 3, index + 1)
plt.imshow(i)
plt.savefig('./result.png')
# plt.show()
if __name__ == '__main__':
root = './data/train/images'
images_path = [os.path.join(root, i) for i in os.listdir(root)]
r = random.randint(0, len(images_path) - 1)
img_path = images_path[r] # 随机取出一张图片
mask_path = img_path.replace('images', 'masks')
mask_path = mask_path.replace('.jpg', '.png')
# opacity 越小,掩膜效果越深
main(imagePath=img_path, labelPath=mask_path, bgr=(0, 0, 255), opacity=0.2)
没有合适的资源?快使用搜索试试~ 我知道了~
温馨提示
项目包含人脸解析图像数据集,文件以文件夹格式储存,可直接用作图像分割数据集,无需额外处理。 可以用于解析短视频中的人脸,做人体姿态估计。将人脸不同区域灰度值阈值处理的话,还可以用作分割的数据集 数据集总督大小:968MB 数据集介绍:分为训练集、测试集 训练集:images图片目录+masks模板目录,9230张图片和9230个对应的mask图片 测试集:images图片目录+masks模板目录,2388张图片和2388个对应的mask图片 除此之外,包含一个图像分割的可视化脚本,随机提取一张图片,将其原始图片、GT图像、GT在原图蒙板的图像展示,并保存在当前目录下
资源推荐
资源详情
资源评论
收起资源包目录
计算机视觉数据集:人脸解析图像数据集 (2000个子文件)
0991_0005.png 42KB
0602_0006.png 33KB
0701_0014.png 33KB
0701_0012.png 32KB
0649_0007.png 32KB
0675_0003.png 32KB
0701_0005.png 31KB
0150_0005.png 31KB
0148_0006.png 31KB
0423_0008.png 31KB
0539_0010.png 31KB
1065_0006.png 30KB
0771_0013.png 30KB
0906_0011.png 30KB
0249_0005.png 29KB
1142_0013.png 29KB
0603_0008.png 29KB
0330_0011.png 29KB
1142_0010.png 29KB
0882_0014.png 29KB
1013_0008.png 29KB
0701_0007.png 28KB
0603_0006.png 28KB
0453_0009.png 28KB
0249_0007.png 28KB
0153_0013.png 28KB
0790_0011.png 28KB
1142_0006.png 27KB
0811_0007.png 27KB
0940_0008.png 27KB
1195_0001.png 27KB
1325_0008.png 27KB
0274_0012.png 27KB
1325_0011.png 27KB
1325_0012.png 27KB
0974_0012.png 26KB
0274_0002.png 26KB
1325_0016.png 26KB
0771_0003.png 26KB
0899_0001.png 26KB
1058_0001.png 26KB
0151_0009.png 26KB
0649_0012.png 26KB
0685_0006.png 26KB
0811_0002.png 25KB
0685_0005.png 25KB
0881_0009.png 25KB
1082_0003.png 25KB
0974_0014.png 25KB
0212_0009.png 25KB
0528_0011.png 25KB
0903_0003.png 24KB
0264_0001.png 24KB
0888_0007.png 24KB
0212_0007.png 24KB
0389_0006.png 23KB
0474_0002.png 23KB
0939_0003.png 23KB
0389_0010.png 23KB
0953_0001.png 23KB
0806_0005.png 23KB
0644_0003.png 23KB
1431_0010.png 22KB
1075_0003.png 22KB
0888_0008.png 22KB
0953_0009.png 22KB
1173_0004.png 22KB
0881_0007.png 22KB
0463_0005.png 22KB
1431_0011.png 22KB
1578_0001.png 22KB
1155_0009.png 21KB
1155_0003.png 21KB
0228_0009.png 21KB
1578_0012.png 21KB
0644_0005.png 21KB
0338_0007.png 21KB
0228_0008.png 21KB
0317_0002.png 21KB
0924_0006.png 21KB
0200_0007.png 20KB
0311_0008.png 20KB
0620_0014.png 20KB
1049_0012.png 20KB
0906_0005.png 20KB
1107_0010.png 20KB
0903_0010.png 20KB
1107_0011.png 20KB
0978_0001.png 20KB
0772_0005.png 20KB
0259_0009.png 20KB
0259_0006.png 20KB
0319_0007.png 20KB
0393_0005.png 20KB
0916_0010.png 20KB
0978_0011.png 19KB
0978_0013.png 19KB
0126_0001.png 19KB
1049_0010.png 19KB
0542_0010.png 19KB
共 2000 条
- 1
- 2
- 3
- 4
- 5
- 6
- 20
资源评论
听风吹等浪起
- 粉丝: 1w+
- 资源: 1215
上传资源 快速赚钱
- 我的内容管理 展开
- 我的资源 快来上传第一个资源
- 我的收益 登录查看自己的收益
- 我的积分 登录查看自己的积分
- 我的C币 登录后查看C币余额
- 我的收藏
- 我的下载
- 下载帮助
安全验证
文档复制为VIP权益,开通VIP直接复制
信息提交成功