Quantcast
Channel: InfoQ - 促进软件开发领域知识与创新的传播
Viewing all articles
Browse latest Browse all 1638

TensorFlow工程实战(二):用tf.layers API在动态图上识别手写数字

$
0
0

通过使用卷积网络在MNIST数据集上进行识别任务的实例,演示如何用tf.layers API构建模型,并在动态图中进行训练。

本文摘选自电子工业出版社出版、李金洪编著的《深度学习之TensorFlow工程化项目实战》一书的实例21:用tf.layers API在动态图上识别手写数字。

实例描述

有一组手写数字图片,要求用tf.layers API在动态图上搭建模型,将其识别出来。

一、启动动态图并加载手写图片数据集

本例加载TFDS模块中集成好的MNIST数据集。该数据集常用于验证模型的功能性实验中。

用tf.enable_eager_execution函数启动动态图,并加载MNIST数据集。具体代码如下:


Viewing all articles
Browse latest Browse all 1638

Trending Articles