import cv2
import numpy as np
import os
import random
from PIL import Image
import matplotlib.pyplot as plt
# 将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)
没有合适的资源?快使用搜索试试~ 我知道了~
图像分割数据集:卫星水体分割【包含数据集(2类别)、标签文件、可视化代码】
共2000个文件
jpg:1999个
py:1个
1.该资源内容由用户上传,如若侵权请联系客服进行举报
2.虚拟产品一经售出概不退款(资源遇到问题,请及时私信上传者)
2.虚拟产品一经售出概不退款(资源遇到问题,请及时私信上传者)
版权申诉
0 下载量 192 浏览量
2024-05-08
13:15:02
上传
评论
收藏 223.16MB 7Z 举报
温馨提示
项目包含:卫星水体分割【包含数据集(2类别)、标签文件、可视化代码】 数据集为500-2000分辨率下的卫星水体分割。分割前景为水体。标签采用255为前景的二值图像,方便观察mask 数据集介绍:总共2273张训练图像和对应的标签、以及568张验证图像和对应的标签 除此之外,包含一个图像分割的可视化脚本,随机提取一张图片,将其原始图片、GT图像、GT在原图蒙板的图像展示,并保存在当前目录下。可以直接运行即可可视化数据
资源推荐
资源详情
资源评论
收起资源包目录
图像分割数据集:卫星水体分割【包含数据集(2类别)、标签文件、可视化代码】 (2000个子文件)
water_body_7316.jpg 4.82MB
water_body_7177.jpg 1020KB
water_body_7185.jpg 858KB
water_body_1.jpg 698KB
water_body_8431.jpg 550KB
water_body_644.jpg 541KB
water_body_1351.jpg 511KB
water_body_6387.jpg 507KB
water_body_7432.jpg 490KB
water_body_8561.jpg 470KB
water_body_22.jpg 425KB
water_body_7149.jpg 409KB
water_body_2318.jpg 388KB
water_body_2295.jpg 378KB
water_body_29.jpg 372KB
water_body_8725.jpg 362KB
water_body_8004.jpg 362KB
water_body_598.jpg 347KB
water_body_2177.jpg 334KB
water_body_7482.jpg 334KB
water_body_8258.jpg 320KB
water_body_2189.jpg 315KB
water_body_20.jpg 312KB
water_body_7348.jpg 301KB
water_body_189.jpg 295KB
water_body_7995.jpg 294KB
water_body_669.jpg 292KB
water_body_7413.jpg 287KB
water_body_7320.jpg 284KB
water_body_8571.jpg 280KB
water_body_2490.jpg 276KB
water_body_8788.jpg 274KB
water_body_7438.jpg 272KB
water_body_7357.jpg 264KB
water_body_8903.jpg 260KB
water_body_7236.jpg 258KB
water_body_7249.jpg 253KB
water_body_596.jpg 252KB
water_body_7681.jpg 251KB
water_body_7195.jpg 242KB
water_body_7381.jpg 238KB
water_body_1846.jpg 238KB
water_body_7360.jpg 235KB
water_body_7341.jpg 228KB
water_body_6.jpg 228KB
water_body_7376.jpg 222KB
water_body_2440.jpg 219KB
water_body_7049.jpg 216KB
water_body_7209.jpg 216KB
water_body_647.jpg 212KB
water_body_2220.jpg 203KB
water_body_1550.jpg 200KB
water_body_2364.jpg 199KB
water_body_7257.jpg 197KB
water_body_7377.jpg 195KB
water_body_636.jpg 194KB
water_body_1567.jpg 192KB
water_body_619.jpg 187KB
water_body_7188.jpg 184KB
water_body_2358.jpg 183KB
water_body_709.jpg 182KB
water_body_5.jpg 181KB
water_body_7374.jpg 177KB
water_body_6386.jpg 169KB
water_body_34.jpg 169KB
water_body_1558.jpg 168KB
water_body_8348.jpg 165KB
water_body_7238.jpg 165KB
water_body_1554.jpg 164KB
water_body_7173.jpg 159KB
water_body_2456.jpg 155KB
water_body_2534.jpg 155KB
water_body_2300.jpg 154KB
water_body_8779.jpg 153KB
water_body_6382.jpg 150KB
water_body_1269.jpg 146KB
water_body_1583.jpg 146KB
water_body_626.jpg 144KB
water_body_2363.jpg 142KB
water_body_1553.jpg 142KB
water_body_6384.jpg 141KB
water_body_1569.jpg 141KB
water_body_7241.jpg 139KB
water_body_2179.jpg 139KB
water_body_120.jpg 138KB
water_body_7373.jpg 138KB
water_body_2383.jpg 136KB
water_body_8754.jpg 135KB
water_body_14.jpg 133KB
water_body_23.jpg 132KB
water_body_187.jpg 131KB
water_body_2350.jpg 131KB
water_body_1830.jpg 130KB
water_body_2188.jpg 129KB
water_body_2526.jpg 128KB
water_body_2181.jpg 127KB
water_body_7674.jpg 127KB
water_body_7388.jpg 127KB
water_body_1097.jpg 126KB
water_body_2448.jpg 124KB
共 2000 条
- 1
- 2
- 3
- 4
- 5
- 6
- 20
资源评论
听风吹等浪起
- 粉丝: 1w+
- 资源: 1359
下载权益
C知道特权
VIP文章
课程特权
开通VIP
上传资源 快速赚钱
- 我的内容管理 展开
- 我的资源 快来上传第一个资源
- 我的收益 登录查看自己的收益
- 我的积分 登录查看自己的积分
- 我的C币 登录后查看C币余额
- 我的收藏
- 我的下载
- 下载帮助
安全验证
文档复制为VIP权益,开通VIP直接复制
信息提交成功