TFHub 中的 ssd_mobilenet_v1 tflite 模型是如何训练的?

时间:2021-01-24 21:37:06

标签: tensorflow tensorflow-lite

如何找到有关如何训练 TFHub 上的 ssd_mobilenet_v1 tflite 模型的更多信息?

它的训练方式是否通过避免 tflite 不支持的某些操作来轻松将其转换为 tflite?或者它是否经过正常训练,然后使用带有 TF Select 的 tflite 转换器和有关此 github issue 的提示进行转换?

另外,有谁知道是否有在 OpenImagesV6 上训练的等效 mobilenet tflite 模型?如果不是,那么训练一个的最佳起点是什么?

1 个答案:

答案 0 :(得分:1)

我不确定模型的确切来源,但看起来它确实具有与 TFLite 兼容的操作。根据我的经验,兼容 TFLite 的 SSD 型号的最佳起点是 TF2 Detection Zoo。您可以使用 these instructions 转换任何 SSD 型号。

要训练您自己的模型,您可以遵循利用 Google Cloud 的 these instructions