Skip to content

Commit 12d782d

Browse files
authored
AC: fix out mapping for multi stage adapters (#3428)
1 parent aa956bc commit 12d782d

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

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

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -128,6 +128,16 @@ class MultiSuperResolutionAdapter(Adapter):
128128
__provider__ = 'multi_super_resolution'
129129
prediction_types = (SuperResolutionPrediction, )
130130

131+
@property
132+
def additional_output_mapping(self):
133+
return getattr(self, '_additional_output_mapping', None)
134+
135+
@additional_output_mapping.setter
136+
def additional_output_mapping(self, value):
137+
self._additional_output_mapping = value
138+
for adapter in self.target_mapping.values():
139+
adapter.additional_output_mapping = value
140+
131141
@classmethod
132142
def parameters(cls):
133143
parameters = super().parameters()

0 commit comments

Comments
 (0)