TensorFlow入门10 -- 编程堆栈(programming stack)

参考:《深度学习图像识别技术--基于TensorFlow Object Detection API 和 OpenVINO》

TensorFlow提供了一个由多个API层组成的编程堆栈(programming stack),如下图所示:


TensorFlow入门10 -- 编程堆栈(programming stack)_第1张图片
TensorFlow Programming Stack

TensorFlow官网强烈推荐用下面两个APIs编写TensorFlow程序

Estimators, Estimator API 提供了完整的训练模型、评估模型精度和生成预测的方法(methods) 

Datasets, 构建数据输入管道(data input pipeline). Dataset API 负责载入数据、操作数据,然后把数据输入模型 。Dataset API 和 Estimators API 啮合的非常好。

获得范例代码

用命令(若没有装git,git安装和使用请参考《Git学习手记3:安装Git》):

git clone https://github.com/tensorflow/models, 如下


TensorFlow入门10 -- 编程堆栈(programming stack)_第2张图片

进入models/samples/core/get_started/ 文件夹下,运行premade_estimator.py


TensorFlow入门10 -- 编程堆栈(programming stack)_第3张图片

运行结果如下:


TensorFlow入门10 -- 编程堆栈(programming stack)_第4张图片

在Visual Studio Code中,打开文件premade_estimator.py,如下图所示:


TensorFlow入门10 -- 编程堆栈(programming stack)_第5张图片

可以看到该程序主要分为以下六个部分:

1,导入和解析数据集。

2,创建描述数据的特征列。

3,选择模型的类型。

4,训练模型。

5,评估模型的有效性。

6,让训练好的模型进行预测。

这六个部分也是大多数TensorFlow程序包含的必要部分

你可能感兴趣的:(TensorFlow入门10 -- 编程堆栈(programming stack))