Search code examples
pythontensorflowopencvobject-detectionmobilenet

Don't understand "TensorFlow Object Detection" with OpenCV GitHub Wiki


I am trying to use this wiki to detect objects with Python OpenCV. But I don't understand this line of code we're supposed to use:

python tf_text_graph_faster_rcnn.py --input /path/to/model.pb --config /path/to/example.config --output /path/to/graph.pbtxt

I want to use MobileNet-SSD v3, so I downloaded this, as well as the tf_text_graph_ssd.py file and the ssdlite_mobilenet_v3_large_320x320_coco.config file.

But none of the files have the *.pb extension listed in the python command line, and this file already has the *.pbtxt output-extension, so I don't get what that line has to do exactly?

This probably is a very basic question, but I've been struggling with this for some time now so I thought I may ask.

Thank you!


Solution

  • this wiki asking you to download the .pb file which is a model from the Model Zoo. However the link is broken. Here is the the working link.

    MobileNet-SSD v3 is here, you can find other MobileNet_SSD v3 models in the link above as well.

    After unzip the downloaded file, you will find the file, frozen_inference_graph.pb. That is the model file you need.