From be73aabffef02dd6bf694477f785d34c7f9c607d Mon Sep 17 00:00:00 2001
From: Hashim Khan <64767361+Hashim1999164@users.noreply.github.com>
Date: Thu, 13 Aug 2026 04:02:15 +0500
Subject: [PATCH] fix: rename nhwc_data_list to nchw_data_list after preprocess

_preprocess_images already converts to NCHW before returning, so the calibration reader should name the cached batch nchw_data_list.
---
 .../image_classification/cpu/resnet50_data_reader.py        | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/quantization/image_classification/cpu/resnet50_data_reader.py b/quantization/image_classification/cpu/resnet50_data_reader.py
index d07198f11..0909c3554 100644
--- a/quantization/image_classification/cpu/resnet50_data_reader.py
+++ b/quantization/image_classification/cpu/resnet50_data_reader.py
@@ -46,16 +46,16 @@ def __init__(self, calibration_image_folder: str, model_path: str):
         (_, _, height, width) = session.get_inputs()[0].shape
 
         # Convert image to input data
-        self.nhwc_data_list = _preprocess_images(
+        self.nchw_data_list = _preprocess_images(
             calibration_image_folder, height, width, size_limit=0
         )
         self.input_name = session.get_inputs()[0].name
-        self.datasize = len(self.nhwc_data_list)
+        self.datasize = len(self.nchw_data_list)
 
     def get_next(self):
         if self.enum_data is None:
             self.enum_data = iter(
-                [{self.input_name: nhwc_data} for nhwc_data in self.nhwc_data_list]
+                [{self.input_name: nchw_data} for nchw_data in self.nchw_data_list]
             )
         return next(self.enum_data, None)