omg工具转换模型时动态维度的解决方法（HarmonyOS 鸿蒙Next）

omg工具转换模型时动态维度的解决方法（HarmonyOS 鸿蒙Next）

  1. 下面以onnx转om为例，假设我们的模型输入维度为inputname[1,n1]其中n1为动态维度，不指定固定，官方omg参数中提供了设置动态维度的选项，但我这样：./omg --model xxx.onnx --framework 5 --output ./modename --input_shape＝inputname:1,-1

  2. 设置命令后会报错：E/AI_FMK (6778): model_build_options_builder.cpp ParseInputShape(151)::“Invalid parameter for input shape: input_ids:1,-1 ,expect positive integer , but value = -1”

不知道是不是只是我会报错，在网上翻阅后查看了昇腾平台atc转换参数后，可以设置一个动态范围，以onnx模型转换为例：命令如下./omg --model xxx.onnx --framework 5 --output ./modename --input_shape＝inputname:1,1~100此命令成功转并设置换动态维度。


更多关于omg工具转换模型时动态维度的解决方法（HarmonyOS 鸿蒙Next）的实战教程也可以访问 https://www.itying.com/category-93-b0.html

3 回复

顶一个

更多关于omg工具转换模型时动态维度的解决方法（HarmonyOS 鸿蒙Next）的实战系列教程也可以访问 https://www.itying.com/category-93-b0.html


重新研究了一下文档，设置shape为-1必须配合-dynamic_dims参数使用才可以，

在HarmonyOS鸿蒙Next中，使用OMG工具转换模型时处理动态维度问题，可以通过以下方式解决：

  1. 指定动态维度范围：在转换过程中，明确指定动态维度的可能取值范围。OMG工具会根据这些范围生成适配的模型。

  2. 使用占位符：在模型定义中使用占位符（Placeholder）来表示动态维度，确保模型在不同输入尺寸下都能正常工作。

  3. 模型优化：通过OMG工具的优化功能，调整模型结构以适应动态维度的变化，确保推理时的兼容性。

  4. 配置文件调整：在OMG工具的配置文件中，设置相关参数以适应动态维度，确保转换后的模型支持动态输入。

这些方法可以帮助在HarmonyOS鸿蒙Next中有效处理OMG工具转换模型时的动态维度问题。

回到顶部