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实现端口转发器的方法
Mar 13 Python
Python实现数通设备端口使用情况监控实例
Jul 15 Python
Python微信库:itchat的用法详解
Aug 14 Python
python中通过预先编译正则表达式提高效率
Sep 25 Python
基于pandas数据样本行列选取的方法
Apr 20 Python
pip安装py_zipkin时提示的SSL问题对应
Dec 29 Python
Python多进程入门、分布式进程数据共享实例详解
Jun 03 Python
Django 创建/删除用户的示例代码
Jul 24 Python
Python3 中作为一等对象的函数解析
Dec 11 Python
opencv 图像加法与图像融合的实现代码
Jul 08 Python
浅谈Python里面None True False之间的区别
Jul 09 Python
python turtle绘图命令及案例
Nov 23 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数据库连接类~~做成一个分页类!
2006/11/25 PHP
php empty,isset,is_null判断比较(差异与异同)
2010/10/19 PHP
打造超酷的PHP数据饼图效果实现代码
2011/11/23 PHP
php实现图片局部打马赛克的方法
2015/02/11 PHP
php的闭包(Closure)匿名函数初探
2016/02/14 PHP
深入理解Yii2.0乐观锁与悲观锁的原理与使用
2017/07/26 PHP
小程序微信支付功能配置方法示例详解【基于thinkPHP】
2019/05/05 PHP
Document 对象的常用方法
2009/07/31 Javascript
jQuery实现简单倒计时功能的方法
2016/07/04 Javascript
Vue 2.X的状态管理vuex记录详解
2017/03/23 Javascript
underscore之Chaining_动力节点Java学院整理
2017/07/10 Javascript
BootStrap中Table隐藏后显示问题的实现代码
2017/08/31 Javascript
浅谈React Native Flexbox布局(小结)
2018/01/08 Javascript
JS插入排序简单理解与实现方法分析
2019/11/25 Javascript
vue+koa2搭建mock数据环境的详细教程
2020/05/18 Javascript
Vue.js中使用Vuex实现组件数据共享案例
2020/07/31 Javascript
说一说Python logging
2016/04/15 Python
用python写一个windows下的定时关机脚本(推荐)
2017/03/21 Python
Django如何配置mysql数据库
2018/05/04 Python
Python实现拷贝/删除文件夹的方法详解
2018/08/29 Python
在Python中给Nan值更改为0的方法
2018/10/30 Python
python3.4爬虫demo
2019/01/22 Python
Python中psutil的介绍与用法
2019/05/02 Python
Django错误:TypeError at / 'bool' object is not callable解决
2019/08/16 Python
python3 assert 断言的使用详解 (区别于python2)
2019/11/27 Python
Python pytesseract验证码识别库用法解析
2020/06/29 Python
python 爬取免费简历模板网站的示例
2020/09/27 Python
web页面录屏实现
2019/02/12 HTML / CSS
Html5之webcoekt播放JPEG图片流
2020/09/22 HTML / CSS
Shopee马来西亚:随拍即卖,最佳行动电商拍卖平台
2017/06/05 全球购物
手机促销活动方案
2014/02/05 职场文书
满月酒主持词
2014/03/27 职场文书
刑事案件上诉状
2015/05/23 职场文书
Django debug为True时,css加载失败的解决方案
2021/04/24 Python
解决vue $http的get和post请求跨域问题
2021/06/07 Vue.js
Python Pytorch查询图像的特征从集合或数据库中查找图像
2022/04/09 Python