OpenVINO开发小组
直播中

从未拥有

9年用户 971经验值
擅长:可编程逻辑 电源/新能源 光电显示
私信 关注
[问答]

将YOLOv5 ONNX模型转换为中间表示(IR)格式时,收到与节点相关的错误怎么解决

      

  • 将 YOLOv5 ONNX 模型转换为 IR 格式:python /opt/intel/openvino_2021/deployment_tools/model_optimizer/mo.py --input_model "model.onnx" --output_dir  "cur_folder/"


  • 收到错误:[ ERROR ] The ExpandDims node Unsqueeze_583 has more than 1 input
            [ ERROR ] Cannot infer shapes or values for node "Slice_4".
            [ ERROR ] Output shape: [0 3 0 0] of node "Slice_4" contains non-positive values
            [ ERROR ] It can happen due to bug in custom shape infer function .
            [ ERROR ] Or because the node inputs have incorrect values/shapes.
            [ ERROR ] Or because input shapes are incorrect (embedded to the model or passed via --input_shape).

   
            

回帖(1)

李煜启

2023-8-15 10:35:13
YOLOv5 中有三个输出节点,所有节点都需要在命令中指定:
模型优化器命令: python mo.py --input_model model.onnx --output  --input_shape [1,3,512,512]
举报

更多回帖

发帖
×
20
完善资料,
赚取积分