@@ -120,7 +120,7 @@ def prepare_mask_and_masked_image(image, mask, height, width):
120
120
image = [image ]
121
121
if isinstance (image , list ) and isinstance (image [0 ], PIL .Image .Image ):
122
122
# 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 ]
124
124
image = [np .array (i .convert ("RGB" ))[None , :] for i in image ]
125
125
image = np .concatenate (image , axis = 0 )
126
126
elif isinstance (image , list ) and isinstance (image [0 ], np .ndarray ):
@@ -134,7 +134,7 @@ def prepare_mask_and_masked_image(image, mask, height, width):
134
134
mask = [mask ]
135
135
136
136
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 ]
138
138
mask = np .concatenate ([np .array (m .convert ("L" ))[None , None , :] for m in mask ], axis = 0 )
139
139
mask = mask .astype (np .float32 ) / 255.0
140
140
elif isinstance (mask , list ) and isinstance (mask [0 ], np .ndarray ):
0 commit comments