轉換 TensorFlow 物件偵測 API 模型以進行Intel® 類神經電腦棒 2推斷 （Intel® NCS 2）

請按照 Intel® NCS 2 的入門指南 安裝OpenVINO™工具組並設定您的 Intel® NCS 2。

mo_tf.py腳本位於~/intel/openvino/deployment_tools/model_optimizer目錄。將模型轉換為中間表示 （IR） 以供 Intel® NCS 2 推斷時，需要指定下列參數。

--input_model <path_to_frozen.pb>

• 如需額外資訊，請參考 Python 中的 凍結自訂模型 。

--tensorflow_use_custom_operations_config <path_to_subgraph_replacement_configuration_file.json>

• 配置檔案位於 ~/intel/openvino/deployment_tools/model_optimizer/擴充/前端/tf 目錄。選取符合您機型拓撲的配置檔案。查看如何為配置檔案清單 轉換模型 。

--tensorflow_object_detection_api_pipeline_config <path_to_pipeline.config>

--reverse_input_channels

• 如果您使用的是經過轉換的 TensorFlow 物件偵測 API 模型與推斷引擎樣品應用程式，則需要此參數。

--data_type FP16

• 指定半精確度浮點格式，以在 Intel® NCS 2 上執行

模型優化程式指令的範例：

python3 ~/intel/openvino/deployment_tools/model_optimizer/mo_tf.py --input_model frozen_model.pb --tensorflow_use_custom_operations_config ~/intel/openvino/deployment_tools/model_optimizer/extensions/front/tf/ssd_v2_support.json --tensorflow_object_detection_api_pipeline_config pipeline.config --reverse_input_channels --data_type FP16

有關模型優化工具的其他資訊可在 OpenVINO™工具組檔中找到。