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').replace('.jpg','.bmp')
# opacity 越小,掩膜效果越深
main(imagePath=img_path, labelPath=mask_path, bgr=(0, 0, 255), opacity=0.5)
没有合适的资源?快使用搜索试试~ 我知道了~
图像分割数据集:水下目标图像语义分割(8分割)【数据集+类别标签+可视化代码】
共2000个文件
bmp:1525个
jpg:474个
py:1个
1.该资源内容由用户上传,如若侵权请联系客服进行举报
2.虚拟产品一经售出概不退款(资源遇到问题,请及时私信上传者)
2.虚拟产品一经售出概不退款(资源遇到问题,请及时私信上传者)
版权申诉
0 下载量 43 浏览量
2024-05-11
14:41:57
上传
评论
收藏 159.07MB 7Z 举报
温馨提示
项目包含:水下目标图像语义分割(8分割)【数据集+类别标签+可视化代码】 数据集为640*480分辨率下的水下图像分割。分割前景为人类、海草、珊瑚、岩石、鱼等等。标签的mask图像为调色后的彩色图像,0为背景,为了方便观察,提供可视化代码。数据集的背景简单,前景区域丰富且标注良好,适用于图像分割。可视化代码效果参考:https://blog.csdn.net/qq_44886601/article/details/135764589 数据做了预处理,对不同的数据进行随机的旋转等等 数据集介绍:分为训练集、测试集 训练集:images图片目录+masks模板目录,1525张图片和1525个对应的mask图片 测试集:images图片目录+masks模板目录,110图片和110个对应的mask图片 除此之外,包含一个图像分割的可视化脚本,随机提取一张图片,将其原始图片、GT图像、GT在原图蒙板的图像展示,并保存在当前目录下
资源推荐
资源详情
资源评论
收起资源包目录
图像分割数据集:水下目标图像语义分割(8分割)【数据集+类别标签+可视化代码】 (2000个子文件)
d_r_617_.bmp 5.89MB
d_r_51_.bmp 5.89MB
d_r_3_.bmp 5.89MB
d_r_198_.bmp 5.89MB
d_r_256_.bmp 5.89MB
d_r_657_.bmp 5.89MB
d_r_404_.bmp 5.89MB
d_r_223_.bmp 5.89MB
d_r_78_.bmp 5.89MB
d_r_474_.bmp 5.89MB
d_r_33_.bmp 5.89MB
d_r_700_.bmp 5.89MB
d_r_788_.bmp 5.89MB
d_r_689_.bmp 5.89MB
d_r_542_.bmp 5.89MB
d_r_145_.bmp 5.89MB
d_r_249_.bmp 3.52MB
d_r_18_.bmp 2.64MB
d_r_189_.bmp 2.64MB
d_r_564_.bmp 2.64MB
d_r_112_.bmp 2.64MB
d_r_179_.bmp 2.64MB
d_r_132_.bmp 2.64MB
d_r_348_.bmp 2.64MB
d_r_428_.bmp 2.64MB
d_r_171_.bmp 2.64MB
d_r_732_.bmp 2.64MB
d_r_59_.bmp 2.64MB
d_r_63_.bmp 2.64MB
d_r_327_.bmp 2.64MB
d_r_293_.bmp 2.64MB
d_r_682_.bmp 2.64MB
d_r_661_.bmp 2.64MB
d_r_776_.bmp 2.64MB
d_r_466_.bmp 2.64MB
d_r_741_.bmp 2.64MB
d_r_52_.bmp 2.64MB
d_r_22_.bmp 2.64MB
d_r_333_.bmp 2.64MB
d_r_303_.bmp 2.64MB
d_r_692_.bmp 2.64MB
d_r_604_.bmp 2.64MB
d_r_789_.bmp 2.64MB
d_r_664_.bmp 2.64MB
d_r_215_.bmp 2.64MB
d_r_546_.bmp 2.64MB
d_r_217_.bmp 2.64MB
d_r_473_.bmp 2.64MB
d_r_20_.bmp 2.64MB
d_r_759_.bmp 2.64MB
d_r_795_.bmp 2.64MB
d_r_61_.bmp 2.64MB
d_r_403_.bmp 2.64MB
d_r_702_.bmp 2.64MB
d_r_5_.bmp 2.64MB
d_r_439_.bmp 2.64MB
d_r_618_.bmp 2.64MB
d_r_162_.bmp 2.64MB
d_r_649_.bmp 2.64MB
d_r_174_.bmp 2.64MB
d_r_301_.bmp 2.64MB
d_r_274_.bmp 2.64MB
d_r_46_.bmp 2.64MB
d_r_14_.bmp 2.64MB
d_r_638_.bmp 2.64MB
d_r_745_.bmp 2.64MB
d_r_800_.bmp 2.64MB
d_r_674_.bmp 2.64MB
d_r_470_.bmp 2.64MB
d_r_258_.bmp 2.64MB
d_r_615_.bmp 2.64MB
d_r_675_.bmp 2.37MB
d_r_656_.bmp 2.37MB
d_r_125_.bmp 2.37MB
d_r_541_.bmp 2.37MB
d_r_690_.bmp 2.37MB
d_r_633_.bmp 2.37MB
d_r_319_.bmp 2.37MB
f_r_1546_.bmp 1.48MB
f_r_1783_.bmp 1.48MB
f_r_1939_.bmp 1.48MB
d_r_77_.bmp 1.48MB
d_r_576_.bmp 1.48MB
f_r_1744_.bmp 1.48MB
n_l_30_.bmp 1.48MB
f_r_320_.bmp 1.48MB
d_r_450_.bmp 1.48MB
d_r_632_.bmp 1.48MB
n_l_51_.bmp 1.48MB
f_r_1828_.bmp 1.48MB
d_r_710_.bmp 1.48MB
n_l_55_.bmp 1.48MB
d_r_587_.bmp 1.48MB
n_l_67_.bmp 1.48MB
f_r_1080_.bmp 1.48MB
d_r_163_.bmp 1.48MB
n_l_105_.bmp 1.48MB
n_l_21_.bmp 1.48MB
f_r_979_.bmp 1.48MB
d_r_549_.bmp 1.48MB
共 2000 条
- 1
- 2
- 3
- 4
- 5
- 6
- 20
资源评论
听风吹等浪起
- 粉丝: 1w+
- 资源: 1245
上传资源 快速赚钱
- 我的内容管理 展开
- 我的资源 快来上传第一个资源
- 我的收益 登录查看自己的收益
- 我的积分 登录查看自己的积分
- 我的C币 登录后查看C币余额
- 我的收藏
- 我的下载
- 下载帮助
安全验证
文档复制为VIP权益,开通VIP直接复制
信息提交成功