File tree Expand file tree Collapse file tree 1 file changed +4
-4
lines changed
monai/transforms/intensity Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Original file line number Diff line number Diff line change @@ -1699,24 +1699,24 @@ def randomize(self, img_size: Sequence[int]) -> None:
1699
1699
1700
1700
def __call__ (self , img : np .ndarray ):
1701
1701
self .randomize (img .shape [1 :])
1702
+ ret = img
1702
1703
if self ._do_transform :
1703
1704
fill_value = (img .min (), img .max ()) if self .fill_value is None else self .fill_value
1704
1705
1705
1706
if self .dropout_holes :
1706
1707
for h in self .hole_coords :
1707
1708
if isinstance (fill_value , (tuple , list )):
1708
- img [h ] = self .R .uniform (fill_value [0 ], fill_value [1 ], size = img [h ].shape )
1709
+ ret [h ] = self .R .uniform (fill_value [0 ], fill_value [1 ], size = img [h ].shape )
1709
1710
else :
1710
- img [h ] = fill_value
1711
- return img
1711
+ ret [h ] = fill_value
1712
1712
else :
1713
1713
if isinstance (fill_value , (tuple , list )):
1714
1714
ret = self .R .uniform (fill_value [0 ], fill_value [1 ], size = img .shape ).astype (img .dtype )
1715
1715
else :
1716
1716
ret = np .full_like (img , fill_value )
1717
1717
for h in self .hole_coords :
1718
1718
ret [h ] = img [h ]
1719
- return ret
1719
+ return ret
1720
1720
1721
1721
1722
1722
class HistogramNormalize (Transform ):
You can’t perform that action at this time.
0 commit comments