yolov5s的pt模型转onnx,在ubuntu20.04环境下进行校验,报错如下:
您好,根据报错提示,是onnx转bin阶段出现问题。看您的截图,请问您是不是用的OE2.3.3版本,后装了horizon-nn库?
为了避免版本不匹配造成的问题,我这边给您提供三个whl包,可以通过链接下载:
https://pan.horizon.ai/index.php/s/xpt9pHtrAY7M7MC
在docker中,解压后,通过命令安装:pip install ***.whl
然后重走模型转换流程。
望采纳!!点击该回答右侧的“采纳”按钮即可采纳!!
ONNX 只支持一些常见的层和操作,并不支持所有 PyTorch 模型中可能使用的层和操作。
因此,如果 PyTorch 模型中使用了不支持的层或操作,则可能会在尝试将其转换为 ONNX 模型时看到“there is an error in pass:add_calibration_node_for_node_all_inputs”错误。
要解决此问题,请尝试替换不支持的层或操作,或者考虑使用其他模型交换格式,如 TensorFlow SavedModel