Skip to content

Commit 703df77

Browse files
committed
AC: fix custom evaluator for segnet
1 parent 7e9ea70 commit 703df77

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tools/accuracy_checker/openvino/tools/accuracy_checker/evaluators/custom_evaluators/segnet_background_matting.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -135,7 +135,7 @@ def set_input_and_output(self):
135135
with_prefix = input_blob.startswith(self.default_model_suffix + '_')
136136
if self.input_blob is None:
137137
self.input_blob = input_blob
138-
self.output_blob = next(iter(self.outputs)).get_any_name()
138+
self.output_blob = next(iter(self.outputs))
139139
if with_prefix != self.with_prefix:
140140
self.input_blob = generate_layer_name(self.input_blob, self.default_model_suffix, with_prefix)
141141
self.output_blob = postprocess_output_name(
@@ -144,7 +144,7 @@ def set_input_and_output(self):
144144
self.output_blob = postprocess_output_name(
145145
generate_layer_name(self.output_blob, self.default_model_suffix, with_prefix),
146146
self.outputs, additional_mapping=self.additional_output_mapping, raise_error=False)
147-
147+
self.adapter.output_blob = self.output_blob
148148
self.with_prefix = with_prefix
149149

150150
def load_network(self, network, launcher):

0 commit comments

Comments
 (0)