File tree Expand file tree Collapse file tree 1 file changed +5
-3
lines changed Expand file tree Collapse file tree 1 file changed +5
-3
lines changed Original file line number Diff line number Diff line change 11
11
12
12
import numpy as np
13
13
import nibabel as nib
14
- import random
15
14
16
15
from torch .utils .data import Dataset
17
16
from torch .utils .data ._utils .collate import np_str_obj_array_pattern
@@ -111,12 +110,15 @@ def __getitem__(self, index):
111
110
seed = np .random .randint (2147483647 )
112
111
113
112
if self .transform is not None :
114
- random .seed (seed )
113
+ np . random .seed (seed )
115
114
img = self .transform (img )
115
+ random_sync_test = np .random .randint (2147483647 )
116
116
117
117
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
119
119
seg = self .seg_transform (seg )
120
+ seg_seed = np .random .randint (2147483647 )
121
+ assert (random_sync_test == seg_seed )
120
122
121
123
if self .image_only or meta_data is None :
122
124
return img , seg
You can’t perform that action at this time.
0 commit comments