1. 程式人生 > >《Gluon 動手學深度學習》顯示影象資料集Fashion-MNIST

《Gluon 動手學深度學習》顯示影象資料集Fashion-MNIST

%matplotlib inline
import sys
sys.path.append('..')
import gluonbook as gb
from mxnet.gluon import data as gdata
import time

mnist_train=gdata.vision.FashionMNIST(train = True)
mnist_test=gdata.vision.FashionMNIST(train = False)
len(mnist_train),len(mnist_test)

feature,label =mnist_train[0]
feature.shape,feature.dtype


# 本函式已儲存在 gluonbook 包中方便以後使用。
def get_fashion_mnist_labels(labels):
    text_labels = ['t-shirt', 'trouser', 'pullover', 'dress', 'coat',
                   'sandal', 'shirt', 'sneaker', 'bag', 'ankle boot']
    return [text_labels[int(i)] for i in labels]

# 本函式已儲存在 gluonbook 包中方便以後使用。
def show_fashion_mnist(images, labels):
    gb.use_svg_display()
    # 這裡的 _ 表示我們忽略(不使用)的變數。
    _, figs = gb.plt.subplots(1, len(images), figsize=(12, 12))
    for f, img, lbl in zip(figs, images, labels):
        f.imshow(img.reshape((28, 28)).asnumpy())
        f.set_title(lbl)
        f.axes.get_xaxis().set_visible(False)
        f.axes.get_yaxis().set_visible(False)
        #print(lbl)
        #print(img.T)

        
X, y = mnist_train[0:10]
show_fashion_mnist(X, get_fashion_mnist_labels(y))

結果為:

使用gluonbook顯示影象非常方便快捷,主要涉及到以下幾個方面:

顯示影象的模組:%matplotlib inline

顯示影象的函式:show_fashion_mnist()

顯示影象標籤及影象每個畫素值:        

print(lbl)
print(img.T)