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

Question

我有一个在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]？

提前感谢！

Answer 1

具有完全相同的问题，并且不能自己解决。如果您找到解决方案，请告诉我！