
pytorch/研究
pytorch
Peanutty
菜鸡一枚
展开
-
torchvision.utils.save_image()保存tensor显示图片异常问题解决
刚开始以为是图像数据分布范围的问题,在保存之前输出图像tensor的最大max和最小min值,出现了 -0.0x和1.0x的数值,说明图像的像素范围超出了0-1。可是通过读utils.save_image()的源码发现,就算超出0-1也不应该出现这种问题,源码中存在如下部分代码。图片和保存路径传入给 save_image()函数就行,不会出现问题。而我在保存之前进行了降维处理,降成了三维(squeeze(0)是降维)有些像素点会显示为全黑(灰度图),如下图所示,第一张和第三张图。原创 2023-06-22 22:57:34 · 1992 阅读 · 1 评论 -
使用torchvision.utils.save_image()将张量 保存 为 单通道灰度图
在这个示例中,我们首先创建了一个假设为128x128的灰度图像张量 gray_image,其形状为 [1, 1, 128, 128]。然后,我们通过将张量的值缩放到0到255的范围(整数)并限制在0到255之间,将其转换为合适的灰度图像格式。最后,我们使用utils.save_image函数将灰度图像保存为PNG文件,命名为gray_image.png。确保在运行代码之前已经安装了torchvision库,该库提供了utils.save_image函数来方便地保存图像。以下是保存灰度图像的示例代码。原创 2023-06-07 11:23:36 · 3375 阅读 · 3 评论 -
将.mat格式的高光谱图像转换成png或jpg格式
在这个示例中,我们假设高光谱图像数据存储在名为“hyperspectral_image.mat”的.mat文件中,并使用了线性拉伸归一化方法将每个波段的数据归一化到0到1的范围内。我们选择了29、53、77三个波段的数据,将它们堆叠在一起,形成RGB图像数据。如果高光谱图像数据为三维矩阵,将其转置为(宽度,高度,波段数)的形式。这是因为MATLAB中高光谱图像通常是以(波段数,高度,宽度)的形式存储的,而我们需要的输出格式是以(宽度,高度,波段数)的形式存储的。根据需要指定输出文件名、图像质量等参数。原创 2023-02-12 01:58:38 · 4667 阅读 · 0 评论 -
matlab显示.mat格式多光谱图像
请注意,在上面的代码中,your_file_name.mat应替换为你要加载的.mat文件的名称。your_image_data应替换为包含多光谱图像数据的变量名。原创 2023-02-11 21:33:41 · 1021 阅读 · 0 评论 -
vscode+anaconda+pytorch报错ModuleNotFoundError: No module named ‘torch‘
在搭建pytorch环境时,已经在终端、pycharm中正常运行torch。但是在vscode中配置python path依然无法正常运行,提示vscode ModuleNotFoundError: No module named ‘torch’先禁用扩展code runner看看是否能正常运行测试代码。启用code runner能够正常运行,环境配置成功。能够正常运行,确定为code runner配置错误。进入code runner配置python地址。将python地址修改与环境地址一样即可。转载 2022-12-22 00:22:53 · 669 阅读 · 0 评论