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

我已经通过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
snoopy717 回答:如何找到Frozen_inference_graph.pb输入和输出节点?

暂时没有好的解决方案,如果你有好的解决方案,请发邮件至:iooj@foxmail.com
本文链接:https://www.f2er.com/3135855.html

大家都在问