获取om的输出结果output_sizes ： acl.mdl.get_output_size_by_index(model_desc, i)为0

一、问题现象（附报错日志上下文）：
获取om的输出结果output_sizes 为[0, 0, 0] ,即acl.mdl.get_output_size_by_index(model_desc, i)结果为0

二、软件版本:
-- CANN 版本 (e.g., CANN 3.0.x，5.x.x):  CANN 8.0.0
--Tensorflow/Pytorch/MindSpore 版本: torch 2.1.0
--Python 版本 (e.g., Python 3.7.5): Python 3.9.2 
-- MindStudio版本 (e.g., MindStudio 2.0.0 (beta3)):
--操作系统版本 (e.g., Ubuntu 18.04):Ubuntu 20.04 LTS

三、测试步骤：
1.使用atc工具把onnx转换成om，命令：atc --model=xxx_sam_xxx.onnx --framework=5 --output=xxx_sam_xxx --soc_version=Ascend310P3 --input_shape="batched_images:1,3,640,640;batched_point_coords:1,1,1,2;batched_point_labels:1,1,1"

2.执行如下代码

import acl

import numpy as np

ret = acl.init()

ret = acl.rt.set_device(0)

model_id, ret = acl.mdl.load_from_file('./xxx_sam_xxx.om')

model_desc = acl.mdl.create_desc()

ret = acl.mdl.get_desc(model_desc, model_id)

input_num = acl.mdl.get_num_inputs(model_desc)

input_sizes = [

acl.mdl.get_input_size_by_index(model_desc, i)

for i in range(input_num)

]

output_num = acl.mdl.get_num_outputs(model_desc)

output_sizes = [

acl.mdl.get_output_size_by_index(model_desc, i)

for i in range(output_num)

]

print("-----")

print(input_num)

print(input_sizes)

print("-----")

print(output_num)

print(output_sizes)

acl.mdl.unload(model_id)

acl.mdl.destroy_desc(model_desc)

acl.rt.reset_device(0)

acl.finalize()

3.代码执行结果output_sizes的值为[0, 0, 0]

-----

3

[4915200, 8, 4]

-----

3

[0, 0, 0]

四、日志信息:
链接: https://pan.baidu.com/s/1GPQyNW04Js5_G9bvgDO_bQ?pwd=tax9 提取码: tax9

Ascend/modelzoo
暂停

内容风险标识

评论 (0)

Ascend/modelzoo暂停 .gitee-modal { width: 500px !important; }

内容风险标识