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=(255, 0, 0), opacity=0.5)
没有合适的资源?快使用搜索试试~ 我知道了~
温馨提示
项目包含:遥感图像下山川、湖泊全景图像分割数据集(划分了训练集和测试集),文件以文件夹格式储存,可直接用作图像分割数据集,无需额外处理。 数据集为遥感图像下山脉、湖泊全景分割数据集,已经做了数据集的划分 数据集总大小:170MB 数据集介绍:分为训练集、测试集 训练集:images图片目录+masks模板目录,12814张图片和12814个对应的mask图片 测试集:images图片目录+masks模板目录,3203张图片和3203个对应的mask图片 除此之外,包含一个图像分割的可视化脚本,随机提取一张图片,将其原始图片、GT图像、GT在原图蒙板的图像展示,并保存在当前目录下
资源推荐
资源详情
资源评论
收起资源包目录
深度学习之图像分割数据集:遥感图像下山川、湖泊全景图像分割数据集(划分了训练集和测试集) (2000个子文件)
012600.jpeg 3KB
002900.jpeg 3KB
009047.jpeg 2KB
009104.jpeg 2KB
008095.jpeg 2KB
015847.jpeg 2KB
006230.jpeg 2KB
013576.jpeg 2KB
015815.jpeg 2KB
009985.jpeg 2KB
011229.jpeg 2KB
000650.jpeg 2KB
007492.jpeg 2KB
007540.jpeg 2KB
010315.jpeg 2KB
001743.jpeg 2KB
005685.jpeg 2KB
002606.jpeg 2KB
001791.jpeg 2KB
001164.jpeg 2KB
011164.jpeg 2KB
000880.jpeg 2KB
006164.jpeg 2KB
012051.jpeg 2KB
011284.jpeg 2KB
014103.jpeg 2KB
005686.jpeg 2KB
014555.jpeg 2KB
004319.jpeg 2KB
002278.jpeg 2KB
015654.jpeg 2KB
012476.jpeg 2KB
003236.jpeg 2KB
012847.jpeg 2KB
002698.jpeg 2KB
006012.jpeg 2KB
003695.jpeg 2KB
000429.jpeg 2KB
003698.jpeg 2KB
012152.jpeg 2KB
001351.jpeg 2KB
009266.jpeg 2KB
008882.jpeg 2KB
003238.jpeg 2KB
011544.jpeg 2KB
002056.jpeg 2KB
013186.jpeg 2KB
007537.jpeg 2KB
006369.jpeg 2KB
005406.jpeg 2KB
008244.jpeg 2KB
011638.jpeg 2KB
010131.jpeg 2KB
004239.jpeg 2KB
010253.jpeg 2KB
012340.jpeg 2KB
015911.jpeg 2KB
010815.jpeg 2KB
014811.jpeg 2KB
000500.jpeg 2KB
011569.jpeg 2KB
000261.jpeg 2KB
000487.jpeg 2KB
006915.jpeg 2KB
005971.jpeg 2KB
007450.jpeg 2KB
000804.jpeg 2KB
003903.jpeg 2KB
009418.jpeg 2KB
002425.jpeg 2KB
015636.jpeg 2KB
001266.jpeg 2KB
008207.jpeg 2KB
007543.jpeg 2KB
004887.jpeg 2KB
012469.jpeg 2KB
012814.jpeg 2KB
009172.jpeg 2KB
003376.jpeg 2KB
012306.jpeg 2KB
001992.jpeg 2KB
005212.jpeg 2KB
002632.jpeg 2KB
014210.jpeg 2KB
005832.jpeg 2KB
008039.jpeg 2KB
000355.jpeg 2KB
006947.jpeg 2KB
011913.jpeg 2KB
005386.jpeg 2KB
009869.jpeg 2KB
000938.jpeg 2KB
009096.jpeg 2KB
004860.jpeg 2KB
003694.jpeg 2KB
001073.jpeg 2KB
001058.jpeg 2KB
014014.jpeg 2KB
012703.jpeg 2KB
009620.jpeg 2KB
共 2000 条
- 1
- 2
- 3
- 4
- 5
- 6
- 20
资源评论
听风吹等浪起
- 粉丝: 1w+
- 资源: 1178
下载权益
C知道特权
VIP文章
课程特权
开通VIP
上传资源 快速赚钱
- 我的内容管理 展开
- 我的资源 快来上传第一个资源
- 我的收益 登录查看自己的收益
- 我的积分 登录查看自己的积分
- 我的C币 登录后查看C币余额
- 我的收藏
- 我的下载
- 下载帮助
最新资源
- 叶片真菌K1-7 L1-9 M1-6 N1-11 P1-8 Q1-3 R1-4 2023-03-22 17 时 39 分.tif
- Simulink建模的基础知识(精)(1)
- IEC 62282-6-106 2024 燃料电池技术-第6-106部分:微型燃料电池动力系统-安全-间接8类(腐蚀性)化合物
- 连续因子要点和难点具体应用和代码实例
- 1715409496108440_file
- 和芯星通UB482导航定位定向卡用户手册
- ASP中国节能环保科技网站(源代码+lw).zip
- 目标检测YOLO实战应用案例100讲-基于YOLOV5的小目标检测
- 1HuffmanTree.cpp
- 基于Java的两个通用安全模块的设计与实现源码.zip
资源上传下载、课程学习等过程中有任何疑问或建议,欢迎提出宝贵意见哦~我们会及时处理!
点击此处反馈
安全验证
文档复制为VIP权益,开通VIP直接复制
信息提交成功