首页 > 解决方案 > 如何将微型 yolov2 模型转换为 tensorflow lite 模型?

问题描述

我已经训练了一个自定义的微型 yolov2 模型,并希望在 Xamarin Android 应用程序中使用这个模型。我已经使用 darkflow 将 yolov2 权重文件转换为 .pb。转换后的模型在 Xamarin Android 应用程序中不起作用,并且出现错误。This blog post说.pb模型应该转换为tensorflow lite版本以在Android上运行,我也将模型转换为tensorflow lite模型,但它仍然给出错误

加载 model.lite 失败,不是有效的 TensorFlow Graph 序列化:无效的 GraphDef

如何将微型 yolov2 模型转换为 tensorflow lite 模型以使其与 Android 应用程序一起使用?

标签: androidtensorflowtensorflow-liteyolodarkflow

解决方案


https://www.tensorflow.org/lite/guide 只需按照以下步骤操作。包括 tensorflow 解释器并将您的模型转换为 tensorflow lite,如链接所示。你在为android上的实时对象检测编码吗?


推荐阅读