5 Fashion数据集--衣裤识别

最新推荐文章于 2024-06-07 10:20:43 发布

CourageLee

最新推荐文章于 2024-06-07 10:20:43 发布

阅读量272

点赞数

分类专栏： tensorflow 文章标签： tensorflow

本文链接：https://blog.csdn.net/qq_35779738/article/details/110140365

版权

通过加入卷积层，训练集上的衣裤识别效果显著提高，测试集正确率达到近乎100%，但验证集正确率仍停留在90%。

摘要由CSDN通过智能技术生成

import  tensorflow as tf
from  tensorflow.keras.datasets import fashion_mnist
from tensorflow.keras.models import  Model
from tensorflow.keras.layers import  Flatten,Dense
import  matplotlib.pyplot as plt
fashion_data = fashion_mnist.load_data()
(x_train,y_train),(x_test,y_test) = fashion_data
x_train,x_test = x_train/255.0, x_test/255.0

class Mymodel(Model):
    def __init__(self):
        super(Mymodel, self).__init__()
        self.flatten = Flatten()
        self