from skimage.segmentation import slic, mark_boundaries import torchvision.transforms as transforms import numpy as np from PIL import Image import matplotlib.pyplot as plt # 加载图像 image = Image.open('img.png') # 转换为 PyTorch 张量 transform = transforms.ToTensor() img_tensor = transform(image).unsqueeze(0) # 将 PyTorch 张量转换为 Numpy 数组 img_np = img_tensor.numpy().transpose(0, 2, 3, 1)[0] # 使用 SLIC 算法生成超像素标记图 segments = slic(img_np, n_segments=100, compactness=10) # 可视化超像素标记图 segment_img = mark_boundaries(img_np, segments) # 将 Numpy 数组转换为 PIL 图像 segment_img = Image.fromarray((segment_img * 255).astype(np.uint8)) # 保存超像素标记图 segment_img.save('segments.jpg') n_segments = np.max(segments) + 1 # 初始化超像素块的区域 segment_regions = np.zeros((n_segments, img_np.shape[0], img_np.shape[1])) # 遍历每个超像素块 for i in range(n_segments): # 获取当前超像素块的掩码 mask = (segments == i) # 将当前超像素块的掩码赋值给超像素块的区域 segment_regions[i][mask] = 1 # 保存超像素块的区域 np.save('segment_regions.npy', segment_regions) # 加载超像素块的区域 segment_regions = np.load('segment_regions.npy') # 取出第一个超像素块的区域 segment_region = segment_regions[0] # 显示超像素块的区域 plt.imshow(segment_region, cmap='gray') plt.show(),上述代码显示超像素块区域时显示的图像为全黑,怎么回事

时间: 2024-03-08 21:47:46 浏览: 40
RAR

project.rar_Project.rar_image segmentation_project_upwardlak_图像处

出现这种情况可能是因为超像素块的区域被归一化为了0-1之间的浮点数。在可视化时,需要将其还原为0-255之间的整数。 可以尝试修改以下代码,将 `segment_region` 进行缩放,将其值映射到0-255之间: ```python # 缩放超像素块的区域,将其值映射到0-255之间 segment_region = (segment_region * 255).astype(np.uint8) # 显示超像素块的区域 plt.imshow(segment_region, cmap='gray') plt.show() ``` 这样就可以正常显示超像素块的区域了。
阅读全文

相关推荐

import torch import torch.nn.functional as F from skimage.segmentation import slic, mark_boundaries import torchvision.transforms as transforms import numpy as np from PIL import Image import matplotlib.pyplot as plt # 加载图像 image = Image.open('3.jpg') # 转换为 PyTorch 张量 transform = transforms.ToTensor() img_tensor = transform(image).unsqueeze(0) # 将 PyTorch 张量转换为 Numpy 数组 img_np = img_tensor.numpy().transpose(0, 2, 3, 1)[0] # 使用 SLIC 算法生成超像素标记图 segments = slic(img_np, n_segments=60, compactness=10) # 可视化超像素索引映射 plt.imshow(segments, cmap='gray') plt.show() # 将超像素索引映射可视化 segment_img = mark_boundaries(img_np, segments) # 将 Numpy 数组转换为 PIL 图像 segment_img = Image.fromarray((segment_img * 255).astype(np.uint8)) # 保存超像素索引映射可视化 segment_img.save('segment_map.jpg') # 定义超像素池化函数 def superpixel_pooling(feature_map, segments): # 获取超像素数量和特征维度 n_segments = np.unique(segments).size n_channels = feature_map.shape[0] # 初始化超像素特征 pooled_features = torch.zeros((n_segments, n_channels)) # 对每个超像素内的像素特征进行聚合 for segment_id in range(n_segments): mask = (segments == segment_id).reshape(-1, 1, 1) pooled_feature = (feature_map * mask.float()).sum(dim=(1, 2)) / mask.sum() pooled_features[segment_id] = pooled_feature return pooled_features # 进行超像素池化 pooled_features = superpixel_pooling(img_tensor, segments) # 可视化超像素特征图 plt.imshow(pooled_features.transpose(0, 1), cmap='gray') plt.show(),上述代码出现问题:AttributeError: 'numpy.ndarray' object has no attribute 'float'

