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)
没有合适的资源?快使用搜索试试~ 我知道了~
温馨提示
项目包含:卫星道路图像分割数据集(划分了训练集和测试集) 数据集为卫星道路图像分割数据集,已经做了数据集的划分。 数据集总大小:329MB 数据集介绍:分为训练集、测试集 训练集:images图片目录+masks模板目录,3600张图片和3600个对应的mask图片 测试集:images图片目录+masks模板目录,900张图片和900个对应的mask图片 除此之外,包含一个图像分割的可视化脚本,随机提取一张图片,将其原始图片、GT图像、GT在原图蒙板的图像展示,并保存在当前目录下
资源推荐
资源详情
资源评论
收起资源包目录
深度学习之图像分割数据集:卫星道路图像分割数据集(划分了训练集和测试集) (2000个子文件)
T3184.png 2KB
T1944.png 2KB
T1449.png 2KB
T3705.png 2KB
T883.png 2KB
T4150.png 2KB
T782.png 2KB
T3125.png 2KB
T2877.png 2KB
T3857.png 2KB
T1557.png 2KB
T2839.png 2KB
T1264.png 2KB
T1672.png 2KB
T4132.png 2KB
T1032.png 2KB
T3277.png 2KB
T2943.png 2KB
T3680.png 2KB
T3953.png 2KB
T2202.png 2KB
T3763.png 2KB
T3816.png 2KB
T2634.png 2KB
T505.png 2KB
T1073.png 2KB
T1724.png 2KB
T1064.png 2KB
T4384.png 2KB
T1525.png 2KB
T4109.png 2KB
T2564.png 2KB
T1561.png 2KB
T237.png 2KB
T3482.png 2KB
T3323.png 2KB
T2487.png 2KB
T2230.png 2KB
T61.png 2KB
T4118.png 2KB
T1736.png 2KB
T1744.png 2KB
T3082.png 2KB
T415.png 2KB
T987.png 2KB
T2168.png 2KB
T1039.png 2KB
T4393.png 2KB
T3972.png 2KB
T4257.png 2KB
T972.png 2KB
T2575.png 2KB
T2489.png 2KB
T1036.png 2KB
T1391.png 2KB
T4361.png 2KB
T2601.png 2KB
T2359.png 2KB
T2729.png 2KB
T3647.png 2KB
T3847.png 2KB
T1778.png 2KB
T3704.png 2KB
T2930.png 2KB
T4078.png 2KB
T3472.png 2KB
T2717.png 2KB
T2902.png 2KB
T3600.png 2KB
T151.png 2KB
T1899.png 2KB
T3062.png 2KB
T1836.png 2KB
T1800.png 2KB
T2486.png 1KB
T3034.png 1KB
T4119.png 1KB
T2774.png 1KB
T1864.png 1KB
T2956.png 1KB
T193.png 1KB
T953.png 1KB
T3510.png 1KB
T2767.png 1KB
T2244.png 1KB
T3356.png 1KB
T396.png 1KB
T2189.png 1KB
T232.png 1KB
T4280.png 1KB
T3963.png 1KB
T3830.png 1KB
T2518.png 1KB
T1723.png 1KB
T79.png 1KB
T20.png 1KB
T3818.png 1KB
T3017.png 1KB
T1458.png 1KB
T3041.png 1KB
共 2000 条
- 1
- 2
- 3
- 4
- 5
- 6
- 20
资源评论
听风吹等浪起
- 粉丝: 1w+
- 资源: 1220
下载权益
C知道特权
VIP文章
课程特权
开通VIP
上传资源 快速赚钱
- 我的内容管理 展开
- 我的资源 快来上传第一个资源
- 我的收益 登录查看自己的收益
- 我的积分 登录查看自己的积分
- 我的C币 登录后查看C币余额
- 我的收藏
- 我的下载
- 下载帮助
最新资源
- kmp算法-使用kmp算法在Golang中实现字符串匹配.zip
- LTC2756 +LTC6244+LTC6655 设计18位乘法串行输入电流输出DAC数模转换器模块硬件(原理图+PCB)工程
- kmp算法-基于C语言实现KMP算法.zip
- DMS智能座舱项目-Python基于深度学习实现驾驶员分心行为监测系统源码.zip
- 2023年数控机床与编程知识点总结归纳(精选试题附答案).docx
- 2023年数控机床与编程考点总结.pdf
- 2023年数控机床与编程考点题型与解题方法(精选试题附答案).docx
- 3CHISS统计软件介绍(Master).pdf
- 3PE防腐绝缘燃气管道接口施工方案.doc
- 51单片机C语言教程.doc
资源上传下载、课程学习等过程中有任何疑问或建议,欢迎提出宝贵意见哦~我们会及时处理!
点击此处反馈
安全验证
文档复制为VIP权益,开通VIP直接复制
信息提交成功