github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-04-29 03:00:14 -04:00
Code Issues Packages Projects Releases Wiki Activity

Revert "reduce img and lbl sizes by half for KiTS19 dataset tests"

Browse Source 
This reverts commit d115b0c664.
This commit is contained in:
Francis Lata
2025-01-15 03:28:57 -08:00
parent d115b0c664
commit 36410069f2
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
test/external/external_test_datasets.py vendored
View File

@@ -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}")