AC: fix segmentation masks in postprocessing (#2921)

Author: eaidova

tools/accuracy_checker/openvino/tools/accuracy_checker/annotation_converters/pascal_voc.py

@@ -13,7 +13,7 @@
 See the License for the specific language governing permissions and
 limitations under the License.
 """
-
+from pathlib import Path
 from ..config import PathField, BoolField
 from ..representation import DetectionAnnotation, SegmentationAnnotation
 from ..representation.segmentation_representation import GTMaskLoader
@@ -169,16 +169,16 @@ def find_images(self, image_id):
         else:
             image_root = self.image_dir
             mask_root = self.mask_dir
-        images = list(image_root.glob('{}.*'.format(image_id)))
+        images = list(Path(image_root).glob('{}.*'.format(image_id)))
         if not images:
             image_file = '{}.jpg'.format(relative_image_subdir + '/' + image_id if relative_image_subdir else image_id)
         else:
             image_file = images[0].name if not relative_image_subdir else relative_image_subdir + '/' + images[0].name
-        masks = list(mask_root.glob('{}.*'.format(image_id)))
+        masks = list(Path(mask_root).glob('{}.*'.format(image_id)))
         if not masks:
             mask_file = '{}.png'.format(relative_image_subdir + '/' + image_id if relative_image_subdir else image_id)
         else:
-            mask_file = images[0].name if not relative_image_subdir else relative_image_subdir + '/' + images[0].name
+            mask_file = masks[0].name if not relative_image_subdir else relative_image_subdir + '/' + masks[0].name
         return image_file, mask_file
 
 

tools/accuracy_checker/openvino/tools/accuracy_checker/postprocessor/extend_segmentation_mask.py

@@ -60,7 +60,13 @@ def process_image(self, annotation, prediction):
             height, width = annotation_mask.shape[-2:]
             if dst_width < width or dst_height < height:
                 warning('size for extending should be not less current mask size. resize operation will be applied')
-                annotation_.mask = self._resize(annotation_.mask, dst_height, dst_width, False)
+                pad_width = min(dst_width, width)
+                pad_height = min(dst_height, height)
+                pad = self.pad_func(dst_width, dst_height, pad_width, pad_height)
+                extended_mask = cv2.copyMakeBorder(
+                    annotation_mask, pad[0], pad[2], pad[1], pad[3], cv2.BORDER_CONSTANT, value=self.filling_label
+                )
+                annotation_.mask = self._resize(extended_mask, dst_height, dst_width, True)
                 continue