• 大小: 2KB
    文件类型: .py
    金币: 1
    下载: 0 次
    发布日期: 2021-06-16
  • 语言: Python
  • 标签: tensor  

资源简介

tensorflow实现服装类别分类,包括模型训练、验证及保存,例子本身从tensorflow官网学习而来,亲自进行了实现,代码可用。环境搭建可参考个人博文。

资源截图

代码片段和文件信息

#!/usr/bin/env python 
# -*- coding:utf-8 -*-

# TensorFlow and tf.keras
import tensorflow as tf
from tensorflow import keras

# Helper libraries
import numpy as np
import matplotlib.pyplot as plt

fashion_mnist = keras.datasets.fashion_mnist

(train_images train_labels) (test_images test_labels) = fashion_mnist.load_data()

class_names = [‘T-shirt/top‘ ‘Trouser‘ ‘Pullover‘ ‘Dress‘ ‘Coat‘
               ‘Sandal‘ ‘Shirt‘ ‘Sneaker‘ ‘Bag‘ ‘Ankle boot‘]

train_images.shape;

train_labels_len = len(train_labels);
print(train_labels_len)
print(train_labels)

test_images.shape

test_labels_len = len(test_labels);
print(test_labels_len)
print(test_labels)

# plt.figure()
# plt.imshow(train_images[0])
# plt.colorbar()
# plt.grid(False)
# plt.show()
train_images = train_images / 255.0

test_images = test_images / 255.0

# plt.figure(figsize=(1010))
# for i in range(25):
#     plt.subplot(55i+1)
#     plt.xticks([])
#     plt.yticks([])
#     plt.grid(F

评论

共有 条评论