diff --git a/mindspore/ccsrc/minddata/dataset/kernels/image/normalize_op.cc b/mindspore/ccsrc/minddata/dataset/kernels/image/normalize_op.cc
index 12790841b08..c286bab74ac 100644
--- a/mindspore/ccsrc/minddata/dataset/kernels/image/normalize_op.cc
+++ b/mindspore/ccsrc/minddata/dataset/kernels/image/normalize_op.cc
@@ -29,7 +29,7 @@ namespace mindspore {
 namespace dataset {
 NormalizeOp::NormalizeOp(const std::vector<float> &mean, const std::vector<float> &std) : mean_(mean), std_(std) {
   // pre-calculate normalized mean to be used later in each Compute
-  for (int8_t i = 0; i < mean.size(); i++) {
+  for (int64_t i = 0; i < mean.size(); i++) {
     mean_[i] = mean_[i] / std_[i];
   }
 }
diff --git a/tests/ut/python/dataset/test_normalizeOp.py b/tests/ut/python/dataset/test_normalizeOp.py
index 4770ae4f847..95c14ff5641 100644
--- a/tests/ut/python/dataset/test_normalizeOp.py
+++ b/tests/ut/python/dataset/test_normalizeOp.py
@@ -367,6 +367,9 @@ def test_multiple_channels():
     util_test(np.ones(shape=[20, 45, 4]) * 1.3, mean=[0.5, 0.6, 0.7, 0.8], std=[0.1, 0.2, 0.3, 0.4])
     util_test(np.ones(shape=[2, 2]), mean=[0.5], std=[0.1])
     util_test(np.ones(shape=[2, 2, 5]), mean=[0.5], std=[0.1])
+    util_test(np.ones(shape=[6, 6, 129]), mean=[0.5]*129, std=[0.1]*129)
+    util_test(np.ones(shape=[6, 6, 129]), mean=[0.5], std=[0.1])
+
 
 
 if __name__ == "__main__":