在ubuntu虚拟机上用atc将yolov5s的onnx转换为om,到昇腾上运行,预测结果不一致
收藏回复举报
在ubuntu虚拟机上用atc将yolov5s的onnx转换为om,到昇腾上运行,预测结果不一致
t('forum.solved') 已解决
新人帖
发表于2024-04-26 11:22:22
0 查看
目的:将自己训练的yolov5s.pt移植到昇腾上运行

过程:用yolov5的export.py导出onnx模型,再用atc将onnx转化成om模型

问题:转换后的om模型在昇腾orangepi AIpro板子上运行结果错误,如图,会出现很多1.000000的预测框。(这个模型是用的官方的yolov5s.pt转化的,怀疑是转换工具出问题了)
%E5%BE%AE%E4%BF%A1%E5%9B%BE%E7%89%87_20240426111926.jpg

环境:
虚拟机ubuntu20.04LTS
转换工具Ascend-cann-toolkit_7.0.RC1_linux-x86_64
python3.8.0

我要发帖子