Empfangene Knotenfehler bei der Konvertierung des YOLOv5 ONNX-Modells in das IR-Format (Intermediate Representation)

• YOLOv5 ONNX-Modell in IR-Format konvertiert:

  python /opt/intel/openvino_2021/deployment_tools/model_optimizer/mo.py --input_model "model.onnx" --output_dir  "cur_folder/"

• Erhaltener Fehler:

  [ 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).

Es gibt drei Ausgabeknoten in YOLOv5, und alle von ihnen müssen im Befehl angegeben werden:

Model Optimizer-Befehl: python mo.py --input_model model.onnx --output --input_shape [1,3,512,512]

Verwenden Sie Netron , um das YOLOv5 ONNX-Gewicht zu visualisieren.