快捷導(dǎo)航

pytorch實(shí)現(xiàn)mnist數(shù)據(jù)集的圖像可視化及保存

更新時間：2020年01月14日 09:40:51 作者：樸素.無恙

今天小編就為大家分享一篇pytorch實(shí)現(xiàn)mnist數(shù)據(jù)集的圖像可視化及保存，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧

如何將pytorch中mnist數(shù)據(jù)集的圖像可視化及保存

導(dǎo)出一些庫

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

數(shù)據(jù)集的準(zhǔn)備

#訓(xùn)練集測試集的準(zhǔn)備

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)

將訓(xùn)練及測試集利用dataloader進(jìn)行迭代

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])#打印出標(biāo)簽
 scipy.misc.toimage(image_array,cmin=0.0,cmax=1.0).save(filename)#保存圖像

以上這篇pytorch實(shí)現(xiàn)mnist數(shù)據(jù)集的圖像可視化及保存就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。

您可能感興趣的文章:

相關(guān)文章

壓縮包密碼破解示例分享(類似典破解)
有一個壓縮包密碼忘了，寫了一個小腳本實(shí)現(xiàn)一個解密的功能,輸入自己常用密碼中的單詞后，腳本將這些密碼組合嘗試解壓壓縮包
2014-01-01
Python中threading模塊join函數(shù)用法實(shí)例分析
這篇文章主要介紹了Python中threading模塊join函數(shù)用法,以實(shí)例形式較為詳細(xì)的分析了join函數(shù)的功能與使用方法,需要的朋友可以參考下
2015-06-06
python實(shí)現(xiàn)根據(jù)文件格式分類
這篇文章主要為大家詳細(xì)介紹了python實(shí)現(xiàn)根據(jù)文件格式分類，文中示例代碼介紹的非常詳細(xì)，具有一定的參考價值，感興趣的小伙伴們可以參考一下
2019-10-10
python中print格式化輸出的問題
所謂格式化輸出，就是創(chuàng)建一個可以嵌入變量內(nèi)容的字符串。這篇文章主要介紹了python中print格式化輸出,需要的朋友可以參考下
2021-04-04
Python基于隨機(jī)采樣一至性實(shí)現(xiàn)擬合橢圓(優(yōu)化版)
這篇文章主要對上一版的Python基于隨機(jī)采樣一至性實(shí)現(xiàn)擬合橢圓的優(yōu)化，文中的示例代碼講解詳細(xì)，具有一定的借鑒價值，感興趣的可以了解一下
2022-11-11
[項目布局配置]Nosql與PythonWeb-Flask框架組合
本文主要描述了怎樣使用輕量級NoSQL數(shù)據(jù)庫與PythonWeb-Flask框架組合來進(jìn)行項目布局及應(yīng)用配置,需要的同學(xué)可以參考下，希望可以對大家有所進(jìn)益
2021-08-08
Pycharm編輯器技巧之自動導(dǎo)入模塊詳解
我們在編程過程中經(jīng)常會不經(jīng)意的使用到一些尚未導(dǎo)入的類和模塊，在這種情況下Pycharm會幫助我們定位模塊文件位置并將其添加到導(dǎo)入列表中，這也就是所謂的自動導(dǎo)入模塊功能。本文給大家介紹了關(guān)于Pycharm編輯器技巧之自動導(dǎo)入模塊的相關(guān)資料,需要的朋友可以參考下。
2017-07-07
用Python Flask創(chuàng)建簡潔高效的URL短鏈接服務(wù)
本文介紹了如何使用Python Flask框架創(chuàng)建URL短鏈接服務(wù)。通過詳細(xì)的步驟和代碼示例，讀者將學(xué)會如何搭建一個高效的URL縮短服務(wù)，包括生成短鏈接、重定向、還原長鏈接等功能。本文還介紹了如何使用Redis數(shù)據(jù)庫實(shí)現(xiàn)短鏈接的存儲和管理和如何優(yōu)化短鏈接的訪問速度和可靠性
2023-04-04
python實(shí)現(xiàn)簡單日期工具類
這篇文章主要為大家詳細(xì)介紹了python實(shí)現(xiàn)簡單日期工具類，具有一定的參考價值，感興趣的小伙伴們可以參考一下
2019-04-04
從零學(xué)Python之引用和類屬性的初步理解
Python是一種解釋型、面向?qū)ο?、動態(tài)數(shù)據(jù)類型的高級程序設(shè)計語言。自從20世紀(jì)90年代初Python語言誕生至今，它逐漸被廣泛應(yīng)用于處理系統(tǒng)管理任務(wù)和Web編程。Python已經(jīng)成為最受歡迎的程序設(shè)計語言之一。2011年1月，它被TIOBE編程語言排行榜評為2010年度語言。自從2004年以后，python的使用率是呈線性增長。
2014-05-05