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