文章 ID: 000088667 內容類型: 疑難排解 最近查看日期: 2022 年 05 月 20 日

如何將 Keras* H5 模型轉換為中間表示 (IR) 格式?

BUILT IN - ARTICLE INTRO SECOND COMPONENT
摘要

將 Keras H5 模型轉換為 IR 格式的步驟。

描述

無法確定將 Keras* H5 模型轉換為 IR 格式的步驟。

解決方法

步驟 1:將 Keras H5 型號 轉換 為已節省的模式格式

  1. 安裝 依存關係:
    cd \deployment_tools\model_optimizer\install_prerequisites
    install_prerequisites_tf2.bat
  2. 使用 TensorFlow* 2載入模型,並以「已存取的模式」格式將其序列化
    import tensorflow as tf
    model = tf.keras.models.load_model('model.h5')
    tf.saved_model.save(model,'model')
  3. 選用:從 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')

 

步驟 2:將已節省的模型格式 轉換 為中間表示

  1. 將目錄變更/deployment_tools/model_optimizer
  2. 執行 mo_tf.py 腳本,其中包含通往 SavedModel 目錄的路徑和可取名的輸出目錄:
    python mo_tf.py --saved_model_dir --output_dir --input_shape

相關產品

本文章適用於 2 產品

這個頁面的內容綜合了英文原始內容的人工翻譯譯文與機器翻譯譯文。本內容是基於一般資訊目的,方便您參考而提供,不應視同完整或準確的內容。如果這個頁面的英文版與譯文之間發生任何牴觸,將受英文版規範及管轄。 查看這個頁面的英文版。