SNPE 模型转换：0 维模型输入的语法

Question

我正在使用 SNPE 将 NN 模型从

.tflite

格式转换为

.dlc

。通常，我转换的模型很容易工作，因为只需将每个输入的尺寸指定为 SNPE 中的

snpe-tflite-to-dlc

工具的参数即可。然而，我当前的模型有 2 个输入，它们根本没有维度。请参阅我的 Netron 模型可视化截图：

有关此工具的文档没有提及在这种情况下应指定哪种输入，仅给出了多个输入的示例。我的命令目前如下所示：

snpe-tflite-to-dlc \
        --input_network facebook_denoiser.tflite \
        --input_dim audio "1,1,100000" \
        --input_dim arg0 "" \
        --input_dim arg1 "" \
        --input_dim arg2 "390,1,768" \
        --input_dim arg3 "1,768" \
        --input_dim arg4 "1,768" \
        --input_dim arg5 "390,1,768" 

运行它不起作用，我尝试了各种其他类似的方法，例如完全省略

arg0

和

arg1

，或者将它们设置为0，但该工具只是抛出一个NoneType错误（这通常意味着参数不是格式正确）。

有谁知道如何转换这样的模型吗？

Answer 1

您好，谢谢您的提问。

为此，您需要对输入节点及其尺寸进行硬编码。目前SNPE不支持动态输入。