object-detection-api - 不可能改变模型来检测

Question

我使用适用于 android 的 tensorflow lite 检测器。例如，我想更改模型以识别狗，我将 dog.tflite 文件和他的标签放在资产中，并在代码中更改要选择的文件（dog.tflite 和他的标签），更改“TF_OD_API_INPUT_SIZE=300”到“TF_OD_API_INPUT_SIZE=224”和“TF_OD_API_IS_QUANTIZED = true;” 到“TF_OD_API_IS_QUANTIZED = false;” 在 DetectorActivity.java 中。但我有错误：“无法在形状为 [1, 25] 的 TensorFlowLite 张量和形状为 [1, 10, 4] 的 Java 对象之间复制。”。如何解决这个问题？

谢谢

标签： object-detection-apitensorflow-lite

Answer 1

您的 TF Lite 模型是“图像分类”模型还是“对象检测”模型？

形状为 [1, 25] 的 TensorFlowLite 张量

从模型输出的形状来看，我相信它是一个图像分类模型。您应该使用此示例代码而不是 object_detection 示例。