Skip to content

Commit 950d497

Browse files
authored
AC: fix ctc decoders output blob (#2990)
1 parent fea4edb commit 950d497

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tools/accuracy_checker/openvino/tools/accuracy_checker/adapters/text_recognition.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -87,7 +87,7 @@ def process(self, raw, identifiers, frame_meta):
8787
self.select_output_blob(raw)
8888
raw_output = self._extract_predictions(raw, frame_meta)
8989
self.select_output_blob(raw_output)
90-
output = raw_output[self.output_blob]
90+
output = raw_output[self.logits_output]
9191
# TBC -> BTC
9292
if output.shape[1] == len(identifiers):
9393
output = np.swapaxes(output, 0, 1)
@@ -229,7 +229,7 @@ def process(self, raw, identifiers=None, frame_meta=None):
229229
self.select_output_blob(raw)
230230
raw_output = self._extract_predictions(raw, frame_meta)
231231
self.select_output_blob(raw_output)
232-
output = raw_output[self.output_blob]
232+
output = raw_output[self.logits_output]
233233
preds_index = np.argmax(output, 2)
234234
preds_index = preds_index.transpose(1, 0)
235235

0 commit comments

Comments
 (0)