From f6e0f5d56b8f4720d73f73e9b2379979bf789e83 Mon Sep 17 00:00:00 2001
From: Mohit Mehrotra <mohit.mehrotra@intel.com>
Date: Tue, 17 Mar 2026 14:27:37 -0700
Subject: [PATCH] add fix issue with shape for object detection

---
 .../src/models/object_detection_ssd_model.cpp | 25 ++++++++++++-------
 1 file changed, 16 insertions(+), 9 deletions(-)

diff --git a/openvino_wrapper_lib/src/models/object_detection_ssd_model.cpp b/openvino_wrapper_lib/src/models/object_detection_ssd_model.cpp
index f5471c56..b8ed438b 100644
--- a/openvino_wrapper_lib/src/models/object_detection_ssd_model.cpp
+++ b/openvino_wrapper_lib/src/models/object_detection_ssd_model.cpp
@@ -162,18 +162,25 @@ bool Models::ObjectDetectionSSDModel::updateLayerProperty(std::shared_ptr<ov::Mo
   ov::Shape input_dims = input_info_map[0].get_shape();
 
   ov::Layout tensor_layout = ov::Layout("NCHW");
-  ov::Layout expect_layout = ov::Layout("NHWC");
-  setInputHeight(input_dims[2]);
-  setInputWidth(input_dims[3]);
-  if (input_dims[1] == 3)
-    expect_layout = ov::Layout("NCHW");
-  else if (input_dims[3] == 3)
-    expect_layout = ov::Layout("NHWC");
-  else
+  ov::Layout model_layout = ov::Layout("NCHW");
+  if (input_dims[1] == 3) {
+    // NCHW: [N, C, H, W]
+    model_layout = ov::Layout("NCHW");
+    setInputHeight(input_dims[2]);
+    setInputWidth(input_dims[3]);
+  } else if (input_dims[3] == 3) {
+    // NHWC: [N, H, W, C]
+    model_layout = ov::Layout("NHWC");
+    setInputHeight(input_dims[1]);
+    setInputWidth(input_dims[2]);
+  } else {
     slog::warn << "unexpect input shape " << input_dims << slog::endl;
+    setInputHeight(input_dims[2]);
+    setInputWidth(input_dims[3]);
+  }
 
   input_info.tensor().set_element_type(ov::element::u8).set_layout(tensor_layout);
-  input_info.preprocess().convert_layout(expect_layout).resize(ov::preprocess::ResizeAlgorithm::RESIZE_LINEAR);
+  ppp.input(input_tensor_name_).model().set_layout(model_layout);
 
   slog::info << "Checking OUTPUTs for model " << getModelName() << slog::endl;
   auto outputs_info = model->outputs();