Skip to content

Commit 40f4738

Browse files
authored
AC: fix mask scaling in concat mode (#3515)
1 parent 143d778 commit 40f4738

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

tools/accuracy_checker/openvino/tools/accuracy_checker/preprocessor/inpainting_preprocessor.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -92,7 +92,7 @@ def process(self, image, annotation_meta=None):
9292
mask = 1 - mask
9393
if self.concat_mask:
9494
image.identifier = image.identifier[0]
95-
image.data = np.concatenate([img, mask], axis=2)
95+
image.data = np.concatenate([img, mask * 255], axis=2)
9696
return image
9797
image.data = [img, mask]
9898
identifier = image.identifier[0]
@@ -151,7 +151,7 @@ def process(self, image, annotation_meta=None):
151151
mask = 1 - mask
152152
if self.concat_mask:
153153
image.identifier = image.identifier[0]
154-
image.data = np.concatenate([img, mask], axis=2)
154+
image.data = np.concatenate([img, mask * 255], axis=2)
155155
return image
156156
image.data = [img, mask]
157157
identifier = image.identifier[0]
@@ -207,7 +207,7 @@ def process(self, image, annotation_meta=None):
207207
img = img * (1 - mask) + 255 * mask
208208
if self.concat_mask:
209209
image.identifier = image.identifier[0]
210-
image.data = np.concatenate([img, mask], axis=2)
210+
image.data = np.concatenate([img, mask * 255], axis=2)
211211
return image
212212
image.data = [img, mask]
213213
identifier = image.identifier[0]

0 commit comments

Comments
 (0)