Skip to content

Commit 7768864

Browse files
authored
Temp fix nifti_reader (#100)
1 parent 448b61a commit 7768864

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

monai/data/nifti_reader.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,6 @@
1111

1212
import numpy as np
1313
import nibabel as nib
14-
import random
1514

1615
from torch.utils.data import Dataset
1716
from torch.utils.data._utils.collate import np_str_obj_array_pattern
@@ -111,12 +110,15 @@ def __getitem__(self, index):
111110
seed = np.random.randint(2147483647)
112111

113112
if self.transform is not None:
114-
random.seed(seed)
113+
np.random.seed(seed)
115114
img = self.transform(img)
115+
random_sync_test = np.random.randint(2147483647)
116116

117117
if self.seg_transform is not None:
118-
random.seed(seed) # ensure randomized transforms roll the same values for segmentations as images
118+
np.random.seed(seed) # ensure randomized transforms roll the same values for segmentations as images
119119
seg = self.seg_transform(seg)
120+
seg_seed = np.random.randint(2147483647)
121+
assert(random_sync_test == seg_seed)
120122

121123
if self.image_only or meta_data is None:
122124
return img, seg

0 commit comments

Comments
 (0)