Skip to content

Commit 9b1d5b7

Browse files
authored
AC: add missed dependency (openvinotoolkit#1204)
* AC: add missed dependency * fix input
1 parent af89f42 commit 9b1d5b7

File tree

4 files changed

+11
-3
lines changed

4 files changed

+11
-3
lines changed

ci/requirements-ac-test.txt

+1
Original file line numberDiff line numberDiff line change
@@ -27,3 +27,4 @@ yamlloader==0.5.5
2727
zipp==1.2.0 # via importlib-metadata
2828
pydicom==1.4.2
2929
tokenizers==0.7
30+
scikit-image==0.15.0

ci/requirements-ac.txt

+1
Original file line numberDiff line numberDiff line change
@@ -13,3 +13,4 @@ tqdm==4.45.0
1313
yamlloader==0.5.5
1414
pydicom==1.4.2
1515
tokenizers==0.7
16+
scikit-image==0.15.0

tools/accuracy_checker/accuracy_checker/launcher/dlsdk_launcher.py

+8-3
Original file line numberDiff line numberDiff line change
@@ -295,7 +295,7 @@ def inputs(self):
295295
if self.network is None:
296296
has_info = hasattr(self.exec_network, 'input_info')
297297
if not has_info:
298-
return self.exec_network.input
298+
return self.exec_network.inputs
299299
return OrderedDict([(name, data.input_data) for name, data in self.exec_network.input_info.items()])
300300
has_info = hasattr(self.network, 'input_info')
301301
if has_info:
@@ -326,11 +326,16 @@ def predict(self, inputs, metadata=None, **kwargs):
326326
if self._use_set_blob:
327327
has_info = hasattr(self.exec_network, 'input_info')
328328
for key, input_data in infer_inputs.items():
329-
ie_input_info = self.exec_network.input_info.input_data if has_info else self.exec_network.inputs
329+
if has_info:
330+
ie_input_info = OrderedDict([
331+
(name, data.input_data) for name, data in self.exec_network.input_info.items()
332+
])
333+
else:
334+
ie_input_info = self.exec_network.inputs
330335
layout = self._target_layout_mapping.get(key, ie_input_info[key].layout)
331336
tensor_desc = TensorDesc(
332337
ie_input_info[key].precision,
333-
ie_input_info[key].input_data.shape,
338+
ie_input_info[key].shape,
334339
layout
335340
)
336341
self.exec_network.requests[0].set_blob(key, Blob(tensor_desc, input_data))

tools/accuracy_checker/requirements.in

+1
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,7 @@ tqdm
33

44
# image reading and preprocessing
55
pillow>=2.6.1
6+
scikit-image
67

78
# reid
89
scikit-learn

0 commit comments

Comments
 (0)