From 74883d83730212bf87763b018943dacaa41d3357 Mon Sep 17 00:00:00 2001 From: SHEN CHUANFU <77523610+chuanfushen@users.noreply.github.com> Date: Wed, 24 Sep 2025 21:17:51 +0800 Subject: [PATCH] Update pretreatment_SUSTech1K.py Fix RGB_image transposed shape --- datasets/SUSTech1K/pretreatment_SUSTech1K.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/datasets/SUSTech1K/pretreatment_SUSTech1K.py b/datasets/SUSTech1K/pretreatment_SUSTech1K.py index e379ffd..3f2e192 100644 --- a/datasets/SUSTech1K/pretreatment_SUSTech1K.py +++ b/datasets/SUSTech1K/pretreatment_SUSTech1K.py @@ -54,8 +54,8 @@ def imgs2pickle(img_groups: Tuple, output_path: Path, img_size: int = 64, verbos HWs = [img.shape[:2] for img in imgs] # transpose to (C, H W) data = [cv2.resize(img, (img_size, img_size), interpolation=cv2.INTER_CUBIC) for img in imgs] - imgs = [img.transpose(2, 0, 1) for img in imgs] - data = np.asarray(data) + imgs = [img.transpose(2, 0, 1) for img in data] + data = np.asarray(imgs) HWs = np.asarray(HWs) elif modality == 'Sils_raw': sils = [cv2.imread(sil, cv2.IMREAD_GRAYSCALE) for sil in data_files] @@ -218,4 +218,4 @@ if __name__ == '__main__': for k, v in args.__dict__.items(): logging.debug(f'{k}: {v}') - pretreat(input_path=Path(args.input_path), output_path=Path(args.output_path), img_size=args.img_size, workers=args.n_workers, verbose=args.verbose, dataset=args.dataset) \ No newline at end of file + pretreat(input_path=Path(args.input_path), output_path=Path(args.output_path), img_size=args.img_size, workers=args.n_workers, verbose=args.verbose, dataset=args.dataset)