diff --git a/src/otx/core/data/dataset/segmentation.py b/src/otx/core/data/dataset/segmentation.py
index 8f9d8f8fe7..ff61871890 100644
--- a/src/otx/core/data/dataset/segmentation.py
+++ b/src/otx/core/data/dataset/segmentation.py
@@ -1,4 +1,4 @@
-# Copyright (C) 2023 Intel Corporation
+# Copyright (C) 2023-2025 Intel Corporation
 # SPDX-License-Identifier: Apache-2.0
 #
 """Module for OTXSegmentationDataset."""
@@ -179,6 +179,7 @@ def __init__(
             image_color_channel,
             stack_images,
             to_tv_image,
+            data_format=data_format,
         )
 
         if self.has_polygons: