使用OpenVINO转换tensorFlow*模型

1.TensorFlow模型转换的复杂度

与Caffe、ONNX不同的是,使用Model-Optimizer将TensorFlow模型转换为IR模型的复杂度更高。
其主要原因是:
(1)TensorFlow框架支持多种操作这些操作优势甚至无法用简单的图形表示,这些操作有时甚至也与计算机视觉无关。
(2)如果不进行处理而试图只使用模型本身,并在神经网络加速器上对其进行加速,极大可能报错。

2.确保模型优化器可以对TensorFlow模型进行转换

(1)进入 cd C:\Program Files(x86)\Intel\openvino_2021\deployment_tools\model_optimizer\install_prerequisites路径
(2)执行脚本install_prerequisites_tf.bat安装转换所必须的安装包。

3.下载TensorFlow模型

使用model-downloader下载ssd-mobilenet_v2_coco模型

4.使用tensorflow_object_detection_api处理和转换模型

(1)进入cd C:\Program Files(x86)\Intel\openvino_2021\deployment_tools\model_optimizer路径
(2)输入命令:
operations_config="<INSTALL_DIR>\deployment_tools\model_optimizer\extensions\front\tf\ssd_v2_support.json" --tensorflow_object_detection_api_pipeline=<model_dir>\pipeline.config --reverse_input_channels --data_type=FP16 --output_dir= <ir_python mo_tf.py --input_model --input_model <model_dir><model_file> --tensorflow_use_custom_dir>


image.png

(3)转换结果


image.png
©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

相关阅读更多精彩内容

友情链接更多精彩内容