Skip to content

Commit 544689f

Browse files
authored
AC: fix multi sr adapter (openvinotoolkit#3434)
1 parent 60b47b7 commit 544689f

File tree

1 file changed

+8
-3
lines changed

1 file changed

+8
-3
lines changed

tools/accuracy_checker/openvino/tools/accuracy_checker/adapters/image_processing.py

+8-3
Original file line numberDiff line numberDiff line change
@@ -135,8 +135,9 @@ def additional_output_mapping(self):
135135
@additional_output_mapping.setter
136136
def additional_output_mapping(self, value):
137137
self._additional_output_mapping = value
138-
for adapter in self.target_mapping.values():
139-
adapter.additional_output_mapping = value
138+
if hasattr(self, '_per_target_adapters'):
139+
for adapter in self._per_target_adapters.values():
140+
adapter.additional_output_mapping = value
140141

141142
@classmethod
142143
def parameters(cls):
@@ -176,7 +177,11 @@ def configure(self):
176177
common_adapter_config = deepcopy(self.launcher_config)
177178
self._per_target_adapters = {}
178179
for key, output_name in self.target_mapping.items():
179-
self._per_target_adapters[key] = SuperResolutionAdapter(common_adapter_config, output_blob=output_name)
180+
self._per_target_adapters[key] = SuperResolutionAdapter(
181+
common_adapter_config, output_blob=output_name,
182+
additional_output_mapping=self.additional_output_mapping
183+
)
184+
180185

181186
def process(self, raw, identifiers=None, frame_meta=None):
182187
predictions = [{}] * len(identifiers)

0 commit comments

Comments
 (0)