From 36410069f24ff0f3a225b6638c344ded3eee5c24 Mon Sep 17 00:00:00 2001 From: Francis Lata Date: Wed, 15 Jan 2025 03:28:57 -0800 Subject: [PATCH] Revert "reduce img and lbl sizes by half for KiTS19 dataset tests" This reverts commit d115b0c66470aae65d5110886c0b80285a05bf39. --- test/external/external_test_datasets.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/external/external_test_datasets.py b/test/external/external_test_datasets.py index bf1b4d30c3..0cd0fceaa8 100644 --- a/test/external/external_test_datasets.py +++ b/test/external/external_test_datasets.py @@ -28,7 +28,7 @@ class TestKiTS19Dataset(ExternalTestDatasets): def _create_samples(self, val, num_samples=2): self._set_seed() - img, lbl = np.random.rand(95, 196, 196).astype(np.float32), np.random.randint(0, 100, size=(95, 196, 196)).astype(np.uint8) + img, lbl = np.random.rand(190, 392, 392).astype(np.float32), np.random.randint(0, 100, size=(190, 392, 392)).astype(np.uint8) img, lbl = nib.Nifti1Image(img, np.eye(4)), nib.Nifti1Image(lbl, np.eye(4)) dataset = "val" if val else "train" preproc_pth = Path(tempfile.gettempdir() + f"/{dataset}")