ID do artigo: 000088667 Tipo de conteúdo: Solução de problemas Última revisão: 20/05/2022

Como converter o modelo Keras* H5 para um formato de Representação Intermediária (IR) ?

BUILT IN - ARTICLE INTRO SECOND COMPONENT
Resumo

Etapas para converter o modelo Keras H5 em um formato IR.

Descrição

Não foi possível determinar etapas para converter o modelo Keras* H5 em um formato IR.

Resolução

Etapa 1: Converter o modelo Keras H5 para o formato SavedModel

  1. Instalar dependências:
    cd \deployment_tools\model_optimizer\install_prerequisites
    install_prerequisites_tf2.bat
  2. Carregue o modelo usando TensorFlow* 2 e serialize-o no formato SavedModel.
    import tensorflow as tf
    model = tf.keras.models.load_model('model.h5')
    tf.saved_model.save(model,'model')
  3. Opcional: modelo com uma camada personalizada CustomLayer de custom_layer.py:
    import tensorflow as tf
    from custom_layer import CustomLayer
    model = tf.keras.models.load_model('model.h5', custom_objects={'CustomLayer': CustomLayer})
    tf.saved_model.save(model,'model')

 

Etapa 2: Converta o formato SavedModel em Representação Intermediária

  1. Mude o diretório para /deployment_tools/model_optimizer
  2. Execute o script mo_tf.py com um caminho para o diretório SavedModel e um diretório de saída contorcível:
    python mo_tf.py --saved_model_dir --output_dir --input_shape

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.