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)
没有合适的资源?快使用搜索试试~ 我知道了~
温馨提示
项目包含海洋鱼类条纹红鲻鱼图像分割数据集(二值图像分割任务),文件以文件夹格式储存,可直接用作图像分割数据集,无需额外处理。 数据集总大小:314MB 数据集介绍:分为训练集、测试集 训练集:images图片目录+masks模板目录,800张图片和800个对应的mask图片 测试集:images图片目录+masks模板目录,200张图片和200个对应的mask图片 除此之外,包含一个图像分割的可视化脚本,随机提取一张图片,将其原始图片、GT图像、GT在原图蒙板的图像展示,并保存在当前目录下
资源推荐
资源详情
资源评论
收起资源包目录
深度学习之图像分割数据集:海洋鱼类条纹红鲻鱼图像分割数据集(二值图像分割任务) (2000个子文件)
00094.png 391KB
00454.png 390KB
00453.png 389KB
00054.png 386KB
00098.png 386KB
00114.png 385KB
00458.png 385KB
00093.png 384KB
00457.png 383KB
00053.png 383KB
00614.png 383KB
00074.png 382KB
00058.png 382KB
00514.png 381KB
00513.png 381KB
00118.png 381KB
00113.png 380KB
00613.png 380KB
00014.png 380KB
00618.png 379KB
00078.png 377KB
00097.png 377KB
00073.png 377KB
00518.png 377KB
00517.png 377KB
00086.png 376KB
00057.png 375KB
00018.png 375KB
00013.png 375KB
00617.png 375KB
00446.png 375KB
00674.png 374KB
00117.png 373KB
00494.png 373KB
00046.png 372KB
00533.png 372KB
00445.png 372KB
00106.png 371KB
00534.png 371KB
00673.png 370KB
00414.png 370KB
00498.png 370KB
00678.png 370KB
00606.png 370KB
00434.png 369KB
00077.png 369KB
00493.png 369KB
00066.png 368KB
00017.png 367KB
00433.png 367KB
00505.png 367KB
00418.png 367KB
00537.png 366KB
00091.png 366KB
00506.png 366KB
00413.png 366KB
00438.png 366KB
00006.png 365KB
00677.png 365KB
00214.png 364KB
00085.png 364KB
00497.png 364KB
00538.png 364KB
00874.png 364KB
00605.png 362KB
00045.png 362KB
00486.png 362KB
00437.png 362KB
00051.png 362KB
00354.png 362KB
00594.png 362KB
00417.png 361KB
00334.png 361KB
00448.png 361KB
00105.png 361KB
00254.png 361KB
00666.png 361KB
00099.png 361KB
00218.png 360KB
00873.png 360KB
00234.png 360KB
00174.png 360KB
00274.png 359KB
00213.png 359KB
00878.png 359KB
00394.png 359KB
00194.png 359KB
00608.png 359KB
00598.png 359KB
00374.png 359KB
00406.png 359KB
00111.png 359KB
00294.png 359KB
00447.png 359KB
00071.png 358KB
00154.png 358KB
00092.png 358KB
00734.png 358KB
00358.png 358KB
00426.png 357KB
共 2000 条
- 1
- 2
- 3
- 4
- 5
- 6
- 20
资源评论
Ai医学图像分割
- 粉丝: 2w+
- 资源: 2298
上传资源 快速赚钱
- 我的内容管理 展开
- 我的资源 快来上传第一个资源
- 我的收益 登录查看自己的收益
- 我的积分 登录查看自己的积分
- 我的C币 登录后查看C币余额
- 我的收藏
- 我的下载
- 下载帮助
最新资源
资源上传下载、课程学习等过程中有任何疑问或建议,欢迎提出宝贵意见哦~我们会及时处理!
点击此处反馈
安全验证
文档复制为VIP权益,开通VIP直接复制
信息提交成功