将tensorflow训练的模型导入到OpenCV使用

本文基于使用tensorflow训练好的车辆检测模型,将其转换成OpenCV能够加载的模式,代码可在tf-car上下载。

首先将下载最新的opencv版本,截止到本文成行时的版本为3.4.3,(3.4.2之前的没法转换模型,但可以加载转换好的模型),将下面的代码存为conver.bat,双击执行即可

@echo off
set OPENCV_DIR=D:/opencv343
python %OPENCV_DIR%/sources/samples/dnn/tf_text_graph_ssd.py --input frozen_inference_graph.pb --config ssd_mobilenet_v1_car.config --output graph.pbtxt
pause

参考

https://github.com/opencv/opencv/wiki/TensorFlow-Object-Detection-API

猜你喜欢

转载自blog.csdn.net/minstyrain/article/details/82529778