在PC端进行yolo11-obb模型训练后得到.pt模型,转换为onnx,再转换为om后,在板卡上推理得到的类别概率发生了移位,导致推理错误;
目前问题是左移了一位,手动挪回来了,但是第0个位置的值找不回来了。如何修正模型转换解决此问题?或者说其他方面的问题。
以下是我用的pt2onnx代码和atc转换onnx2om代码:

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
目前问题是左移了一位,手动挪回来了,但是第0个位置的值找不回来了。如何修正模型转换解决此问题?或者说其他方面的问题。
以下是我用的pt2onnx代码和atc转换onnx2om代码:
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