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 1111
1212import numpy as np
1313import nibabel as nib
14- import random
1514
1615from torch .utils .data import Dataset
1716from 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
You can’t perform that action at this time.
0 commit comments