训练yolo11-obb后,进行模型转换,边缘推理出现问题
收藏回复举报
训练yolo11-obb后,进行模型转换,边缘推理出现问题
t('forum.solved') 已解决
发表于2026-01-11 22:27:55
0 查看
在PC端进行yolo11-obb模型训练后得到.pt模型,转换为onnx,再转换为om后,在板卡上推理得到的类别概率发生了移位,导致推理错误;
目前问题是左移了一位,手动挪回来了,但是第0个位置的值找不回来了。如何修正模型转换解决此问题?或者说其他方面的问题。
以下是我用的pt2onnx代码和atc转换onnx2om代码:
cke_4963.png
atc --model=bestl.onnx --framework=5 --output=50newyolo11obbl_om --input_shape="images:1,3,640,640" --soc_version=Ascend310B1 --input_format=NCHW --precision_mode=allow_mix_precision

我要发帖子