在完成Tensorflow Object Detection API模型训练后,可以使用OpenVINO工具套件中的Model Optimitzer工具对原有模型进行优化。操作步骤如下:
①打开cmd,输入如下指令,指令中的含义见下表所示
(指令如下:python mo_tf.py…)
python mo_tf.py --input_model=E:\T-Tensorflow\tf_train\workspaces\cats_dogs\trained_frozen_models\cats_dogs_model\frozen_inference_graph.pb --tensorflow_use_custom_operations_config="C:\Intel\openvino_2021.1.110\deployment_tools\model_optimizer\extensions\front\tf\ssd_v2_support.json" --tensorflow_object_detection_api_pipeline_config=E:\T-Tensorflow\tf_train\workspaces\cats_dogs\trained_frozen_models\cats_dogs_model\pipeline.config --reverse_input_channels --data_type=FP16 --output_dir=E:\T-Tensorflow\tf_train\workspaces\cats_dogs\IR_Model\ --model_name=cats_dogs_detector
mo_tf.py中各指令的含义:
指令 |
---|