PyTorch读取Cifar数据集并显示图片的实例讲解


Posted in Python onJuly 27, 2018

首先了解一下需要的几个类所在的package

PyTorch读取Cifar数据集并显示图片的实例讲解

from torchvision import transforms, datasets as ds
from torch.utils.data import DataLoader
import matplotlib.pyplot as plt
import numpy as np

#transform = transforms.Compose是把一系列图片操作组合起来,比如减去像素均值等。
#DataLoader读入的数据类型是PIL.Image
#这里对图片不做任何处理,仅仅是把PIL.Image转换为torch.FloatTensor,从而可以被pytorch计算
transform = transforms.Compose(
 [
 transforms.ToTensor()
 ]
)

Step 1,得到torch.utils.data.Dataset实例。

torch.utils.data.Dataset是一个抽象类,CIFAR100是它的一个实例化子类

train=True,读取训练集;train=False,读取测试集

download=False,不下载。如果为True,则先检查root下有无该数据集,如果没有就先下载。

train_set = ds.CIFAR100(root='.', train=True, transform=transform, target_transform=None, download=True)

Step 2,把Dataset封装成torch.utils.data.DataLoader

data_loader = DataLoader(dataset=train_set,
  batch_size=1,
  shuffle=False,
  num_workers=2)


# # 生成torch.utils.data.DataLoaderIter
# # 不过DataLoaderIter它会被DataLoader自动创建并且调用,我们用不到
# data_iter = iter(data_loader)
# images, labels = next(data_iter)

step 3,从DataLoader里读取数据,并将图片显示出来。

注意:

1)使用for...in...循环读取数据的时候,会自动调用DataLoader里的__next__()函数

而且只能对Tensor实例进行迭代,所以之前的transforms必须最后加一个transforms.ToTensor()

2)显示图片有两种方式:Image.show()和plt.imshow(ndarray)

Image.show():

通过transforms.ToPILImage()把FloatTensor转化为Image

plt.imshow(ndarray):

通过FloatTensor.numpy()转化为ndarray,再调用plt.imshow()

to_pil_image = transforms.ToPILImage()
cnt = 0
for image,label in data_loader:
 if cnt>=3: # 只显示3张图片
 break
 print(label) # 显示label

 # 方法1:Image.show()
 # transforms.ToPILImage()中有一句
 # npimg = np.transpose(pic.numpy(), (1, 2, 0))
 # 因此pic只能是3-D Tensor,所以要用image[0]消去batch那一维
 img = to_pil_image(image[0])
 img.show()

 # 方法2:plt.imshow(ndarray)
 img = image[0] # plt.imshow()只能接受3-D Tensor,所以也要用image[0]消去batch那一维
 img = img.numpy() # FloatTensor转为ndarray
 img = np.transpose(img, (1,2,0)) # 把channel那一维放到最后

 # 显示图片
 plt.imshow(img)
 plt.show()

 cnt += 1

另外补一句np.transpose()的用法。

第一个参数是要transpose的图片;

第二个是shape。比如一个ndarray是(channel, height, width),如果给第二个参数(height, width,channel),就会把第0维channel整个搬到最后。

以上这篇PyTorch读取Cifar数据集并显示图片的实例讲解就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
python发布模块的步骤分享
Feb 21 Python
Python搜索引擎实现原理和方法
Nov 27 Python
对django layer弹窗组件的使用详解
Aug 31 Python
python实现梯度下降法
Mar 24 Python
如何基于python实现不邻接植花
May 01 Python
Django中的AutoField字段使用
May 18 Python
Keras:Unet网络实现多类语义分割方式
Jun 11 Python
Pycharm常用快捷键总结及配置方法
Nov 14 Python
用ldap作为django后端用户登录验证的实现
Dec 07 Python
Python echarts实现数据可视化实例详解
Mar 03 Python
Python探索生命起源 matplotlib细胞自动机动画演示
Apr 21 Python
使用Python获取字典键对应值的方法
Apr 26 Python
pytorch 数据集图片显示方法
Jul 26 #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
You might like
yii去掉必填项中星号的方法
2015/12/28 PHP
php session实现多级目录存放实现代码
2016/02/03 PHP
php实现HTML实体编号与非ASCII字符串相互转换类实例
2016/11/02 PHP
js类 from qq
2006/11/13 Javascript
silverlight线程与基于事件驱动javascript引擎(实现轨迹回放功能)
2011/08/09 Javascript
javascript ready和load事件的区别示例介绍
2013/08/30 Javascript
跟我学习javascript的Date对象
2015/11/19 Javascript
Treegrid的动态加载实例代码
2016/04/29 Javascript
js实现上下左右弹框划出效果
2017/03/08 Javascript
详解Vue使用 vue-cli 搭建项目
2017/04/20 Javascript
jQuery实现百度登录框的动态切换效果
2017/04/21 jQuery
推荐10款扩展Web表单的JS插件
2017/12/25 Javascript
vue 右键菜单插件 简单、可扩展、样式自定义的右键菜单
2018/11/29 Javascript
JS/HTML5游戏常用算法之追踪算法实例详解
2018/12/12 Javascript
JS字符串常用操作方法实例小结
2019/06/24 Javascript
Vue项目结合Vue-layer实现弹框式编辑功能(实例代码)
2020/03/11 Javascript
Vue使用Element实现增删改查+打包的步骤
2020/11/25 Vue.js
[03:11]2014DOTA2国际邀请赛-VG掉入败者组 独家专访357
2014/07/19 DOTA
Python基于递归算法求最小公倍数和最大公约数示例
2018/07/27 Python
Python matplotlib生成图片背景透明的示例代码
2019/08/30 Python
Python监控服务器实用工具psutil使用解析
2019/12/19 Python
Python unittest单元测试openpyxl实现过程解析
2020/05/27 Python
一款简洁的纯css3代码实现的动画导航
2014/10/31 HTML / CSS
CSS3实现点击放大的动画实例代码
2017/02/27 HTML / CSS
梅西百货澳大利亚:Macy’s Australia
2017/07/26 全球购物
李宁官方网店:中国运动品牌
2017/11/02 全球购物
中国京东和泰国中央集团合资的网站:JD CENTRAL
2020/08/22 全球购物
锐步香港官方网上商店:Reebok香港
2020/11/05 全球购物
音乐学院硕士生的自我评价分享
2013/11/01 职场文书
纠纷协议书
2014/04/16 职场文书
走群众路线学习笔记
2014/11/06 职场文书
2015年团支部工作总结
2015/04/03 职场文书
会议通知格式范文
2015/04/15 职场文书
金正昆讲礼仪观后感
2015/06/11 职场文书
邹越演讲观后感
2015/06/15 职场文书
2019终止劳动合同协议书最新范本!
2019/07/09 职场文书