我有一个在https://customvision.ai中创建的对象检测模型。如果我将它导出为 TensorFlow Lite 模型,我会得到一个期望FLOAT32 [1, 416, 416, 3]
作为输入并FLOAT32 [1, 13, 13, 35]
作为输出返回的模型(根据 TensorFlow Lite 的visualize.py
)。
我想在 Android 应用程序中使用该模型。我尝试将.tflite
模型文件加载到TensorFlow Lite 对象检测示例应用程序中,但它需要不同的格式。运行应用程序时出现以下异常。java.lang.IllegalArgumentException: Cannot copy between a TensorFlowLite tensor with shape [1, 13, 13, 35] and a Java object with shape [1, 10, 4].
调整示例应用程序以使用 customvision.ai 中的模型是否可行?
我应该如何解释形状[1, 13, 13, 35]
?
提前致谢!