如何使用customvision.ai为TensorFlow Lite创建对象检测模型?

时间:2019-10-09 16:05:10

标签: tensorflow microsoft-cognitive tensorflow-lite microsoft-custom-vision

我有一个在https://customvision.ai中创建的对象检测模型。如果将其导出为TensorFlow Lite模型,则会得到一个期望FLOAT32 [1, 416, 416, 3]作为输入并返回FLOAT32 [1, 13, 13, 35]作为输出的模型(根据TensorFlow Lite的visualize.py)。

我想在Android应用中使用该模型。我尝试将.tflite模型文件加载到TensorFlow Lite object detection sample app中,但是期望使用不同的格式。运行该应用程序时出现以下异常。 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]

谢谢!

0 个答案:

没有答案