Как я могу найти узел ввода-вывода frozen_inference_graph.pb? - PullRequest
Новая
       65

Как я могу найти узел ввода-вывода frozen_inference_graph.pb?

0 голосов
/ 08 ноября 2019

Я уже обучил модель по модели 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/

, затем я попытался преобразовать model.ckpt.meta с помощью mvNCCompiler в график 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
...