pytorch实现mnist数据集的图像可视化及保存


Posted in Python onJanuary 14, 2020

如何将pytorch中mnist数据集的图像可视化及保存

导出一些库

import torch
import torchvision 
import torch.utils.data as Data 
import scipy.misc
import os
import matplotlib.pyplot as plt   
BATCH_SIZE = 50  
DOWNLOAD_MNIST = True

数据集的准备

#训练集测试集的准备

train_data = torchvision.datasets.MNIST(root='./mnist/', train=True,transform=torchvision.transforms.ToTensor(),              
  download=DOWNLOAD_MNIST, )
test_data = torchvision.datasets.MNIST(root='./mnist/', train=False)

将训练及测试集利用dataloader进行迭代

train_loader = Data.DataLoader(dataset=train_data, batch_size=BATCH_SIZE, shuffle=True)
test_x = Variable(torch.unsqueeze(test_data.test_data, dim=1), requires_grad=True).type(torch.FloatTensor)[:20]/255 
test_y = test_data.test_labels[:20]#前两千张
 #具体查看图像形式为:
 
a_data, a_label = train_data[0]
print(type(a_data))#tensor 类型
#print(a_data)
print(a_label)

#把原始图片保存至MNIST_data/raw/下
save_dir="mnist/raw/"
if os.path.exists(save_dir)is False:
 os.makedirs(save_dir)
 
for i in range(20):
 image_array,_=train_data[i]#打印第i个
 image_array=image_array.resize(28,28)
 filename=save_dir + 'mnist_train_%d.jpg' % i#保存文件的格式
 print(filename)
 print(train_data.train_labels[i])#打印出标签
 scipy.misc.toimage(image_array,cmin=0.0,cmax=1.0).save(filename)#保存图像

pytorch实现mnist数据集的图像可视化及保存

pytorch实现mnist数据集的图像可视化及保存

以上这篇pytorch实现mnist数据集的图像可视化及保存就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
Django框架下在视图中使用模版的方法
Jul 16 Python
简单介绍Python的Django框架加载模版的方式
Jul 20 Python
Python操作csv文件实例详解
Jul 31 Python
Python创建二维数组实例(关于list的一个小坑)
Nov 07 Python
用十张图详解TensorFlow数据读取机制(附代码)
Feb 06 Python
Flask框架各种常见装饰器示例
Jul 17 Python
python+mysql实现教务管理系统
Feb 20 Python
pandas中read_csv的缺失值处理方式
Dec 19 Python
在Django中预防CSRF攻击的操作
Mar 13 Python
利用python实现后端写网页(flask框架)
Feb 28 Python
Python requests用法和django后台处理详解
Mar 19 Python
pytest实现多进程与多线程运行超好用的插件
Jul 15 Python
Pytorch在dataloader类中设置shuffle的随机数种子方式
Jan 14 #Python
python3.7通过thrift操作hbase的示例代码
Jan 14 #Python
解决pytorch DataLoader num_workers出现的问题
Jan 14 #Python
PyTorch实现ResNet50、ResNet101和ResNet152示例
Jan 14 #Python
python重要函数eval多种用法解析
Jan 14 #Python
关于ResNeXt网络的pytorch实现
Jan 14 #Python
Python属性和内建属性实例解析
Jan 14 #Python
You might like
PHP __autoload函数(自动载入类文件)的使用方法
2012/02/04 PHP
关于PHP session 存储方式的详细介绍
2013/06/25 PHP
PHP使用flock实现文件加锁的方法
2015/07/01 PHP
一张表搞清楚php is_null、empty、isset的区别
2015/07/07 PHP
Laravel框架基于ajax和layer.js实现无刷新删除功能示例
2019/01/17 PHP
基于jQuery的一个扩展form序列化到json对象
2010/12/09 Javascript
基于jquery的15款幻灯片插件
2011/04/10 Javascript
jquery foreach使用示例
2013/09/12 Javascript
IE6下javasc#ipt:void(0) 无效的解决方法
2013/12/23 Javascript
js鼠标滑轮滚动事件绑定的简单实例(兼容主流浏览器)
2014/01/14 Javascript
JS实现的一个简单的Autocomplete自动完成例子
2014/04/16 Javascript
node.js中的buffer.Buffer.isBuffer方法使用说明
2014/12/14 Javascript
js实现格式化金额,字符,时间的方法
2015/02/26 Javascript
js剪切板应用clipboardData实例解析
2016/05/29 Javascript
ES6中Iterator与for..of..遍历用法分析
2017/03/31 Javascript
js将键值对字符串转为json字符串的方法
2018/03/30 Javascript
详解Vue-cli中的静态资源管理(src/assets和static/的区别)
2018/06/19 Javascript
使用Object.defineProperty如何巧妙找到修改某个变量的准确代码位置
2018/11/02 Javascript
jquery实现动态改变css样式的方法分析
2019/05/27 jQuery
如何在vue项目中嵌入jsp页面的方法(2种)
2020/02/06 Javascript
springboot+vue+对接支付宝接口+二维码扫描支付功能(沙箱环境)
2020/10/15 Javascript
使用Py2Exe for Python3创建自己的exe程序示例
2018/10/31 Python
Django框架中序列化和反序列化的例子
2019/08/06 Python
python爬虫使用requests发送post请求示例详解
2020/08/05 Python
sklearn中的交叉验证的实现(Cross-Validation)
2021/02/22 Python
HTML5 本地存储和内容按需加载的思路和方法
2011/04/07 HTML / CSS
amazeui页面校验功能的实现代码
2020/08/24 HTML / CSS
巴西最大的玩具连锁店:Ri Happy
2020/06/17 全球购物
投标单位介绍信
2014/01/09 职场文书
保密工作责任书
2014/04/16 职场文书
大学生社会实践方案
2014/05/11 职场文书
文明家庭先进事迹材料
2014/05/14 职场文书
关于读书的演讲稿600字
2014/08/27 职场文书
师德师风事迹材料
2014/12/20 职场文书
淘宝客服专员岗位职责
2015/04/07 职场文书
MySQL自定义函数及触发器
2022/08/05 MySQL