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

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可视化模型。这可能有助于找出与输入形状的任何不一致之处。