本文基于使用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