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')
mask_path = mask_path.replace('.jpeg', '.png')
# opacity 越小,掩膜效果越深
main(imagePath=img_path, labelPath=mask_path, bgr=(255, 0, 0), opacity=0.2)
没有合适的资源?快使用搜索试试~ 我知道了~
图像分割数据集:风景图像中的水体分割(2分割,包含训练集和测试集)
共2000个文件
png:1322个
jpeg:677个
py:1个
1.该资源内容由用户上传,如若侵权请联系客服进行举报
2.虚拟产品一经售出概不退款(资源遇到问题,请及时私信上传者)
2.虚拟产品一经售出概不退款(资源遇到问题,请及时私信上传者)
版权申诉
0 下载量 6 浏览量
2024-04-10
15:04:25
上传
评论
收藏 115.45MB ZIP 举报
温馨提示
项目包含:风景图像中的水体分割(2分割,包含训练集和测试集) 数据集为风景图像下的水体分割,包含湖泊啊、大海啊、小溪等等类型的水体分割,为二值图像分割任务,前景像素点为255。水体区域占图像很大比例,前景区域丰富,可以用于分割数据 数据集总大小:115 MB 数据集介绍:分为训练集、测试集 训练集:images图片目录+masks模板目录,1322张图片和132个对应的mask图片 测试集:images图片目录+masks模板目录,566张图片和566个对应的mask图片 除此之外,包含一个图像分割的可视化脚本,随机提取一张图片,将其原始图片、GT图像、GT在原图蒙板的图像展示,并保存在当前目录下
资源推荐
资源详情
资源评论
收起资源包目录
图像分割数据集:风景图像中的水体分割(2分割,包含训练集和测试集) (2000个子文件)
ADE_train_00016658.jpeg 649KB
ADE_train_00004970.jpeg 435KB
ADE_train_00010793.jpeg 379KB
ADE_train_00015900.jpeg 304KB
ADE_train_00007913.jpeg 300KB
ADE_train_00015956.jpeg 298KB
ADE_train_00010885.jpeg 288KB
ADE_train_00013677.jpeg 283KB
ADE_train_00019759.jpeg 279KB
ADE_train_00019874.jpeg 276KB
ADE_train_00008684.jpeg 272KB
ADE_train_00011698.jpeg 268KB
ADE_train_00005276.jpeg 265KB
ADE_train_00004914.jpeg 258KB
ADE_train_00017740.jpeg 258KB
ADE_train_00019010.jpeg 257KB
ADE_train_00004363.jpeg 254KB
ADE_train_00016666.jpeg 251KB
ADE_train_00003153.jpeg 250KB
ADE_train_00007663.jpeg 244KB
ADE_train_00009944.jpeg 243KB
ADE_train_00013620.jpeg 240KB
ADE_train_00008669.jpeg 240KB
ADE_train_00011579.jpeg 229KB
ADE_train_00010800.jpeg 222KB
ADE_train_00005006.jpeg 220KB
ADE_val_00000275.jpeg 218KB
ADE_train_00010010.jpeg 215KB
ADE_train_00014918.jpeg 215KB
ADE_train_00013577.jpeg 214KB
ADE_train_00008889.jpeg 212KB
ADE_train_00008678.jpeg 211KB
ADE_train_00010879.jpeg 211KB
ADE_train_00004358.jpeg 209KB
ADE_train_00014924.jpeg 209KB
ADE_train_00011009.jpeg 208KB
ADE_val_00000501.jpeg 207KB
ADE_train_00007759.jpeg 206KB
ADE_train_00012020.jpeg 205KB
ADE_train_00019837.jpeg 201KB
ADE_train_00009937.jpeg 199KB
ADE_train_00012592.jpeg 197KB
ADE_train_00015014.jpeg 194KB
ADE_train_00007643.jpeg 194KB
ADE_val_00000917.jpeg 191KB
ADE_train_00004571.jpeg 189KB
ADE_train_00000650.jpeg 188KB
ADE_train_00004581.jpeg 188KB
ADE_train_00008689.jpeg 185KB
ADE_train_00003056.jpeg 183KB
ADE_train_00007651.jpeg 182KB
ADE_train_00005271.jpeg 182KB
ADE_train_00012889.jpeg 182KB
ADE_train_00019879.jpeg 181KB
ADE_train_00004345.jpeg 177KB
ADE_train_00012824.jpeg 176KB
ADE_val_00001782.jpeg 175KB
ADE_train_00007650.jpeg 173KB
ADE_train_00012365.jpeg 171KB
ADE_train_00010013.jpeg 166KB
ADE_train_00004579.jpeg 166KB
ADE_train_00008886.jpeg 165KB
ADE_train_00007915.jpeg 165KB
ADE_train_00019814.jpeg 164KB
ADE_train_00002530.jpeg 163KB
ADE_train_00005011.jpeg 162KB
ADE_train_00012965.jpeg 158KB
ADE_train_00014266.jpeg 153KB
ADE_train_00000569.jpeg 151KB
ADE_val_00000594.jpeg 150KB
ADE_train_00005018.jpeg 150KB
ADE_train_00007758.jpeg 149KB
ADE_train_00008898.jpeg 149KB
ADE_train_00019828.jpeg 146KB
ADE_train_00011844.jpeg 146KB
ADE_val_00001555.jpeg 146KB
ADE_train_00002226.jpeg 145KB
ADE_val_00001580.jpeg 144KB
ADE_train_00008253.jpeg 143KB
ADE_val_00001190.jpeg 142KB
ADE_train_00007648.jpeg 141KB
ADE_train_00015013.jpeg 140KB
ADE_train_00011057.jpeg 138KB
ADE_train_00009925.jpeg 137KB
ADE_train_00012845.jpeg 137KB
ADE_train_00019808.jpeg 137KB
ADE_train_00013943.jpeg 137KB
ADE_val_00001200.jpeg 135KB
ADE_val_00001571.jpeg 135KB
ADE_train_00005886.jpeg 134KB
ADE_train_00013145.jpeg 134KB
ADE_train_00012297.jpeg 134KB
ADE_val_00000919.jpeg 134KB
ADE_train_00006597.jpeg 130KB
ADE_train_00014829.jpeg 129KB
ADE_train_00012025.jpeg 128KB
ADE_train_00008276.jpeg 127KB
ADE_train_00012326.jpeg 125KB
ADE_train_00013741.jpeg 124KB
ADE_train_00003149.jpeg 124KB
共 2000 条
- 1
- 2
- 3
- 4
- 5
- 6
- 20
资源评论
听风吹等浪起
- 粉丝: 1w+
- 资源: 1239
上传资源 快速赚钱
- 我的内容管理 展开
- 我的资源 快来上传第一个资源
- 我的收益 登录查看自己的收益
- 我的积分 登录查看自己的积分
- 我的C币 登录后查看C币余额
- 我的收藏
- 我的下载
- 下载帮助
安全验证
文档复制为VIP权益,开通VIP直接复制
信息提交成功