之前写了android demo在手机上的运用,但是模型都是官方给的,想要替换成自己的模型,所以尝试了下将自己训练的yolo模型来替换demo给的模型。
首先,darknet的训练和.weight文件到.pb文件的转化,以及android demo的实现见之前的博客。在此不再叙述sdk,nkd等配置问题,且直接使用.pb文件。其次,默认已安装android studio。
(1)终端进入(android安装目录)/bin,输入./stuodio.sh
开启android studio
(2)点击new,import project导入(tensorflow所在路径)/tensorflow/examples/android文件夹
(可连接手机先run,保证demo能够正常运行后再行修改)
(3)将build.gradle中68行的bazelLocation改为自己bazel的路径:def bazelLocation = '/home/seven/bin/bazel'
185行apply from: "download-models.gradle"
注释掉,并在第112行,增加//*/
(不然后面的内容都被当做注释了):
if (nativeBuildSystem == 'bazel' || nativeBuildSystem ==