最近做的一个项目需要在c++程序中调用目标检测模型,但是在使用opencv调用模型时总是遇到各种各样的错误,一直没有成功。最近终于打通了这一过程,总结如下(本文以SSD_MobileNet模型为例)。
运行环境:
opencv >= 3.4.4,
Visual Studio 2015,
训练环境参考本文引用的其他文章。
1. opencv 调用训练好的目标检测模型
具体调用代码参考 opencv dnn模块 示例(5) 目标检测 object_detection (4) TensorFlow SSD,代码中调用的两个模型文件分别来自 ssd_mobilenet_v1_coco_2017_11_17.pb,ssd_mobilenet_v1_coco_2017_11_17.pbtxt。文件 object_detection_classes_coco.txt 可以在 opencv 安装目录下的 \sources\samples\data\dnn 目录中找到。