Intel® Distribution of OpenVINO™ Toolkit
Community assistance about the Intel® Distribution of OpenVINO™ toolkit, OpenCV, and all aspects of computer vision-related on Intel® platforms.

Compiling TensorFlow™ Model Zoo Networks

idata
Employee
987 Views

Hi,

 

I follow the guidance from https://movidius.github.io/ncsdk/tf_modelzoo.html. to compile Inception v3 Model. However, when I did the step of 'Freeze model for inference', it reports that:

 

Traceback (most recent call last):

 

File "/home/stella/tensorflow/tensorflow/python/tools/freeze_graph.py", line 58, in

 

from tensorflow.python.training import checkpoint_management

 

ImportError: cannot import name 'checkpoint_management'

 

I see a similar problem from the forum, the solution is re-cloning the ncappzoo. But I did not make the example from ncappzoo, and I just want to know about the process to compile a tensorflow model of TensorFlow Model Zoo Networks.

 

How I can solve this problem? I use tensorflow 1.4.0 thx
0 Kudos
2 Replies
idata
Employee
612 Views

Hi @Stella2424

 

I believe the issue is related to the tensorflow version you have installed. I was able to freeze the model using tensorflow version 1.11.0 and the NCSDK v 2.10.01.

 

Could you try installing these versions?

 

Regards,

 

Jesus
0 Kudos
idata
Employee
612 Views

@Jesus_at_Intel Thanks! I install tf 1.11.0, and it worked. And 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,
0 Kudos
Reply