Pytorch 小知识点汇总一 -- save_image, Image对象size, dataloader之getitem返回值, VScode 报错’already started‘

最新推荐文章于 2024-07-10 20:59:51 发布

xiangyong58

最新推荐文章于 2024-07-10 20:59:51 发布

阅读量5.3k

点赞数 2

分类专栏： Machine & Deep Learning 文章标签： pytorch

本文链接：https://blog.csdn.net/xiangyong58/article/details/104379353

版权

Machine & Deep Learning 专栏收录该内容

75 篇文章 132 订阅 ¥9.90 ¥99.00

订阅专栏

超级会员免费看

这篇博客总结了Pytorch使用中的常见问题，包括torchvision.utils.save_image报TypeError，如何保存图片，理解图片尺寸，save_image参数，dataloader的getitem返回值限制，以及在VScode中调试时遇到的'already started'错误及其解决方案。

摘要由CSDN通过智能技术生成

一、torchvision.utils.save_image 报错TypeError: Cannot handle this data type

1. img如果是uint16的矩阵而不转为uint8，Image.fromarray这句会报错。所以加上np.uint8(img)很有必要

rgb_image = rgb_image.astype(np.uint8)

2. 在pytorch中tensor默认是CHW，而PIL中是HWC. 顾img需是CHW形式。

 rgb_image = rgb_image.transpose((2,0,1))

二、保存生成图片的两个方式：

1. opencv方法：

cv2.imwrite('./temp/imgIn1.png', image_temp0)

2. torchvision.utils.save_image 方法：要特别注意除255.0 即/255.0

