原因：java.lang.IllegalArgumentException：标签编号 6 与部署我自己的模型后轴 1 上的形状不匹配答案

【问题标题】：Caused by: java.lang.IllegalArgumentException: Label number 6 mismatch the shape on axis 1 afted deploying my own model原因：java.lang.IllegalArgumentException：标签编号 6 与部署我自己的模型后轴 1 上的形状不匹配
【发布时间】：2021-08-04 03:53:36
【问题描述】：

您好，我正在尝试使用 tflite 模型构建一个 android 图像分类项目来对棋子进行分类。我已经训练了我的模型并在我的 android 项目的 assets 文件夹中部署了 tflite 模型以及标签映射，但是我收到了这个错误： Caused by: java.lang.IllegalArgumentException: Label number 6 mismatch the shape on axis 1.

由于棋子的数量，我确实有六堂课。我尝试了具有不同模型和标签图的应用程序，并且该应用程序可以正常工作。我不知道我在哪里犯了错误。我的 labelmap 没有空行或额外的类。这是 tflite 文件：https://www.pastefile.com/a44ydg 这是我的标签映射文件： https://www.pastefile.com/9rg9v7 如果需要，我可以提供整个应用程序，但同样只有此模型才会发生这种情况。

附：这是来自 android studio 的完整错误日志：https://pastebin.com/bxdq9x1r

【问题讨论】：

标签： android tensorflow tensorflow2.0 tensorflow-lite

【解决方案1】：

您能否仔细检查模型 (https://www.pastefile.com/a44ydg) 中的 4 个输出以查看形状是否与 6 个标签匹配？

我使用netron 来检查您的model，但不知道输出的形状。

您可以参考page 了解 4 个输出“TFLite_Detection_PostProcess”、“TFLite_Detection_PostProcess:1”、“TFLite_Detection_PostProcess:2”和“TFLite_Detection_PostProcess:3”是什么。

【讨论】：

您好，对不起，我是新手。我阅读了您链接我的文档。我有这样的命令，我在以 tflite 格式制作模型之前执行。您想查看命令以了解这 4 个输出吗？
首先，也许您可以将错误消息的完整调用堆栈粘贴到 Java 中以显示更多信息。其次，关键是您的输出与形状不匹配。例如，形状为[1, N]，但标签映射包含6 行（其中轴1：N != 6），因此会发生错误。您可以尝试python 中的推理代码以了解输出。通常，interpreter.get_output_details() 将返回包含类型和形状的字典，因此您可以检查形状。
如果可以帮助您获得更好的图片，我可以将评估结果粘贴到 tensorboard 中。我从张量板上看到对形状的检测不是很好。我现在尝试增加步骤并添加更多图片。
我在原帖中添加了完整的错误信息。
这是我用于生成 tflite 的命令：command = "tflite_convert \ --saved_model_dir={} \ --output_file={} \ --input_shapes=1,300,300,3 \ --input_arrays=normalized_input_image_tensor \ --output_arrays='TFLite_Detection_PostProcess','TFLite_Detection_PostProcess:1','TFLite_Detection_PostProcess:2','TFLite_Detection_PostProcess:3' \ --inference_type=FLOAT \ --allow_custom_ops".format(FROZEN_TFLITE_PATH, TFLITE_MODEL, ) 如果您查看 input_shapes，我认为这是模型将知道的图像形状，当我用手机拍照时，我获取例如 2000x3000px