Skip to content

Commit 25c5917

Browse files
author
Anna Grebneva
authored
Fixed output verification in aocr adapter (#3229)
1 parent 432b0f8 commit 25c5917

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

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

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -373,10 +373,16 @@ def configure(self):
373373
self.labels = self.get_value_from_config('labels')
374374
self.eos_index = self.get_value_from_config('eos_index')
375375
self.lower_case = self.get_value_from_config('to_lower_case')
376+
self.outputs_verified = False
377+
378+
def select_output_blob(self, outputs):
379+
self.output_blob = self.check_output_name(self.output_blob, outputs)
380+
self.outputs_verified = True
376381

377382
def process(self, raw, identifiers, frame_meta):
378383
raw_out = self._extract_predictions(raw, frame_meta)
379-
self.select_output_blob(raw_out)
384+
if not self.outputs_verified:
385+
self.select_output_blob(raw_out)
380386
result = []
381387
if isinstance(raw_out[self.output_blob], bytes):
382388
out_str = raw_out[self.output_blob].decode('iso-8859-1')

0 commit comments

Comments
 (0)