File tree Expand file tree Collapse file tree 1 file changed +6
-1
lines changed
tools/accuracy_checker/openvino/tools/accuracy_checker/launcher Expand file tree Collapse file tree 1 file changed +6
-1
lines changed Original file line number Diff line number Diff line change @@ -495,19 +495,24 @@ def _log_versions(self):
495
495
def _create_network (self , input_shapes = None ):
496
496
model_path = Path (self ._model )
497
497
compiled_model = model_path .suffix == '.blob'
498
+ self .out_tensor_name_to_node = {}
498
499
if compiled_model :
499
500
self .network = None
500
501
with open (str (self ._model ), 'rb' ) as f : #pylint:disable=unspecified-encoding
501
502
self .exec_network = self .ie_core .import_model (io .BytesIO (f .read ()), self ._device )
502
503
self .original_outputs = self .exec_network .outputs
503
504
model_batch = self ._get_model_batch_size ()
504
505
self ._batch = model_batch if model_batch is not None else 1
506
+ for out in self .original_outputs :
507
+ if not out .names :
508
+ continue
509
+ for name in out .names :
510
+ self .out_tensor_name_to_node [name ] = out .get_node ().friendly_name
505
511
return
506
512
if self ._weights is None and self ._model .suffix != '.onnx' :
507
513
self ._weights = model_path .parent / (model_path .name .split (model_path .suffix )[0 ] + '.bin' )
508
514
self .network = self .read_network (self ._model , self ._weights )
509
515
self .original_outputs = self .network .outputs
510
- self .out_tensor_name_to_node = {}
511
516
for out in self .original_outputs :
512
517
if not out .names :
513
518
continue
You can’t perform that action at this time.
0 commit comments