1.模型导入
Android Studio 版本要4.1版本以上,没有的需要更新。
接下来,右键单击要使用 TFLite 模型的模块或单击,然后FileNew > Other > TensorFlow Lite Model ,如图所示:
导入后模型存在在这:
2.程序解析
2.1 layout布局
猫狗分类APP只有一个activity——也就是只有一个界面,布局是放置在“layout”这里的,如下所示
2.2 ImageClassifierActivity
这里是主要进行APP操作的类文件,就相当于C语言中的mian函数
这个接口函数,引用了布局layout,定义了两个函数,一个是 initClassifier(),这个函数调用了 Classifier这个类的对于TF模型的一些关键参数,initViews()这是一个图片点击函数,就是APP最基本的函数功能。
override fun onCreate(savedInstanceState: Bundle?) {
super.onCreate(savedInstanceState)
setContentView(R.layout.activity_image_classifier)
initClassifier()
initViews()
}
这个接口函数,用Toast告诉点击后的结果,也就是具体的分类的标准
override fun onClick(view: View?) {
val bitmap = ((view as ImageView).drawable as BitmapDrawable).bitmap
val result = classifier.recognizeImage(bitmap)
runOnUiThread { Toast.makeText(this, result.get(0).title, Toast.LENGTH_SHORT).show() }
}
package co