Skip to content

Commit f1b9479

Browse files
authored
AC: fix for unknown lauouts pattern (#2934)
1 parent b56806f commit f1b9479

File tree

1 file changed

+8
-3
lines changed
  • tools/accuracy_checker/openvino/tools/accuracy_checker/launcher

1 file changed

+8
-3
lines changed

tools/accuracy_checker/openvino/tools/accuracy_checker/launcher/input_feeder.py

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -351,7 +351,8 @@ def _parse_inputs_config(self, inputs_entry, default_layout='NCHW', precisions_l
351351
value = re.compile(value) if not isinstance(value, int) else value
352352
non_constant_inputs_mapping[name] = value
353353
layout = input_.get('layout', layouts_info.get(name, default_layout))
354-
layouts[name] = LAYER_LAYOUT_TO_IMAGE_LAYOUT[layout]
354+
if layout in LAYER_LAYOUT_TO_IMAGE_LAYOUT:
355+
layouts[name] = LAYER_LAYOUT_TO_IMAGE_LAYOUT[layout]
355356
self.get_layer_precision(input_, name, precision_info, precisions)
356357

357358
all_config_inputs = (
@@ -364,9 +365,10 @@ def _parse_inputs_config(self, inputs_entry, default_layout='NCHW', precisions_l
364365
non_constant_inputs = not_config_inputs + config_non_constant_inputs
365366
if not_config_inputs and (precision_info or isinstance(precision_info, defaultdict)) or layouts_info:
366367
inputs_entry = self.provide_input_config_for_not_config(
367-
inputs_entry, precision_info, not_config_inputs, precisions, layouts_info
368+
inputs_entry, precision_info, not_config_inputs, precisions, layouts_info, layouts
368369
)
369370

371+
370372
return (
371373
constant_inputs,
372374
non_constant_inputs,
@@ -513,11 +515,14 @@ def get_layer_precision(input_config, input_name, precision_info, precisions):
513515
precisions[input_name] = input_precision
514516
return input_precision
515517

516-
def provide_input_config_for_not_config(self, inputs_entry, precision_info, not_config_inputs, precisions, layouts):
518+
def provide_input_config_for_not_config(
519+
self, inputs_entry, precision_info, not_config_inputs, precisions, layouts, layouts_mapping):
517520
for input_name in not_config_inputs:
518521
input_config = {'name': input_name, 'type': 'INPUT'}
519522
precision = self.get_layer_precision(input_config, input_name, precision_info, precisions)
520523
layout = layouts.get(input_name)
524+
if layout in LAYER_LAYOUT_TO_IMAGE_LAYOUT:
525+
layouts_mapping[input_name] = LAYER_LAYOUT_TO_IMAGE_LAYOUT[layout]
521526
input_config['layout'] = layout
522527
if precision is not None or layout is not None:
523528
inputs_entry.append(input_config)

0 commit comments

Comments
 (0)