TensorFlow自己训练的SSD mobilenet模型 安卓移植

在做本项目之前，是一个非常非常小白的半吊子深度学习爱好者，但是一个月来接触了并主导了一个计算机视觉小项目，收获颇多。这篇文章详细介绍一下从0到apk的一个TensorFlow模型移植的demo。

不得不说，Google真是非常强大，尤其是TensorFlow和Android出自一家，使两者的迁移降低了很大难度。

由于之前从未接触过Android开发，因此下面主要分为以下几部分来实现：

1. 模型准备（生成.pb模型和.txt标签文件）

2. 安装Android studio（从0开始）

3. 导入模型并编译（很简单的操作==）

4. 迁移自己的模型（目前只实现了同类模型迁移）

下面开始详细介绍：

1.模型准备

模型只需要两个：一个是训练生成的模型.pb文件和标签文件.txt

2.安装Android studio

这个很麻烦啊，整个这个迁移过程做了大概一整天，但是一上午的时间就是安装这个ide了。我是参照各个前辈的教程上写的，主要步骤依次是：下载安装JDK和SDK（其实这两个都可以先不安装，我是参考别人的博客说是先安装这两个，但是其实在安装AS的时候可以再装。）JDK就是直接用AS推荐的嵌入式的就可以，SDK我是用的自己的，到时候把路径改到SDK安装的位置就行。

其他的貌似没有特别需要注意的了。但是创建第一个工程的时候需要等待一段时间，命令窗显示下载gradle的东西。

3.导入模型并编译

把TensorFlow的Android模型导入进去，记得把配置文件（build gradle）的相关版本改对，改成SDK的最高版本。如果导入成功了就可以插上安卓手机准备编译啦。但是如果不成功，就根据报错的日志信息去google，基本上都可以找到对应的改正方法，因为这时候的错误都是配置啊、版本啊这类问题。

编译的时候，有一个点很容易忽略，就是update AS，因为如果不是最新版本配置的化会报错！这个报错后的提示特别难发现是要更新的问题，我是多方尝试后无果最终在Stack Overflow上找到的解决方法----update，重启之后就好了啦，就可以run了。

run成功后，我们的手机上就可以发现同时安装了4个TFapp，分别是图片分类、检测、风格转换和语音转换。点开就可以测试啦。（还有个小问题需要注意，就是安卓版本的问题，因为高版本的SDK生成的apk可能在较低版本的安卓设备上不能安装）

4.迁移到自己的模型

这里因为我训练的是基于SSD_mobileNet的物体识别模型，因此是针对其中的检测模型修改。

其实若只是要在手机上安装观察效果，操作起来非常简单，把第一部分中准备的两个模型放到assets文件夹中，把detectactivity.java文件中的API模型路径改成自己的，因为这个检测模型优先ssdmobilenet，其后才是yolo和multibox算法。

特别注意一下，txt文件的第一行要是？？？，否则会报一个索引错误。

然后run就好啦。