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?spm=1001.2014.3001.5501
# 将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/Blowhole/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('.jpg', '.png')
# opacity 越小,掩膜效果越深
main(imagePath=img_path, labelPath=mask_path, bgr=(0, 0, 255), opacity=0.5)
没有合适的资源?快使用搜索试试~ 我知道了~
温馨提示
项目包含5种常见瓷砖缺陷分割数据集(二值图像分割),按照文件夹格式存储,可直接用作图像分割数据集,无需额外处理。 数据共有5个子文件夹,代表5种不同的瓷砖缺陷数据。(气孔、破碎、裂纹、磨损、凹凸不平) 每种缺陷子文件夹,均包含原始图像和对应的缺陷GT模板
资源推荐
资源详情
资源评论
收起资源包目录
计算机视觉之图像分割数据集:5种常见瓷砖缺陷分割数据集(二值图像分割) (786个子文件)
exp6_num_972.jpg 73KB
exp6_num_3279.jpg 72KB
exp5_num_9549.jpg 72KB
exp6_num_7102.jpg 71KB
exp5_num_7054.jpg 67KB
exp6_num_28424.jpg 66KB
exp6_num_109350.jpg 66KB
exp6_num_117060.jpg 66KB
exp6_num_135686.jpg 66KB
exp6_num_25007.jpg 65KB
exp6_num_342249.jpg 65KB
exp4_num_9520.jpg 65KB
exp5_num_339932.jpg 64KB
exp6_num_33541.jpg 63KB
exp5_num_109313.jpg 63KB
exp5_num_28404.jpg 63KB
exp6_num_139916.jpg 63KB
exp6_num_105927.jpg 62KB
exp4_num_105851.jpg 62KB
exp3_num_3223.jpg 62KB
exp5_num_342212.jpg 62KB
exp5_num_149023.jpg 62KB
exp4_num_109286.jpg 62KB
exp6_num_149097.jpg 62KB
exp5_num_24876.jpg 61KB
exp5_num_139883.jpg 61KB
exp6_num_356817.jpg 61KB
exp6_num_339965.jpg 61KB
exp6_num_45116.jpg 61KB
exp3_num_342176.jpg 61KB
exp3_num_148977.jpg 60KB
exp4_num_24843.jpg 60KB
exp4_num_7036.jpg 60KB
exp5_num_135651.jpg 60KB
exp5_num_356780.jpg 60KB
exp3_num_853.jpg 60KB
exp5_num_348696.jpg 60KB
exp5_num_33487.jpg 59KB
exp6_num_88201.jpg 59KB
exp5_num_45084.jpg 59KB
exp4_num_135615.jpg 59KB
exp6_num_124787.jpg 58KB
exp5_num_3247.jpg 58KB
exp4_num_28388.jpg 58KB
exp5_num_105879.jpg 58KB
exp6_num_352572.jpg 58KB
exp3_num_7023.jpg 58KB
exp3_num_139838.jpg 58KB
exp4_num_139856.jpg 58KB
exp4_num_348673.jpg 58KB
exp4_num_116999.jpg 57KB
exp6_num_348888.jpg 57KB
exp3_num_109271.jpg 57KB
exp4_num_88149.jpg 57KB
exp5_num_88170.jpg 57KB
exp6_num_85665.jpg 57KB
exp6_num_86064.jpg 57KB
exp3_num_24829.jpg 56KB
exp5_num_124744.jpg 56KB
exp4_num_356767.jpg 56KB
exp5_num_352540.jpg 56KB
exp6_num_346414.jpg 56KB
exp4_num_124690.jpg 56KB
exp2_num_3211.jpg 56KB
exp4_num_342190.jpg 55KB
exp5_num_346381.jpg 55KB
exp3_num_28373.jpg 55KB
exp4_num_33407.jpg 55KB
exp3_num_9508.jpg 55KB
exp2_num_342162.jpg 54KB
exp5_num_86031.jpg 54KB
exp5_num_85642.jpg 54KB
exp4_num_85602.jpg 54KB
exp3_num_33395.jpg 54KB
exp4_num_20426.jpg 54KB
exp4_num_85998.jpg 54KB
exp3_num_135593.jpg 53KB
exp3_num_45042.jpg 53KB
exp6_num_26244.jpg 53KB
exp4_num_45057.jpg 53KB
exp3_num_339854.jpg 53KB
exp3_num_348661.jpg 53KB
exp5_num_77692.jpg 53KB
exp6_num_3789.jpg 53KB
exp2_num_105807.jpg 52KB
exp3_num_105826.jpg 52KB
exp5_num_26212.jpg 52KB
exp6_num_77744.jpg 52KB
exp4_num_352527.jpg 52KB
exp3_num_124629.jpg 51KB
exp2_num_7009.jpg 51KB
exp3_num_356756.jpg 51KB
exp3_num_116981.jpg 51KB
exp4_num_26163.jpg 51KB
exp3_num_346347.jpg 51KB
exp5_num_3746.jpg 51KB
exp2_num_109256.jpg 50KB
exp1_num_3191.jpg 50KB
exp6_num_291134.jpg 49KB
exp4_num_339867.jpg 49KB
共 786 条
- 1
- 2
- 3
- 4
- 5
- 6
- 8
资源评论
- yidaqiqi2024-05-18资源很实用,对我启发很大,有很好的参考价值,内容详细。
听风吹等浪起
- 粉丝: 1w+
- 资源: 1242
上传资源 快速赚钱
- 我的内容管理 展开
- 我的资源 快来上传第一个资源
- 我的收益 登录查看自己的收益
- 我的积分 登录查看自己的积分
- 我的C币 登录后查看C币余额
- 我的收藏
- 我的下载
- 下载帮助
安全验证
文档复制为VIP权益,开通VIP直接复制
信息提交成功