pytorch 数据集图片显示方法


Posted in Python onJuly 26, 2018

图片显示

pytorch 载入的数据集是元组tuple 形式,里面包括了数据及标签(train_data,label),其中的train_data数据可以转换为torch.Tensor形式,方便后面计算使用。

同样给一些刚入门的同学在使用载入的数据显示图片的时候带来一些难以理解的地方,这里主要是将Tensor与numpy转换的过程,理解了这些就可以就行转换了

CIAFA10数据集

首先载入数据集,这里做了一些数据处理,包括图片尺寸、数据归一化等

import torch
from torch.autograd import Variable 
import matplotlib.pyplot as plt 
import torchvision.datasets as dset
import torchvision.transforms as transforms
from autoencoder import AutoEncoder
import torch.nn as nn
import torchvision
import numpy as np
dataset = dset.CIFAR10(root='../train/data', download=True, 
    transform=transforms.Compose([
    transforms.Scale(200),
    transforms.ToTensor(),
    transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5)),
    transforms.Gray()
    ]))

在这里 dataset 是一个CIFAR10对象,(大家可以查看一下他的源代码)

方式一

dataset[1] = ([torch.FloatTensor of size 1x200x200],9)

载入的第二个数据是个tensor格式,包含一个标签 9

这里我们做的就是将torch.FloatTensor 转换为numpy,然后显示

b = dataset[1][0].numpy()
#取数据,不取标签

因为这里的b仍然是1*200*200的大小,所以要重新reshape一下,适合输出图像

plt.imshow(b.reshape(200,200),cmap = 'gray')
plt.show()

然后可以显示图像了

方式二

利用torch的接口

img = torchvision.utils.make_grid(dataset[1][0]).numpy()
plt.imshow(np.transpose(img,(1,2,0)))
plt.show()

这用np.transpose 是因为plt.imshow在显示 时候输入的是(imgsize,imgsieze,channels),而这里得到的img是(3,200,200)的格式,所以进行了转换,才能显示

以上这篇pytorch 数据集图片显示方法就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
python用10行代码实现对黄色图片的检测功能
Aug 10 Python
Python制作豆瓣图片的爬虫
Dec 28 Python
python如何通过twisted实现数据库异步插入
Mar 20 Python
python 删除非空文件夹的实例
Apr 26 Python
python ftp 按目录结构上传下载的实现代码
Sep 12 Python
解决python有时候import不了当前的包问题
Aug 28 Python
Python猴子补丁知识点总结
Jan 05 Python
python识别验证码图片实例详解
Feb 17 Python
Python流程控制语句的深入讲解
Jun 15 Python
python中return不返回值的问题解析
Jul 22 Python
PyCharm常用配置和常用插件(小结)
Feb 06 Python
Python还能这么玩之只用30行代码从excel提取个人值班表
Jun 05 Python
mac安装pytorch及系统的numpy更新方法
Jul 26 #Python
浅谈pytorch和Numpy的区别以及相互转换方法
Jul 26 #Python
pytorch: tensor类型的构建与相互转换实例
Jul 26 #Python
pytorch中tensor的合并与截取方法
Jul 26 #Python
Python爬虫框架Scrapy常用命令总结
Jul 26 #Python
Python退火算法在高次方程的应用
Jul 26 #Python
Python爬虫框架Scrapy基本用法入门教程
Jul 26 #Python
You might like
虚拟主机中对PHP的特殊设置
2006/10/09 PHP
php 将bmp图片转为jpg等其他任意格式的图片
2009/06/21 PHP
php生成略缩图代码
2012/07/16 PHP
使用php转义输出HTML到JavaScript
2015/03/27 PHP
摘自织梦CMS中的图片处理类
2015/08/08 PHP
php简单解析mysqli查询结果的方法(2种方法)
2016/06/29 PHP
thinkPHP5.1框架使用SemanticUI实现分页功能示例
2019/08/03 PHP
RSA实现JS前端加密与PHP后端解密功能示例
2019/08/05 PHP
window.location.hash知识汇总
2015/11/09 Javascript
form表单序列化详解(推荐)
2017/08/15 Javascript
Vue基于NUXT的SSR详解
2017/10/24 Javascript
快速解决brew安装特定版本flow的问题
2018/05/17 Javascript
JavaScript数组方法的错误使用例子
2018/09/13 Javascript
Vue中的Props(不可变状态)
2018/09/29 Javascript
vue中watch和computed为什么能监听到数据的改变以及不同之处
2019/12/27 Javascript
[04:11]DOTA2亚洲邀请赛小组赛第一日 TOP10精彩集锦
2015/01/30 DOTA
python模拟enum枚举类型的方法小结
2015/04/30 Python
Python生成随机验证码的两种方法
2015/12/22 Python
python3学习笔记之多进程分布式小例子
2018/02/13 Python
python 遍历目录(包括子目录)下所有文件的实例
2018/07/11 Python
jupyter lab的目录调整及设置默认浏览器为chrome的方法
2020/04/10 Python
Python字符串格式化常用手段及注意事项
2020/06/17 Python
python怎么删除缓存文件
2020/07/19 Python
最新PyCharm从安装到PyCharm永久激活再到PyCharm官方中文汉化详细教程
2020/11/17 Python
几个CSS3的flex弹性盒模型布局的简单例子演示
2016/05/12 HTML / CSS
Aveda美国官网:天然护发产品、洗发水、护发素和沙龙
2016/12/09 全球购物
美国领先的家庭健康检测试剂盒提供商:LetsGetChecked
2019/03/18 全球购物
施华洛世奇波兰官网:SWAROVSKI波兰
2019/06/18 全球购物
群众路线党课主持词
2014/04/01 职场文书
《池塘边的叫声》教学反思
2014/04/12 职场文书
给妈妈洗脚活动方案
2014/08/16 职场文书
医德医风个人工作总结2014
2014/11/14 职场文书
企业内部管理控制:银行存款控制制度范本
2020/01/10 职场文书
Python-OpenCV实现图像缺陷检测的实例
2021/06/11 Python
Python可视化学习之seaborn调色盘
2022/02/24 Python
html,css,javascript是怎样变成页面的
2023/05/07 HTML / CSS