def _get_ie_output_blob_name(self, outputs: Dict,
ng_result: result) -> str:
if len(self.results) == 1:
return next(iter(outputs.keys()))
else:
prev_layer_output = ng_result.input(0).get_source_output()
return get_ie_output_name(prev_layer_output)
def _get_ie_output_blob_name(self, outputs: Dict, ng_result: result) -> str:
if len(self.results) == 1:
return next(iter(outputs.keys()))
else:
prev_layer = ng_result.input(0).get_source_output()
out_name = prev_layer.get_node().get_friendly_name()
if prev_layer.get_node().get_output_size() != 1:
out_name += "." + str(prev_layer.get_index())
return out_name
def __get_ie_output_blob_buffer(self, output_blobs: Dict[str, Blob],
ng_result: result) -> np.ndarray:
if len(self.results) == 1:
return next(iter(output_blobs.values())).buffer
else:
prev_layer = ng_result.input(0).get_source_output()
out_name = prev_layer.get_node().get_friendly_name()
if prev_layer.get_node().get_output_size() != 1:
out_name += "." + str(prev_layer.get_index())
return output_blobs[out_name].buffer