Skip to content

Commit 63db5f7

Browse files
authored
demos: copy array for async (#3591)
Ticket 82580
1 parent 16fc96a commit 63db5f7

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

demos/common/python/openvino/model_zoo/model_api/adapters/openvino_adapter.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -121,16 +121,18 @@ def reshape_model(self, new_shape):
121121
self.model.reshape(new_shape)
122122

123123
def get_raw_result(self, request):
124-
raw_result = {key: request.get_tensor(key).data[:] for key in self.get_output_layers().keys()}
125-
return raw_result
124+
return {key: request.get_tensor(key).data for key in self.get_output_layers()}
125+
126+
def copy_raw_result(self, request):
127+
return {key: request.get_tensor(key).data.copy() for key in self.get_output_layers()}
126128

127129
def infer_sync(self, dict_data):
128130
self.infer_request = self.async_queue[self.async_queue.get_idle_request_id()]
129131
self.infer_request.infer(dict_data)
130132
return self.get_raw_result(self.infer_request)
131133

132134
def infer_async(self, dict_data, callback_data) -> None:
133-
self.async_queue.start_async(dict_data, (self.get_raw_result, callback_data))
135+
self.async_queue.start_async(dict_data, (self.copy_raw_result, callback_data))
134136

135137
def set_callback(self, callback_fn):
136138
self.async_queue.set_callback(callback_fn)

0 commit comments

Comments
 (0)