我有一个在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]
?
谢谢!