官方文档
https://docs.openvinotoolkit.org/latest/openvino_docs_MO_DG_prepare_model_convert_model_tf_specific_Convert_YOLO_From_Tensorflow.html
假设你已经在Ubuntu配置好openvino。其实跟着官网操作,问题不大,一些小问题百度谷歌基本能解决,主要注意的是tensorflow的版本要一致,这里有两个地方需要用到tensorflow,一是darknet的yolov3.weights转tf的pb模型,二是pb模型转openvino的IR模型。
运行环境
Ubuntu 16.04
python 3.5
tensorflow-gpu 1.12.0
CUDA 9.0 # 如果没有安装cuda9,选择tensorflow 1.12.0 cpu版本也可以
openvino 2020.4.287
第一步下载官方的darket模型,这里默认用的是416*416的yolov3.weights,还有coco数据标签coco.names,这个随便网上都可以搜到.
第二步下载yolov3的tf实现,这个官方有给我们指定了一个版本,主要是用到里面的转换工具convert