/garbage-classification-android

Primary LanguageKotlin

garbage-classification-android

使用 TensorFlow Lite 在移动端进行图像识别，识别不同的垃圾并进行分类。

基础模型在 MobileNetV2（ImageNet）基础上进行迁移学习，识别 40 种不同的垃圾。

如何更换模型

将模型文件重命名为 garbage.tflite，复制到/android/app/src/main/assets，然后重新构建。

如果模型的预处理步骤、输入尺寸或输出类别等参数有变化，则需要修改 Classifier 中的对应代码。

如何转换模型

TensorFlow Lite 的模型使用 FlatBuffer 格式，可以使用 TensorFlow Lite converter 对 TensorFlow 模型进行转换。