目的:将自己训练的yolov5s.pt移植到昇腾上运行
过程:用yolov5的export.py导出onnx模型,再用atc将onnx转化成om模型
问题:转换后的om模型在昇腾orangepi AIpro板子上运行结果错误,如图,会出现很多1.000000的预测框。(这个模型是用的官方的yolov5s.pt转化的,怀疑是转换工具出问题了)
环境:
虚拟机ubuntu20.04LTS
转换工具Ascend-cann-toolkit_7.0.RC1_linux-x86_64
python3.8.0
过程:用yolov5的export.py导出onnx模型,再用atc将onnx转化成om模型
问题:转换后的om模型在昇腾orangepi AIpro板子上运行结果错误,如图,会出现很多1.000000的预测框。(这个模型是用的官方的yolov5s.pt转化的,怀疑是转换工具出问题了)
环境:
虚拟机ubuntu20.04LTS
转换工具Ascend-cann-toolkit_7.0.RC1_linux-x86_64
python3.8.0