如何找到Frozen_inference_graph.pb输入和输出节点?

时间:2019-11-08 18:18:30

标签: tensorflow movidius

我已经通过ssd inception v2模型训练了一个模型,并使用“ export_inference_graph.py”生成了Frozen_inference_graph.pb

我尝试使用以下命令生成模型,然后尝试使用mvNCCompiler将其转换为movidius图

python3 /home/chanchihang/Special_Topic/models/research/object_detection/export_inference_graph.py 
--input_type image_tensor 
--pipeline_config_path /home/chanchihang/Special_Topic/ssd_inception_v2_coco.config 
--trained_checkpoint_prefix /home/chanchihang/Special_Topic/models/research/inceptionV2_Training/model.ckpt-20000 
--output_directory /home/chanchihang/Special_Topic/models/research/inceptionV2_Training/

然后我尝试将mvNCCompiler的model.ckpt.meta转换为movidius图形,并在下面显示错误

mvNCCompile model.ckpt.meta -in=image_tensor -on=detection_scores -is 224 224 -o test.graph


Traceback (most recent call last):
   File "/usr/local/bin/mvNCCompile", line 118, in <module>
    create_graph(args.network, args.inputnode, args.outputnode, args.outfile, args.nshaves, args.inputsize, args.weights)
  File "/usr/local/bin/mvNCCompile", line 104, in create_graph
    net = parse_tensor(args, myriad_config)
  File "/usr/local/bin/ncsdk/Controllers/TensorFlowParser.py", line 259, in parse_tensor
    input_data = np.random.uniform(0, 1, shape)
  File "mtrand.pyx", line 961, in numpy.random.mtrand.RandomState.uniform
  File "common.pyx", line 557, in numpy.random.common.cont
TypeError: 'NoneType' object cannot be interpreted as an integer

0 个答案:

没有答案