rom skimage.segmentation import slic, mark_boundaries import torchvision.transforms as transforms import numpy as np from PIL import Image import matplotlib.pyplot as plt # 加载图像 image = Image.open('3.jpg') # 转换为 PyTorch 张量 transform = transforms.ToTensor() img_tensor = transform(image).unsqueeze(0) # 将 PyTorch 张量转换为 Numpy 数组 img_np = img_tensor.numpy().transpose(0, 2, 3, 1)[0] # 使用 SLIC 算法生成超像素标记图 segments = slic(img_np, n_segments=60, compactness=10) # 可视化超像素索引映射 plt.imshow(segments, cmap='gray') plt.show() # 将超像素索引映射可视化 segment_img = mark_boundaries(img_np, segments) # 将 Numpy 数组转换为 PIL 图像 segment_img = Image.fromarray((segment_img * 255).astype(np.uint8)) # 保存超像素索引映射可视化 segment_img.save('segment_map.jpg') 将上述代码中引入超像素池化代码:import cv2 import numpy as np # 读取图像 img = cv2.imread('3.jpg') # 定义超像素分割器 num_segments = 60 # 超像素数目 slic = cv2.ximgproc.createSuperpixelSLIC(img, cv2.ximgproc.SLICO, num_segments) # 进行超像素分割 slic.iterate(10) # 获取超像素标签和数量 labels = slic.getLabels() num_label = slic.getNumberOfSuperpixels() # 对每个超像素进行池化操作,这里使用平均值池化 pooled = [] for i in range(num_label): mask = labels == i region = img[mask] pooled.append(region.mean(axis=0)) # 将池化后的特征图可视化 pooled = np.array(pooled, dtype=np.uint8) pooled_features = pooled.reshape(-1) pooled_img = cv2.resize(pooled_features, (img.shape[1], img.shape[0]), interpolation=cv2.INTER_NEAREST) print(pooled_img.shape) cv2.imshow('Pooled Image', pooled_img) cv2.waitKey(0),并显示超像素池化后的特征图

最新推荐

recommend-type

华为云:云服务器ECS实战部署.docx

华为云:云服务器ECS实战部署.docx
recommend-type

【光学】基于matlab OAM涡旋光束(拉盖尔-高斯光束)【Matlab仿真 7508期】.zip

【光学】基于matlab OAM涡旋光束(拉盖尔-高斯光束)【Matlab仿真 7508期】.zip
recommend-type

【无线定位】基于matlab改进的TOA求解信号最优的基站定位优化问题【Matlab仿真 8797期】.zip

CSDN Matlab武动乾坤上传的资料均有对应的代码,代码均可运行,亲测可用,适合小白; 1、代码压缩包内容 主函数:main.m; 调用函数:其他m文件;无需运行 运行结果效果图; 2、代码运行版本 Matlab 2019b;若运行有误,根据提示修改;若不会,私信博主; 3、运行操作步骤 步骤一:将所有文件放到Matlab的当前文件夹中; 步骤二:双击打开main.m文件; 步骤三:点击运行,等程序运行完得到结果; 4、仿真咨询 如需其他服务,可私信博主或扫描博客文章底部QQ名片; 4.1 博客或资源的完整代码提供 4.2 期刊或参考文献复现 4.3 Matlab程序定制 4.4 科研合作
recommend-type

Terraform:Terraform与Google Cloud集成实战.docx

Terraform:Terraform与Google Cloud集成实战.docx
recommend-type

【姿态估计】基于matlab扩展卡尔曼滤波器DEKF和9轴IMU姿态估计【Matlab仿真 8017期】.zip

CSDN Matlab武动乾坤上传的资料均有对应的代码,代码均可运行,亲测可用,适合小白; 1、代码压缩包内容 主函数:main.m; 调用函数:其他m文件;无需运行 运行结果效果图; 2、代码运行版本 Matlab 2019b;若运行有误,根据提示修改;若不会,私信博主; 3、运行操作步骤 步骤一:将所有文件放到Matlab的当前文件夹中; 步骤二:双击打开main.m文件; 步骤三:点击运行,等程序运行完得到结果; 4、仿真咨询 如需其他服务,可私信博主或扫描博客文章底部QQ名片; 4.1 博客或资源的完整代码提供 4.2 期刊或参考文献复现 4.3 Matlab程序定制 4.4 科研合作
recommend-type

NIST REFPROP问题反馈与解决方案存储库

资源摘要信息:"NIST REFPROP是一个计算流体热力学性质的软件工具,由美国国家标准技术研究院(National Institute of Standards and Technology,简称NIST)开发。REFPROP能够提供精确的热力学和传输性质数据,广泛应用于石油、化工、能源、制冷等行业。它能够处理多种纯组分和混合物的性质计算,并支持多种方程和混合规则。用户在使用REFPROP过程中可能遇到问题,这时可以利用本存储库报告遇到的问题,寻求帮助。需要注意的是,在报告问题前,用户应确保已经查看了REFPROP的常见问题页面,避免提出重复问题。同时,提供具体的问题描述和示例非常重要,因为仅仅说明“不起作用”是不足够的。在报告问题时,不应公开受知识产权保护或版权保护的代码或其他内容。"
recommend-type

管理建模和仿真的文件

管理Boualem Benatallah引用此版本:布阿利姆·贝纳塔拉。管理建模和仿真。约瑟夫-傅立叶大学-格勒诺布尔第一大学,1996年。法语。NNT:电话:00345357HAL ID:电话:00345357https://theses.hal.science/tel-003453572008年12月9日提交HAL是一个多学科的开放存取档案馆,用于存放和传播科学研究论文,无论它们是否被公开。论文可以来自法国或国外的教学和研究机构,也可以来自公共或私人研究中心。L’archive ouverte pluridisciplinaire
recommend-type

