1.4 安装 TensorFlow.js 的相关包
TensorFlow.js可以在浏览器和 Node.js 中运行,本节将重点介绍如何在浏览器中使用TensorFlow.js。在基于浏览器的项目中获取TensorFlow.js主要有两种方法。第一种是使用脚本标签,第二种是通过npm安装并且使用Parcel构建工具。
1.使用<script/>标签
我们可以将代码清单1-9所示脚本标签添加到项目的主HTML文件中。
代码清单1-9
<scriptsrc="https://cdn.jsdelivr.net/npm/@tensorflow/tfjs@1.0.0/dist/tf.min.js"></script>
添加完上述脚本标签后,我们就可以在浏览器中使用 TensorFlow.js 库了。该方法最大的缺点是当我们在代码编辑器中使用相关API时没有智能提示。
2.使用npm包
相对于通过<script/>标签引入的方法,我们更推荐通过 npm 包进行安装并使用 Parcel构建工具。
首先,创建一个项目文件夹,并在该文件夹中创建主页面文件index.html和脚本文件script.js。然后,在index.html文件中通过<script/>标签引入script.js,如图1-33所示。
接下来,我们在项目文件夹的根目录下通过代码清单 1-10 所示命令完成项目的初始化。
代码清单1-10
npminit
完成初始化工作后,继续通过代码清单1-11所示命令安装TensorFlow.js包。
代码清单1-11
npminstall@tensorflow/tfjs
接着,通过代码清单1-12所示命令安装Parcel构建工具。
代码清单1-12
npminstall-gparcel-bundler
安装完成后,我们就可以打包该Web项目了。Parcel中内置了一个当文件内容发生改变时
能够重新构建应用的开发服务器。为了实现快速开发,该开发服务器支持热模块替换,因此我
们只需指出入口文件(Parcel中所有指令的入口可以是一个或多个文件,一个或多个通配符表
达式,一个或多个目录)即可,如代码清单1-13所示。
代码清单1-13
parceldemo/*html
接下来,我们在浏览器中打开 http://localhost:1234,弹出的启动页面如图 1-34 所示。
我们可以在http://localhost:1234对应的启动页面中查看相应的效果。关于Parcel的使用,我们不再赘述。最后,我们介绍如何通过npm包安装TensorFlow.js的相关模块。
1.TensorFlow.js Core
TensorFlow.js Core模块主要提供与数值计算和硬件加速相关的低阶API,是TensorFlow.js生态系统的一部分,可以通过代码清单1-14所示命令完成安装。
代码清单1-14
npminstall@tensorflow/tfjs-core
2.TensorFlow.js Layers
TensorFlow.js Layers模块以TensorFlow.js Core为基础,是用来在浏览器中构建、训练并运行深度学习模型的高阶API,属于TensorFlow.js生态系统的一部分,可以通过代码清单1-15所示命令完成安装。
代码清单1-15
npminstall@tensorflow/tfjs-layers
3.TensorFlow.js Data
TensorFlow.js Data模块用于加载并解析磁盘或互联网上各种格式的数据,可以通过代码清单1-16所示命令完成安装。
代码清单1-16
npminstall@tensorflow/tfjs-data
4.TensorFlow.js Vis
TensorFlow.js Vis模块能够将TensorFlow.js模型进行可视化,可以通过代码清单1-17所示命令完成安装。
代码清单1-17
npminstall@tensorflow/tfjs-vis
5.TensorFlow.js AutoML
TensorFlow.js AutoML模块提供一系列用于加载和运行AutoMLEdge相关模型的API,可以通过代码清单1-18所示命令完成安装。
代码清单1-18
npminstall@tensorflow/tfjs-automl
6.TensorFlow.js Converter
TensorFlow.js Converter是一个开源包,用于将预训练的 TensorFlowSavedModel 或TensorFlow
Hub模块加载到浏览器中。注意,该库依赖Python 3.6.8版本。为了防止版本冲突,我们必须借助conda安装Python虚拟环境。关于conda工具的安装这里不再赘述。我们看一下如何在虚拟环境中安装tfjs-converter。
首先,打开终端工具,并输入代码清单1-19所示命令,开始虚拟环境的安装。
代码清单1-19
condacreate-ndemopython=3.6.8
安装完成后,通过代码清单1-20所示命令查看虚拟环境的相关信息,如图1-35所示。
代码清单1-20
condainfo--envs
接着,通过代码清单 1-21 所示命令进入对应的虚拟环境并查看虚拟环境中 Python 的版本 信息,如图 1-36 所示。
代码清单 1-21
condaactivatedemo
最后,通过代码清单 1-22 所示命令在虚拟环境中安装 tfjs-converter 包。安装细节如图 1-37 所示。
代码清单 1-22
pip install tensorflowjs
安装成功后,通过代码清单 1-23 所示命令查看版本信息,如图 1-38 所示。
代码清单 1-23
tensorflowjs_converter --version
至此,关于通过 npm 安装 TensorFlow.js 相关包的方法已经全部讲解完毕。