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[:, :, ::-1]
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)
没有合适的资源?快使用搜索试试~ 我知道了~
温馨提示
项目包含水泥道路裂缝数据集(二值图像分割),文件以文件夹格式储存,可直接用作图像分割数据集,无需额外处理。 数据集介绍:分为训练集、测试集 训练集:images图片目录+masks模板目录,95张图片和95个对应的mask图片 测试集:images图片目录+masks模板目录,23张图片和23个对应的mask图片 除此之外,包含一个图像分割的可视化脚本,随机提取一张图片,将其原始图片、GT图像、GT在原图蒙板的图像展示,并保存在当前目录下
资源推荐
资源详情
资源评论
收起资源包目录
计算机视觉之图像分割数据集:水泥道路裂缝数据集(二值图像分割) (238个子文件)
089.jpg 43KB
090.jpg 41KB
054.jpg 40KB
098.jpg 39KB
035.jpg 38KB
096.jpg 37KB
088.jpg 37KB
097.jpg 37KB
095.jpg 36KB
109.jpg 36KB
091.jpg 36KB
092.jpg 36KB
055.jpg 35KB
021.jpg 35KB
012.jpg 35KB
115.jpg 35KB
099.jpg 35KB
027.jpg 35KB
073.jpg 35KB
118.jpg 35KB
082.jpg 35KB
111.jpg 35KB
108.jpg 34KB
114.jpg 34KB
107.jpg 34KB
112.jpg 34KB
101.jpg 34KB
094.jpg 33KB
053.jpg 33KB
113.jpg 33KB
116.jpg 33KB
083.jpg 33KB
103.jpg 33KB
030.jpg 33KB
052.jpg 33KB
031.jpg 33KB
013.jpg 33KB
024.jpg 33KB
085.jpg 33KB
093.jpg 33KB
019.jpg 33KB
077.jpg 33KB
110.jpg 32KB
048.jpg 32KB
105.jpg 32KB
029.jpg 32KB
070.jpg 32KB
016.jpg 31KB
065.jpg 31KB
087.jpg 31KB
102.jpg 31KB
100.jpg 31KB
028.jpg 31KB
084.jpg 31KB
117.jpg 31KB
006.jpg 31KB
036.jpg 31KB
067.jpg 31KB
050.jpg 30KB
009.jpg 30KB
106.jpg 30KB
104.jpg 30KB
026.jpg 30KB
020.jpg 30KB
035.jpg 29KB
023.jpg 29KB
078.jpg 29KB
069.jpg 29KB
001.jpg 29KB
064.jpg 29KB
061.jpg 28KB
002.jpg 28KB
086.jpg 28KB
066.jpg 28KB
076.jpg 28KB
022.jpg 28KB
017.jpg 28KB
059.jpg 28KB
075.jpg 28KB
072.jpg 28KB
041.jpg 28KB
062.jpg 27KB
015.jpg 27KB
060.jpg 27KB
071.jpg 27KB
097.jpg 27KB
074.jpg 27KB
044.jpg 27KB
039.jpg 27KB
079.jpg 27KB
068.jpg 26KB
011.jpg 26KB
014.jpg 26KB
033.jpg 26KB
045.jpg 26KB
005.jpg 26KB
046.jpg 25KB
032.jpg 25KB
063.jpg 25KB
025.jpg 25KB
共 238 条
- 1
- 2
- 3
资源评论
听风吹等浪起
- 粉丝: 1w+
- 资源: 1239
下载权益
C知道特权
VIP文章
课程特权
开通VIP
上传资源 快速赚钱
- 我的内容管理 展开
- 我的资源 快来上传第一个资源
- 我的收益 登录查看自己的收益
- 我的积分 登录查看自己的积分
- 我的C币 登录后查看C币余额
- 我的收藏
- 我的下载
- 下载帮助
最新资源
- 谷歌浏览器自动化测试版113.0.5672.0(包含linux,windows32/64,mac三个版本,不会自动更新)
- uniapp中tab切换,底部内容跟着移动,相反,底部移动,tab也跟着切换-组件
- 基于JS+TS实现跨平台3D相机控制器-附项目源码-优质项目分享.zip
- 跨相机-基于Rust实现的跨平台相机捕获-附项目源码-优质项目分享.zip
- odise 14离线安装包 大众斯柯达奥迪 5054 6153
- 网页设计期末作业-纯html加css+少量js-盗墓笔记旅游导航网站.rar
- 算法笔记模拟退火.rar
- MATLAB大数据仿真案例-蚁群算法(ACO)用于求解旅行商(TSP)问题.rar
- 基于yolov5的吸烟行为检测源码+模型.zip
- MySQL基础知识-个人笔记.rar
资源上传下载、课程学习等过程中有任何疑问或建议,欢迎提出宝贵意见哦~我们会及时处理!
点击此处反馈
安全验证
文档复制为VIP权益,开通VIP直接复制
信息提交成功