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多线程编程(四):使用Lock互斥锁
Apr 05 Python
Python语言实现机器学习的K-近邻算法
Jun 11 Python
关于Python面向对象编程的知识点总结
Feb 14 Python
Python中装饰器兼容加括号和不加括号的写法详解
Jul 05 Python
python中的for循环
Sep 28 Python
浅析Python函数式编程
Oct 06 Python
python 解压pkl文件的方法
Oct 25 Python
python打包生成的exe文件运行时提示缺少模块的解决方法
Oct 31 Python
Django之模型层多表操作的实现
Jan 08 Python
python爬虫爬取笔趣网小说网站过程图解
Nov 18 Python
Python实现数值积分方式
Nov 20 Python
python实现输入三角形边长自动作图求面积案例
Apr 12 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 URL参数获取方式的四种例子
2014/02/28 PHP
Zend Framework教程之请求对象的封装Zend_Controller_Request实例详解
2016/03/07 PHP
php实现简单爬虫的开发
2016/03/28 PHP
PHP模板引擎Smarty内建函数详解
2016/04/11 PHP
PHP递归遍历指定文件夹内的文件实现方法
2016/11/15 PHP
用PHP将Unicode 转化为UTF-8的实现方法(推荐)
2017/02/08 PHP
Jquery对数组的操作技巧整理
2014/03/25 Javascript
学习Angular中作用域需要注意的坑
2016/08/17 Javascript
js获取ip和地区
2017/03/10 Javascript
angular中实现控制器之间传递参数的方式
2017/04/24 Javascript
微信小程序商品到详情的实现
2017/06/27 Javascript
jQuery中的$是什么意思及 $. 和 $().的区别
2018/04/20 jQuery
JS实现table表格内针对某列内容进行即时搜索筛选功能
2018/05/11 Javascript
arctext.js实现文字平滑弯曲弧形效果的插件
2019/05/13 Javascript
[43:24]完美世界DOTA2联赛PWL S3 INK ICE vs DLG 第二场 12.12
2020/12/17 DOTA
Python XML RPC服务器端和客户端实例
2014/11/22 Python
Python表示矩阵的方法分析
2017/05/26 Python
Python sorted函数详解(高级篇)
2018/09/18 Python
python实现五子棋小游戏
2020/03/25 Python
python制作简单五子棋游戏
2019/06/18 Python
使用Python给头像戴上圣诞帽的图像操作过程解析
2019/09/20 Python
TensorFlow查看输入节点和输出节点名称方式
2020/01/04 Python
浅谈html5增强的页面元素
2016/06/14 HTML / CSS
澳大利亚墨水站Ink Station:墨水和碳粉打印机墨盒
2019/03/24 全球购物
DNA基因检测和分析:23andMe
2019/05/01 全球购物
英国时尚配饰、珠宝和服装网站:KJ Beckett
2020/01/23 全球购物
乌克兰的第一家手表店:Deka
2020/03/05 全球购物
2014村务公开实施方案
2014/02/25 职场文书
授权委托书(完整版)
2014/09/10 职场文书
2014年出纳工作总结与计划
2014/12/09 职场文书
主持稿开场白
2015/06/01 职场文书
焦裕禄纪念馆观后感
2015/06/09 职场文书
体育教师教学随笔
2015/08/15 职场文书
村官2015年度工作总结
2015/10/14 职场文书
导游词之台湾安平古堡
2019/12/25 职场文书
python playwrigh框架入门安装使用
2022/07/23 Python