gpuR包在R Markdown中的应用:创建动态报告的5大技巧

![ gpuR包在R Markdown中的应用:创建动态报告的5大技巧](https://codingclubuc3m.rbind.io/post/2019-09-24_files/image1.png) # 1. gpuR包简介与安装 ## gpuR包简介 gpuR是一个专为R语言设计的GPU加速包,它充分利用了GPU的强大计算能力,将原本在CPU上运行的计算密集型任务进行加速。这个包支持多种GPU计算框架,包括CUDA和OpenCL,能够处理大规模数据集和复杂算法的快速执行。 ## 安装gpuR包 安装gpuR包是开始使用的第一步,可以通过R包管理器轻松安装: ```r insta
recommend-type

如何利用matrix-nio库,通过Shell脚本和Python编程,在***网络中创建并运行一个机器人?请提供详细的步骤和代码示例。

matrix-nio库是一个强大的Python客户端库,用于与Matrix网络进行交互,它可以帮助开发者实现机器人与***网络的互动功能。为了创建并运行这样的机器人,你需要遵循以下步骤: 参考资源链接:[matrix-nio打造***机器人下载指南](https://wenku.csdn.net/doc/2oa639sw55?spm=1055.2569.3001.10343) 1. 下载并解压《matrix-nio打造***机器人下载指南》资源包。资源包中的核心项目文件夹'tiny-matrix-bot-main'将作为你的工作目录。 2. 通过命令行工具进入'tiny-
recommend-type

掌握LeetCode习题的系统开源答案

资源摘要信息:"LeetCode答案集 - LeetCode习题解答详解" 1. LeetCode平台概述: LeetCode是一个面向计算机编程技能提升的在线平台,它提供了大量的算法和数据结构题库,供编程爱好者和软件工程师练习和提升编程能力。LeetCode习题的答案可以帮助用户更好地理解问题,并且通过比较自己的解法与标准答案来评估自己的编程水平,从而在实际面试中展示更高效的编程技巧。 2. LeetCode习题特点: LeetCode题目设计紧贴企业实际需求,题目难度从简单到困难不等,涵盖了初级算法、数据结构、系统设计等多个方面。通过不同难度级别的题目,LeetCode能够帮助用户全面提高编程和算法设计能力,同时为求职者提供了一个模拟真实面试环境的平台。 3. 系统开源的重要性: 所谓系统开源,指的是一个系统的源代码是可以被公开查看、修改和发布的。开源对于IT行业至关重要,因为它促进了技术的共享和创新,使得开发者能够共同改进软件,同时也使得用户可以自由选择并信任所使用的软件。开源系统的透明性也使得安全审计和漏洞修补更加容易进行。 4. LeetCode习题解答方法: - 初学者应从基础的算法和数据结构题目开始练习,逐步提升解题速度和准确性。 - 在编写代码前,先要分析问题,明确算法的思路和步骤。 - 编写代码时,注重代码的可读性和效率。 - 编写完毕后,测试代码以确保其正确性,同时考虑边界条件和特殊情况。 - 查看LeetCode平台提供的官方解答和讨论区的其他用户解答,学习不同的解题思路。 - 在社区中与他人交流,分享自己的解法,从反馈中学习并改进。 5. LeetCode使用技巧: - 理解题目要求,注意输入输出格式。 - 学习并掌握常见的算法技巧,如动态规划、贪心算法、回溯法等。 - 练习不同类型的题目,增强问题解决的广度和深度。 - 定期回顾和复习已解决的问题,巩固知识点。 - 参加LeetCode的比赛,锻炼在时间压力下的编程能力。 6. 关键标签“系统开源”: - 探索LeetCode的源代码,了解其后端架构和前端界面是如何实现的。 - 了解开源社区如何对LeetCode这样的平台贡献代码,以及如何修复bug和增强功能。 - 学习开源社区中代码共享的文化和最佳实践。 7. 压缩包子文件“leetcode-master”分析: - 该文件可能是一个版本控制工具(如Git)中的一个分支,包含了LeetCode习题答案的代码库。 - 用户可以下载此文件来查看不同用户的习题答案,分析不同解法的差异,从而提升自己的编程水平。 - “master”通常指的是主分支,意味着该分支包含了最新的、可以稳定部署的代码。 8. 使用LeetCode资源的建议: - 将LeetCode作为提升编程能力的工具,定期练习,尤其是对准备技术面试的求职者来说,LeetCode是提升面试技巧的有效工具。 - 分享和讨论自己的解题思路和代码,参与到开源社区中,获取更多的反馈和建议。 - 理解并吸收平台提供的习题答案,将其内化为自己解决问题的能力。 通过上述知识点的详细分析,可以更好地理解LeetCode习题答案的重要性和使用方式,以及在IT行业开源系统中获取资源和提升技能的方法。