【OpenVINO】YOLOv3的tensorflow pb模型转换成IR格式数据(.xml和.bin)

默认已经配置好OpenVINO,配置方法可参考链接:【OpenVINO】Win 10安装配置OpenVINO指南。

pb文件需是冻结之后的模型文件。

打开cmd,进入目录

cd C:\Program Files (x86)\IntelSWTools\openvino\deployment_tools\model_optimizer

YOLOv3:

python mo_tf.py --input_model D:\tensorflow-yolo-v3\frozen_darknet_yolov3_model.pb --tensorflow_use_custom_operations_config C:\Program Files (x86)\IntelSWTools\openvino\deployment_tools\model_optimizer\extensions\front\tf\yolo_v3.json --input_shape=[1,416,416,3]  --data_type=FP16

在当前文件夹下生成三个文件:

YOLOv3-tiny:

python mo_tf.py --input_model D:\tensorflow-yolo-v3\frozen_darknet_yolov3_tiny_model.pb --tensorflow_use_custom_operations_config C:\Program Files (x86)\IntelSWTools\openvino\deployment_tools\model_optimizer\extensions\front\tf\yolo_v3_tiny.json --input_shape=[1,416,416,3]  --data_type=FP16 --tiny

你可能感兴趣的:(深度学习,tensorflow,OpenVINO)