
160
|
第
9
章
有关模型的元数据,例如用于创建模型的单词词典。为了有一个良好的开始,让我们先
探索一下 TensorFlow Lite Model Maker。
9.1
使用
TensorFlow Lite Model Maker
创建模型
TensorFlow Lite Model Maker 设计的一个核心场景是图像分类,它声称只需四行代码
就可以帮助你创建一个基本模型。此外,该模型将与 Android Studio 的导入功能一起使
用,因此你不必在资产文件夹中搞得乱七八糟,它会生成代码来帮助你入门。正如你将
在第 10 章看到的,使用 Model Maker 制作的模型使你的应用程序更容易使用,因为将
Android 数据表示转换为模型所需的张量的艰巨任务已被抽象为使用 Android Studio 导
入模型时自动生成的辅助类。不幸的是,在撰写本书时,Xcode 还没有等效的工具,因
此你需要使用自己编写的代码来完成大部分模型数据的输入 / 输出,但我们将通过一些
示例在第 11 章中给你一个良好的开始。
使用 Model Maker 创建模型非常简单。你使用 Python,而我已经为你创建了一个 Python
notebook,它可以在本书的代码库(
https://oreil.ly/9ImxO
)中找到。
首先安装
tflite-model-maker
包:
!pip install -q tflite-model-maker
完成后,你可以像这样在 TensorFlow Lite Model ...