File tree Expand file tree Collapse file tree 1 file changed +5
-3
lines changed
demos/common/python/openvino/model_zoo/model_api/adapters Expand file tree Collapse file tree 1 file changed +5
-3
lines changed Original file line number Diff line number Diff line change @@ -121,16 +121,18 @@ def reshape_model(self, new_shape):
121
121
self .model .reshape (new_shape )
122
122
123
123
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 ()}
126
128
127
129
def infer_sync (self , dict_data ):
128
130
self .infer_request = self .async_queue [self .async_queue .get_idle_request_id ()]
129
131
self .infer_request .infer (dict_data )
130
132
return self .get_raw_result (self .infer_request )
131
133
132
134
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 ))
134
136
135
137
def set_callback (self , callback_fn ):
136
138
self .async_queue .set_callback (callback_fn )
You can’t perform that action at this time.
0 commit comments