ID do artigo: 000088394 Tipo de conteúdo: Mensagens de erro Última revisão: 20/05/2022

Erros relacionados ao nó recebido ao converter o modelo YOLOv5 ONNX em formato de representação intermediária (IR)

BUILT IN - ARTICLE INTRO SECOND COMPONENT
Resumo

Corrigir o comando Model Optimizer para converter o modelo YOLOv5 ONNX em formato IR.

Descrição
  • Modelo YOLOv5 ONNX convertido em formato IR:

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

  • Erro recebido:

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

Resolução

Há três nós de saída no YOLOv5 e todos eles precisam ser especificados no comando:

Comando Model Optimizer: python mo.py -input_model model.onnx --nós de saída --input_shape [1,3.512.512]

Outras informações

Use a Netron para visualizar o peso YOLOv5 ONNX.

Produtos relacionados

Este artigo aplica-se a 2 produtos

O conteúdo desta página é uma combinação de tradução humana e por computador do conteúdo original em inglês. Este conteúdo é fornecido para sua conveniência e apenas para informação geral, e não deve ser considerado completo ou exato. Se houver alguma contradição entre a versão em inglês desta página e a tradução, a versão em inglês prevalecerá e será a determinante. Exibir a versão em inglês desta página.