【機器學習No.1】Tensorflow智慧之別衣帽種類模型

Bonstoppo發表於2019-03-26

這是第一個專案,我打算一步步更新這個東西。

我的Git Repository:Fashion_mnist

FashionMNIST 是一個替代 MNIST 手寫數字集的影像資料集。 它是由 Zalando(一家德國的時尚科技公司)旗下的研究部門提供。其涵蓋了來自 10 種類別的共 7 萬個不同商品的正面圖片。

第一步:搭建環境:

#匯入tensorflow + keras
import tensorflow as tf
from tensorflow import keras
import numpy as np
import matplotlib.pyplot as plt

print("輸出當前TF的環境:" + tf.__version__)

第二步:匯入資料集:

在這一步中需要提醒的是,fashion_mnist資料集需要外網連結,沒有外網的話可以直接去下載資料集。

#獲得相應的資料集(首次執行會下載 , 但是似乎需要開啟代理 , 沒有開啟的話無法下載。)
fashion_mnist = keras.datasets.fashion_mnist

#獲得訓練集合,測試集合(大批量訓練,小批量訓練)
(train_images , train_labels) , (test_images , test_labels) = fashion_mnist.load_data()

#驗證訓練集合,測試集合的數量
print("訓練集合樣本數圖片寬 , 長:" + str(train_images.shape) + "測試集合樣本數 圖片寬 , 長:" + str(test_images.shape))

 

 

相關文章