使用yolov3-tiny.pth官方权重文件转onnx后使用昇腾Atlas 200I DK A2做模型推理,报错execute model failed, modelId is 1
收藏回复举报
使用yolov3-tiny.pth官方权重文件转onnx后使用昇腾Atlas 200I DK A2做模型推理,报错execute model failed, modelId is 1
发表于2024-04-01 09:46:26
0 查看

首先,下载官方yolov3.pth模型之后运行export.py 命令将模型转为.onnx文件。

其次,使用atc转换命令将.onnx文件转为.om文件,转化命令如下:

atc --model=yolov3-tiny.onnx --framework=5 --output=yolov3 --input_shape="images:1,3,360,360" --soc_version=Ascend310B4

之后,运行在官方下载的yolov3目标检测sample文件,运行命令如下:

cd $HOME/samples/cplusplus/level2_simple_inference/2_object_detection/YOLOV3_coco_detection_picture/scripts

bash sample_build.sh

bash sample_run.sh

在运行bash sample_run.sh命令时报如下错误:

[ERROR]  execute model failed, modelId is 1

[ERROR]  Execute model inference failed

[ERROR]  Inference model inference output data failed

cke_11897.png

求教大佬是哪里的问题?

本帖最后由 匿名用户2024/04/01 10:09:33 编辑

我要发帖子