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 相关文章推荐
python列表与元组详解实例
Nov 01 Python
python实现的jpg格式图片修复代码
Apr 21 Python
Python fileinput模块使用实例
May 28 Python
Python正则表达式完全指南
May 25 Python
python记录程序运行时间的三种方法
Jul 14 Python
python3实现名片管理系统
Nov 29 Python
python 监听salt job状态,并任务数据推送到redis中的方法
Jan 14 Python
浅谈Django+Gunicorn+Nginx部署之路
Sep 11 Python
python实现操作文件(文件夹)
Oct 31 Python
python多线程使用方法实例详解
Dec 30 Python
Python连接Postgres/Mysql/Mongo数据库基本操作大全
Jun 29 Python
Python first-order-model实现让照片动起来
Jun 25 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动态生成虚拟现实VRML网页
2006/10/09 PHP
php 字符转义 注意事项
2009/05/27 PHP
PHP单例模式详细介绍
2015/07/01 PHP
PHP判断密码强度的方法详解
2017/05/26 PHP
laravel-admin select框默认选中的方法
2019/10/03 PHP
JQuery 国际象棋棋盘 实现代码
2009/06/26 Javascript
jQuery LigerUI 使用教程表格篇(1)
2012/01/18 Javascript
jQuery实现MSN中文网滑动Tab菜单效果代码
2015/09/09 Javascript
jqGrid中文文档之选项设置
2015/12/02 Javascript
使用bootstrap实现多窗口和拖动效果
2016/09/22 Javascript
Javascript 引擎工作机制详解
2016/11/30 Javascript
如何提高数据访问速度
2016/12/26 Javascript
js仿iphone秒表功能 计算平均数
2017/01/11 Javascript
vue将单页面改造成多页面应用的方法
2018/11/25 Javascript
JavaScript两种计时器的实例讲解
2019/01/31 Javascript
ES5新增数组的实现方法
2020/05/12 Javascript
微信小程序调用wx.getImageInfo遇到的坑解决
2020/05/31 Javascript
vue 子组件修改data或调用操作
2020/08/07 Javascript
详解webpack的文件监听实现(热更新)
2020/09/11 Javascript
通过代码实例展示Python中列表生成式的用法
2015/03/31 Python
浅谈用Python实现一个大数据搜索引擎
2017/11/28 Python
使用python将时间转换为指定的格式方法
2018/11/12 Python
python如何实现数据的线性拟合
2019/07/19 Python
PIL对上传到Django的图片进行处理并保存的实例
2019/08/07 Python
Python3操作Excel文件(读写)的简单实例
2019/09/02 Python
浅谈Django中的QueryDict元素为数组的坑
2020/03/31 Python
Python selenium自动化测试模型图解
2020/04/15 Python
Python matplotlib读取excel数据并用for循环画多个子图subplot操作
2020/07/14 Python
Python getsizeof()和getsize()区分详解
2020/11/20 Python
浅谈CSS3 动画卡顿解决方案
2019/01/02 HTML / CSS
施华洛世奇匈牙利官网:SWAROVSKI匈牙利
2019/07/06 全球购物
C/C++有关内存的思考题
2015/12/04 面试题
高一生物教学反思
2014/01/17 职场文书
北京英文导游词
2015/02/12 职场文书
大二学年个人总结
2015/03/03 职场文书
2015年秋季学校开学标语
2015/07/16 职场文书