ID del artículo: 000088394 Tipo de contenido: Mensajes de error Última revisión: 20/05/2022

Errores relacionados con nodos recibidos al convertir el modelo YOLOv5 ONNX en formato de representación intermedia (IR)

BUILT IN - ARTICLE INTRO SECOND COMPONENT
Resumen

Corrija el comando Optimizador de modelo para convertir el modelo YOLOv5 ONNX en formato IR.

Descripción
  • Modelo YOLOv5 ONNX convertido en formato IR:

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

  • Error recibido:

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

Resolución

Hay tres nodos de salida en YOLOv5 y todos ellos deben especificarse en el comando:

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

Más información

Utilice Netron para visualizar el peso YOLOv5 ONNX.

Productos relacionados

Este artículo se aplica a 2 productos

El contenido de esta página es una combinación de la traducción humana y automática del contenido original en inglés. Este contenido se ofrece únicamente para su comodidad como información general y no debe considerarse completa o precisa. Si hay alguna contradicción entre la versión en inglés de esta página y la traducción, prevalecerá la versión en inglés. Consulte la versión en inglés de esta página.