如何找到有关如何训练 TFHub 上的 ssd_mobilenet_v1 tflite 模型的更多信息?
它的训练方式是否通过避免 tflite 不支持的某些操作来轻松将其转换为 tflite?或者它是否经过正常训练,然后使用带有 TF Select 的 tflite 转换器和有关此 github issue 的提示进行转换?
另外,有谁知道是否有在 OpenImagesV6 上训练的等效 mobilenet tflite 模型?如果不是,那么训练一个的最佳起点是什么?
答案 0 :(得分:1)
我不确定模型的确切来源,但看起来它确实具有与 TFLite 兼容的操作。根据我的经验,兼容 TFLite 的 SSD 型号的最佳起点是 TF2 Detection Zoo。您可以使用 these instructions 转换任何 SSD 型号。
要训练您自己的模型,您可以遵循利用 Google Cloud 的 these instructions。