OpenCV调用TensorFlow的目标检测

旧城等待， 2024-04-18 22:35 77阅读 0赞

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

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly96aGFuZ3BpbmcuYmxvZy5jc2RuLm5ldA_size_16_color_FFFFFF_t_70][]

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

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly96aGFuZ3BpbmcuYmxvZy5jc2RuLm5ldA_size_16_color_FFFFFF_t_70 1][]

生成的文件为网路结构

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

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly96aGFuZ3BpbmcuYmxvZy5jc2RuLm5ldA_size_16_color_FFFFFF_t_70 2][]

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

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly96aGFuZ3BpbmcuYmxvZy5jc2RuLm5ldA_size_16_color_FFFFFF_t_70]: https://img-blog.csdnimg.cn/20190912210242757.PNG?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly96aGFuZ3BpbmcuYmxvZy5jc2RuLm5ldA==,size_16,color_FFFFFF,t_70
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly96aGFuZ3BpbmcuYmxvZy5jc2RuLm5ldA_size_16_color_FFFFFF_t_70 1]: https://img-blog.csdnimg.cn/20190912210331587.PNG?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly96aGFuZ3BpbmcuYmxvZy5jc2RuLm5ldA==,size_16,color_FFFFFF,t_70
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly96aGFuZ3BpbmcuYmxvZy5jc2RuLm5ldA_size_16_color_FFFFFF_t_70 2]: https://img-blog.csdnimg.cn/20190912214559355.PNG?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly96aGFuZ3BpbmcuYmxvZy5jc2RuLm5ldA==,size_16,color_FFFFFF,t_70