tensorflow 下载模型

tensorflow 下载模型

六、下载模型 （可选）

andriod demo源码是不包含训练好的tensorflow模型的，但是在编译时需要模型。模型一共有三个，分别对应物体识别、行人检测和图像风格迁移。

在Bazel编译时，可以不用手动下载模型，因为在文件//tensorflow/examples/android/BUILD中设定了自动下载模型的操作。

但是如果使用Android Studio等其他方式编译，则不会运行BUILD文件中的自动下载操作。

手动下载模型zip文件：

$ curl -L .tensorflow/models/inception5h.zip -o /tmp/inception5h.zip$ curl -L .tensorflow/models/mobile_multibox_v1.zip -o /tmp/mobile_multibox_v1.zip$ curl -L .tensorflow/models/stylize_v1.zip -o /tmp/stylize_v1.zip
1
2
3
4
5

将上述三个zip文件分别解压到文件夹： 
//tensorflow/examples/android/assets

下载的模型链接随着　tf android 项目版本的更新可能会有变动，具体版本对应的链接请参考tensorflow根目录下的WORKSPACE文件，它其中标明了当前项目版本对应的模型下载链接。

解压完毕后，打开//tensoeflow/examples/android/BUILD文件，找到以 android_binary 开头的那段代码片，将 assets = [ ] 内以@开头的三句代码用#号 注释掉，如下图黄色部分所示

如果不注释掉该段内容，即使手动将模型放在assets文件中，使用bazel编译时也会重新下载模型。