torchvision.utils.save_image(torch.from_numpy(image_temp0).float()/255.0, './temp/0image_'+str(index_temp)+'.jpg&#

了解本专栏

订阅专栏解锁全文

超级会员免费看

确定要放弃本次机会？

福利倒计时

: :

立减 ¥

普通VIP年卡可用

立即使用

xiangyong58

关注关注

2
点赞
踩
4

收藏

觉得还不错? 一键收藏
打赏
1
评论
复制链接

分享到 QQ

分享到新浪微博

扫一扫

专栏目录

订阅专栏

【笔记】torchvision.utils.save_image（）：单通道灰度图的保存（大坑）

nyist_yangguang的博客

08-26

7945

pytorch的torchvision无法保存真正的灰度图像在图像相关的深度学习模型中，有时候需要保存训练中的图像或者测试得到的图像（尤其是低级视觉任务比如去噪、超分辨率、压缩重建等），一般使用如下方式进行图像保存(torchvision.utils中的save_image()函数）： torchvision.utils.save_image(output.data,'%d.bmp'% (idx), padding=0) 但这种方式只能保存RGB彩色图像，如果网络的输出是单通道灰度图...

pytorch修改图片尺寸大小

qq_45868384的博客

04-02

6686

import cv2 from torch.utils.data import Dataset from PIL import Image import os import matplotlib.pyplot as plt class MyData(Dataset): # 初始化 root_dir大的路径和label_dir具体内容/获取地址 def __init__(self, root_dir, label_dir): # 创建全局变量 self...

1 条评论您还未登录，请先登录后发表或查看评论

Pytorch自定义Dataset，及__getitem__()访问

qq_43949166的博客

07-10

289

首先，实现继承自的自定义类，并重写方法。"""初始化自定义数据集。Args:data (list): 数据列表。""""""返回数据集的大小。Returns:int: 数据集的大小。""""""根据索引获取数据项。Args:idx (int): 数据项的索引。Returns:data: 索引对应的数据项。"""实现自定义数据集类，并重写方法。通过实例化自定义数据集类并使用索引访问数据项。通常使用语法来调用方法，而不是直接调用。可以使用DataLoader。

Pytorch中的Dataloader返回哪些东西？

开飞机的小毛驴儿

01-28

1万+

参考https://zhuanlan.zhihu.com/p/30934236，https://blog.csdn.net/jzwong/article/details/108867297?ops_request_misc=%25257B%252522request%25255Fid%252522%25253A%252522161179811616780269879401%252522%25252C%252522scm%252522%25253A%25252220140713.130102334.pc%25

torchvision存tensor为图片时报错TypeError: Cannot handle this data type

Archer_northern的博客

11-04

2704

博主在玩retro上的atari小游戏时，想存一下observation的图片，代码如下： obs_ = torch.Tensor(obs) print(obs_.shape) vutils.save_image(obs_, 'kan.png') 张量维度为(250, 160, 3) 结果直接报错 TypeError: Cannot handle this data type (1, 1, 250) 查阅资料后尝试n种解决方法无果，直接深入源码查看：回溯到倒数第二层： t

图片保存：torchvision.utils.save_image(img, imgPath)

weixin_43723625的博客

08-24

2万+

torchvision.utils.save_image(img, imgPath) 深度学习模型中，一般使用如下方式进行图像保存(torchvision.utils中的save_image()函数），这种方式只能保存RGB彩色图像，如果网络的输出是单通道灰度图像，则该函数依然会输出三个通道，每个通道的数值都是相同的，即“伪灰度图像”，虽然从视觉效果上看不出区别，但是图像所占内存比正常情况大了两倍。 save_image()函数： def save_image(tensor, filename, nro

a-PyTorch-Tutorial-to-Image-Captioning-master_pytorch_

09-30

**PyTorch简介** PyTorch是Facebook开源的一个深度学习框架，它以其灵活性和易用性受到广大开发者和研究人员的喜爱。PyTorch的核心是动态计算图，这使得它在构建复杂的神经网络结构时更加直观和便捷。本教程将引导...

a-PyTorch-Tutorial-to-Image-Captioning:显示，参加和讲述| PyTorch教程以进行图像字幕

05-07

假定具备PyTorch，卷积和递归神经网络的基本知识。如果您是PyTorch的新手，请先阅读PyTorch的和。问题，建议或更正可以作为问题发布。我在Python 3.6使用PyTorch 0.4 。 2020年1月27日：添加了两个新教程的...

Image-to-Image_Translation_in_PyTorch_pytorch-

热门推荐

风口IT猪的成长录

07-27

22万+

Save Image As Type 2.0-crx插件

03-25

语言:English 通过图像上的上下文菜单将图像另存为PNG，JPG或WebP。允许您通过右键单击图像来专门通过PNG，JPG，PDF或WebP格式保存图像。别忘了留下5星级评论！！这确实可以帮助我们！ :star:将图像保存为所需的文件类型！ :star:另外，如果扩展名不起作用或某些功能不起作用，请转到“支持”选项卡，然后在其中写下问题。如果可以轻松解决某些问题，请不要留下负面评价。我们的团队一直在监视审阅和支持选项卡，因此我们会尽快做出回应。如果您对扩展名有任何建议，请告诉我们，我们将考虑添加！！您也可以通过页面右侧“开发人员”部分下的“联系开发人员”超链接向我们发送电子邮件！易于使用：1.）只需转到要保存到桌面的图像即可。 2.）右键单击图像上。 3.）转到“将图像另存为类型2.0。4.）选择要保存图像的文件类型（PNG，JPG或WebP）功能：1-易于使用。 2-超级安全。 3-保存图像为PNG，JPG，PDF，或WebP的在四种格式安装图片另存新文件类型2.0和保存图像，我们目前拥有可利用的今天！

Save image as Type-crx插件

04-04

语言:English 将图像另存为PNG，JPG。将图像另存为类型

torchvision.utils.save_image()保存tensor显示图片异常问题解决

hhb442的博客

06-22

1751

刚开始以为是图像数据分布范围的问题，在保存之前输出图像tensor的最大max和最小min值，出现了 -0.0x和1.0x的数值，说明图像的像素范围超出了0-1。可是通过读utils.save_image()的源码发现，就算超出0-1也不应该出现这种问题，源码中存在如下部分代码。图片和保存路径传入给 save_image（）函数就行，不会出现问题。而我在保存之前进行了降维处理，降成了三维（squeeze(0)是降维）有些像素点会显示为全黑（灰度图），如下图所示，第一张和第三张图。

使用torchvision.utils.save_image()将张量保存为单通道灰度图

hhb442的博客

06-07

2927

在这个示例中，我们首先创建了一个假设为128x128的灰度图像张量 gray_image，其形状为 [1, 1, 128, 128]。然后，我们通过将张量的值缩放到0到255的范围（整数）并限制在0到255之间，将其转换为合适的灰度图像格式。最后，我们使用utils.save_image函数将灰度图像保存为PNG文件，命名为gray_image.png。确保在运行代码之前已经安装了torchvision库，该库提供了utils.save_image函数来方便地保存图像。以下是保存灰度图像的示例代码。

【pytorch】带batch的tensor类型图像显示

XavierJ的博客

11-22

3263

项目场景 pytorch训练时我们一般把数据集放到数据加载器里，然后分批拿出来训练。训练前我们一般还要看一下训练数据长啥样，也就是训练数据集可视化。那么如何显示dataloader里面带batch的tensor类型的图像呢？显示图像绘图最常用的库就是matplotlib： pip install matplotlib 显示图像会用到matplotlib.pyplot.imshow方法。查阅官方文档可知，该方法接收的图像的通道数要放到后面：数据加载器中数据的维度是[B, C, H, W]，我们每次只

torchvision.utils.save_image 讲解

aaatomaaa的博客

03-05

1019

torchvision.utils.save_image

【Pytorch API笔记8】用torchvision.utils.save_image批量保存图像数据

hjxu2016的博客

04-04

763

将给定的Tensor保存成image文件。如果给定的是mini-batch tensor，那就用make-grid做成雪碧图，再保存。将一个Batch内的数据，都保存下来，省的一个一个保存了。

pytorch拼接与保存图片

qq_40600539的博客

01-15

6803

在深度学习训练时，我们常需要对某层的输出或最终模型的预测结果进行输出查看，如featmap、output、predict结果等。在pytorch的设定中有tensor类型变量，不可直接查看和保存图片。虽然有torchvision.utils.save_image 方法可以方便地保存tensor类型图片，调用方法如下： from torchvision.utils import save...

06-06

PyTorch是一个流行的深度学习框架，它具有易于使用的API和灵活性。安装PyTorch可以使您在计算机上运行深度学习模型。本文将介绍如何在Windows和Ubuntu上安装PyTorch最新版的详细步骤。对于Windows用户，首先需要...