@@ -120,7 +120,7 @@ def prepare_mask_and_masked_image(image, mask, height, width):
120120 image = [image ]
121121 if isinstance (image , list ) and isinstance (image [0 ], PIL .Image .Image ):
122122 # resize all images w.r.t passed height an width
123- image = [i .resize ((width , height ), resample = PIL .Image .LANCZOS ) for i in image ]
123+ image = [i .resize ((width , height ), resample = PIL .Image .LANCZOS ) for i in image ]
124124 image = [np .array (i .convert ("RGB" ))[None , :] for i in image ]
125125 image = np .concatenate (image , axis = 0 )
126126 elif isinstance (image , list ) and isinstance (image [0 ], np .ndarray ):
@@ -134,7 +134,7 @@ def prepare_mask_and_masked_image(image, mask, height, width):
134134 mask = [mask ]
135135
136136 if isinstance (mask , list ) and isinstance (mask [0 ], PIL .Image .Image ):
137- mask = [i .resize ((width , height ), resample = PIL .Image .LANCZOS ) for i in mask ]
137+ mask = [i .resize ((width , height ), resample = PIL .Image .LANCZOS ) for i in mask ]
138138 mask = np .concatenate ([np .array (m .convert ("L" ))[None , None , :] for m in mask ], axis = 0 )
139139 mask = mask .astype (np .float32 ) / 255.0
140140 elif isinstance (mask , list ) and isinstance (mask [0 ], np .ndarray ):
0 commit comments