From 16a5611c2d16afce6ba0e17669645be522f56996 Mon Sep 17 00:00:00 2001
From: Chandraveer <chandraveersingh6377@gmail.com>
Date: Thu, 5 Feb 2026 15:36:24 +0530
Subject: [PATCH] Fix incorrect dimension handling for grayscale images in
 base_loader()

---
 R/folder-dataset.R | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/R/folder-dataset.R b/R/folder-dataset.R
index e7409239..b8cdc698 100644
--- a/R/folder-dataset.R
+++ b/R/folder-dataset.R
@@ -153,7 +153,7 @@ base_loader <- function(path) {
   if (length(dim(img)) == 2)
     img <- abind::abind(img, img, img, along = 3)
   else if (length(dim(img)) == 3 && dim(img)[1] == 1)
-    img <- abind::abind(img, img, img, along = 1) %>% aperm(c(3,1,2))
+    img <- abind::abind(img[1,,], img[1,,], img[1,,], along = 3)
 
   # all readers default to channel last
   img