android - 如何在移动设备上重新训练 TensorFlow lite 模型以进行 MLKit 对象检测？我选择重新训练什么 Tensorflow Hub 模型？

Question

我一直在开发一个使用 CameraX API 的 Android 应用程序，我想使用 MLKit 创建一个图像分析器。我的主要目标是检测特定大小的某种类型的文档，一个检测标签用于正面，一个用于背面。所以我想训练一个可以满足我特定需求的 TensorFlow Lite 模型。我已经看过很多很棒的文档，这些文档描述了如何在电话上本地使用我自己的模型来进行分析用例，所以这不是问题。

问题是我是一名移动开发人员，我对实际训练我的模型知之甚少。我有大量的训练数据，因为我已经完成了像这样的一些很棒的教程，并且我有很多要分类的文档的图像（数千张），我已经按照描述将它们组织到各自的目录中。我还观看了这个视频，很明显，使用 TensorFlow 图像分类器和经过训练的模型进行再训练是一种非常有益的方法，但我没有看到太多关于他们选择 Tensorflow Hub lite 模型的确切原因的信息。是什么让一个比其他更好，我应该寻找什么来选择最好的一个来重新培训和满足我的特定需求（检测特定文档）？

这是正确的方法吗？任何建议表示赞赏，谢谢！

标签： androidcomputer-visiontensorflow2.0tensorflow-liteandroid-camerax