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
# 将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
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')
# opacity 越小,掩膜效果越深
main(imagePath=img_path, labelPath=mask_path, bgr=(0, 0, 255), opacity=0.5)
没有合适的资源?快使用搜索试试~ 我知道了~
医学图像分割数据集:肺分割数据(包含训练集和测试集)
共2000个文件
png:1999个
py:1个
1.该资源内容由用户上传,如若侵权请联系客服进行举报
2.虚拟产品一经售出概不退款(资源遇到问题,请及时私信上传者)
2.虚拟产品一经售出概不退款(资源遇到问题,请及时私信上传者)
版权申诉
0 下载量 22 浏览量
2024-04-29
15:20:41
上传
评论
收藏 253.01MB 7Z 举报
温馨提示
项目包含:肺分割数据(包含训练集和测试集) 数据集为256*256 分辨率下的肺部分割。分割的前景包含左肺、右肺等等。标签的mask图像为前景为255的二值图像,方便观察 数据集介绍:分为训练集、测试集 训练集:images图片目录+masks模板目录,6849张图片和6849个对应的mask图片 测试集:images图片目录+masks模板目录,1712图片和1712个对应的mask图片 除此之外,包含一个图像分割的可视化脚本,随机提取一张图片,将其原始图片、GT图像、GT在原图蒙板的图像展示,并保存在当前目录下
资源推荐
资源详情
资源评论
收起资源包目录
医学图像分割数据集:肺分割数据(包含训练集和测试集) (2000个子文件)
Normal (5790).png 1KB
Normal (3648).png 1KB
Normal (10989).png 1KB
Normal (11295).png 1KB
Normal (5579).png 1KB
Normal (4833).png 1KB
Normal (11651).png 1KB
Normal (2887).png 1KB
Normal (3673).png 1KB
Normal (10437).png 1KB
Normal (4251).png 1KB
Normal (11918).png 1KB
Normal (3044).png 1KB
Normal (3056).png 1KB
Normal (11248).png 1KB
Normal (11257).png 1KB
Normal (11251).png 1KB
Normal (2005).png 1KB
Normal (11660).png 1KB
Normal (2858).png 1KB
Normal (2717).png 1KB
Normal (2484).png 1KB
Normal (2238).png 1KB
Normal (5558).png 1KB
Normal (3799).png 1KB
Normal (5510).png 1KB
Normal (1986).png 1KB
Normal (2991).png 1KB
Normal (5042).png 1KB
Normal (10947).png 1KB
Normal (5112).png 1KB
Normal (4058).png 1KB
Normal (11156).png 1KB
Normal (5098).png 1KB
Normal (6349).png 1KB
Normal (11009).png 1KB
Normal (5562).png 1KB
Normal (2808).png 1KB
Normal (3102).png 1KB
Normal (10310).png 1KB
Normal (5859).png 1KB
Normal (3677).png 1KB
Normal (10649).png 1KB
Normal (11171).png 1KB
Normal (3419).png 1KB
Normal (10736).png 1KB
Normal (10390).png 1KB
Normal (10903).png 1KB
Normal (5255).png 1KB
Normal (3882).png 1KB
Normal (11343).png 1KB
Normal (3272).png 1KB
Normal (1861).png 1KB
Normal (1672).png 1KB
Normal (10605).png 1KB
Normal (11959).png 1KB
Normal (3899).png 1KB
Normal (1844).png 1KB
Normal (5664).png 1KB
Normal (5097).png 1KB
Normal (6316).png 1KB
Normal (11155).png 1KB
Normal (4033).png 1KB
Normal (10361).png 1KB
Normal (10824).png 1KB
Normal (1740).png 1KB
Normal (3908).png 1KB
Normal (4262).png 1KB
Normal (4794).png 1KB
Normal (11263).png 1KB
Normal (4001).png 1KB
Normal (5622).png 1KB
Normal (11728).png 1KB
Normal (3961).png 1KB
Normal (4706).png 1KB
Normal (11232).png 1KB
Normal (11676).png 1KB
Normal (3545).png 1KB
Normal (1556).png 1KB
Normal (5831).png 1KB
Normal (2123).png 1KB
Normal (2362).png 1KB
Normal (6513).png 1KB
Normal (1600).png 1KB
Normal (1667).png 1KB
Normal (5092).png 1KB
Normal (4870).png 1KB
Normal (1647).png 1KB
Normal (10681).png 1KB
Normal (2781).png 1KB
Normal (11318).png 1KB
Normal (6062).png 1KB
Normal (2989).png 1KB
Normal (3850).png 1KB
Normal (5780).png 1KB
Normal (2096).png 1KB
Normal (6218).png 1KB
Normal (10321).png 1KB
Normal (3281).png 1KB
Normal (11443).png 1KB
共 2000 条
- 1
- 2
- 3
- 4
- 5
- 6
- 20
资源评论
听风吹等浪起
- 粉丝: 1w+
- 资源: 1288
上传资源 快速赚钱
- 我的内容管理 展开
- 我的资源 快来上传第一个资源
- 我的收益 登录查看自己的收益
- 我的积分 登录查看自己的积分
- 我的C币 登录后查看C币余额
- 我的收藏
- 我的下载
- 下载帮助
安全验证
文档复制为VIP权益,开通VIP直接复制
信息提交成功