@@ -92,7 +92,7 @@ def process(self, image, annotation_meta=None):
92
92
mask = 1 - mask
93
93
if self .concat_mask :
94
94
image .identifier = image .identifier [0 ]
95
- image .data = np .concatenate ([img , mask ], axis = 2 )
95
+ image .data = np .concatenate ([img , mask * 255 ], axis = 2 )
96
96
return image
97
97
image .data = [img , mask ]
98
98
identifier = image .identifier [0 ]
@@ -151,7 +151,7 @@ def process(self, image, annotation_meta=None):
151
151
mask = 1 - mask
152
152
if self .concat_mask :
153
153
image .identifier = image .identifier [0 ]
154
- image .data = np .concatenate ([img , mask ], axis = 2 )
154
+ image .data = np .concatenate ([img , mask * 255 ], axis = 2 )
155
155
return image
156
156
image .data = [img , mask ]
157
157
identifier = image .identifier [0 ]
@@ -207,7 +207,7 @@ def process(self, image, annotation_meta=None):
207
207
img = img * (1 - mask ) + 255 * mask
208
208
if self .concat_mask :
209
209
image .identifier = image .identifier [0 ]
210
- image .data = np .concatenate ([img , mask ], axis = 2 )
210
+ image .data = np .concatenate ([img , mask * 255 ], axis = 2 )
211
211
return image
212
212
image .data = [img , mask ]
213
213
identifier = image .identifier [0 ]
0 commit comments