OpenCV调用TensorFlow的目标检测

最新推荐文章于 2023-02-09 10:19:18 发布

原创最新推荐文章于 2023-02-09 10:19:18 发布 · 1.8k 阅读

1 ·

CC 4.0 BY-SA版权

本文介绍如何使用Python脚本生成SSD Inception V2和Faster R-CNN Inception ResNet V2 Atrous两种模型的结构文件，并展示如何利用这些模型进行目标检测。具体步骤包括：生成网络结构.pbtxt文件，设置输入图片、模型、配置文件路径，指定输出文件名等。

python PATH\tf_text_graph_ssd.py 
--input path\frozen_inference_graph.pb 
--config path\pipeline.config
--output path\ssd_inception_v2.pbtxt

运行以上脚本会生成网络的结构.pbtxt

生成的文件为网路结构

python .\object_detection.py 
--input .\model\timg.jpg 
--model .\model\frozen_inference_graph.pb 
--config .\model\ssd_inception_v2_graph.pbtxt 
--height 300 
--width 300 
--classes .\model\mscoco.txt

python PATH\tf_text_graph_faster_rcnn.py 
--input .\frozen_inference_graph.pb
--config .\pipeline.config
--output .\faster_rcnn_inception_resnet_v2_atrous.pbtxt