AC: enable auto resize for NHWC layout (#3007)

* AC: enable auto resize for NHWC layout

* Update tools/accuracy_checker/openvino/tools/accuracy_checker/preprocessor/resize.py

Co-authored-by: Anna Grebneva <[email protected]>

Co-authored-by: Anna Grebneva <[email protected]>

Author: eaidova

tools/accuracy_checker/openvino/tools/accuracy_checker/launcher/openvino_launcher.py

@@ -814,7 +814,7 @@ def _fill_lstm_inputs(self, infer_outputs=None):
         for lstm_var, output_layer in self._lstm_inputs.items():
             layer_shape = parse_partial_shape(self.inputs[lstm_var].partial_shape)
             if infer_outputs and output_layer not in infer_outputs:
-                raise 'Output node with name {} not found'.format(output_layer)
+                raise ConfigError('Output node with name {} not found'.format(output_layer))
             input_data = infer_outputs[output_layer].reshape(layer_shape) if infer_outputs else np.zeros(
                 layer_shape, dtype=format_map[self.inputs[lstm_var].element_type.get_type_name()]
             )

tools/accuracy_checker/openvino/tools/accuracy_checker/preprocessor/resize.py

@@ -479,17 +479,22 @@ def __init__(self, config, name=None):
 
     def set_input_shape(self, input_shape):
         def is_image_input(shape):
-            return len(shape) == 4 and shape[1] in [1, 3, 4]
+            return len(shape) == 4 and (shape[1] in [1, 3, 4] or shape[-1] in [1, 3, 4])
+
+        def is_nhwc(shape):
+            return shape[-1] in [1, 3, 4]
         if input_shape is None:
             raise ConfigError('resize to input size impossible')
         image_inputs = [value for value in input_shape.values() if is_image_input(value)]
         if not image_inputs:
             raise ConfigError('image input is not detected')
         if len(image_inputs) == 1:
-            self.dst_height, self.dst_width = image_inputs[0][2:]
+            self.dst_height, self.dst_width = (
+                image_inputs[0][2:] if not is_nhwc(image_inputs[0]) else image_inputs[0][1:3]
+            )
         else:
-            self.dst_height = [im_input[2] for im_input in image_inputs]
-            self.dst_width = [im_input[3] for im_input in image_inputs]
+            self.dst_height = [im_input[2] if not is_nhwc(im_input) else im_input[1] for im_input in image_inputs]
+            self.dst_width = [im_input[3] if not is_nhwc(im_input) else im_input[2] for im_input in image_inputs]
 
     def process(self, image, annotation_meta=None):
         is_simple_case = not isinstance(image.data, list)  # otherwise -- pyramid, tiling, etc