Community
cancel
Showing results for 
Search instead for 
Did you mean: 
idata
Community Manager
381 Views

Compiling TensorFlow ssd_mobilenet_v1_coco model

I want to compile the tf model ssd_mobilenet_v1_coco of the zoo. However, when I use the following command to Export GraphDef file,

 

python3 /home/stella/models/research/slim/export_inference_graph.py \

 

--alsologtostderr \

 

--model_name=ssd_mobilenet_v1_coco \

 

--batch_size=1 \

 

--dataset_name=imagenet \

 

--image_size=299 \

 

--output_file=ssd_mobilenet_v1_coco.pb

 

Traceback (most recent call last):

 

File "/home/stella/models/research/slim/export_inference_graph.py", line 156, in

 

tf.app.run()

 

File "/usr/local/lib/python3.5/dist-packages/tensorflow/python/platform/app.py", line 125, in run

 

_sys.exit(main(argv))

 

File "/home/stella/models/research/slim/export_inference_graph.py", line 129, in main

 

is_training=FLAGS.is_training)

 

File "/home/stella/models/research/slim/nets/nets_factory.py", line 148, in get_network_fn

 

raise ValueError('Name of network unknown %s' % name)

 

ValueError: Name of network unknown ssd_mobilenet_v1_coco

 

It reports error!

 

Can you help me compile the ssd_mobilenet_v1_coco model thx!

 

regards,
Tags (1)
0 Kudos
1 Reply
idata
Community Manager
79 Views

Hi @Stella2424

 

You're using OpenVINO, correct? Here is a helpful guide on converting Tensorflow Object Detection models. Please let me know if you have any further questions.

 

Best Regards,

 

Sahira
Reply