TFLite 对象检测模型的 Sagemaker Neo 编译失败

Question

我正在尝试使用 Sagemaker Neo 为 imx8qm 编译 TFLite 对象检测模型。我提供了一个 .tflite 文件并像这样指定输入配置：{"input":[1, 300, 300, 3]}

但是编译失败并出现以下错误： ClientError: InputConfiguration: Framework cannot load TFLite model. Unable to infer tensor data type for all inputs/outputs: Please specify all input layers in data_shape.

如何解决这个问题？

Answer 1

Neo 需要将模型文件压缩为 tar 文件 (.tar.gz)，然后再将其作为输入。

如果这不能解决问题，您可以使用 Netron 可视化您的模型。它可能有助于找出与输入形状的任何不一致之处。