Skip to content
Advertisement

Tag: object-detection-api

Tensorflow’ pb and pbtxt files don’t work with OpenCV after retraining MobileNet SSD V1 COCO

I have followed this tutorial to retrain MobileNet SSD V1 using Tensorflow GPU as described and got 0.5 loss after training using GPU (below more info about config) and got model.ckpt. This is the command I used for Training: python ../models/research/object_detection/legacy/train.py –logtostderr –train_dir=./data/ –pipeline_config_path=./ssd_mobilenet_v1_pets.config And this is the command for freezing (generate pb file): python ../models/research/object_detection/export_inference_graph.py –input_type image_tensor –pipeline_config_path ./ssd_mobilenet_v1_pets.config

Advertisement