AC: support different image types in voc segmentation (#2920)

* AC: support different image types in voc segmentation

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

Author: eaidova

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

@@ -139,8 +139,7 @@ def convert(self, check_content=False, progress_callback=None, progress_interval
         images_set = read_txt(self.image_set_file)
         num_iterations = len(images_set)
         for image_id, image in enumerate(images_set):
-            image_file = '{}.jpg'.format(image)
-            mask_file = '{}.png'.format(image)
+            image_file, mask_file = self.find_images(image)
             annotation = SegmentationAnnotation(image_file, mask_file, mask_loader=GTMaskLoader.SCIPY)
             annotations.append(annotation)
             if check_content:
@@ -161,6 +160,28 @@ def convert(self, check_content=False, progress_callback=None, progress_interval
 
         return ConverterReturn(annotations, meta, content_check_errors)
 
+    def find_images(self, image_id):
+        relative_image_subdir = ''
+        if '/' in image_id:
+            relative_image_subdir, image_id =image_id.rsplit('/', 1)
+            image_root = self.image_dir / relative_image_subdir
+            mask_root = self.mask_dir / relative_image_subdir
+        else:
+            image_root = self.image_dir
+            mask_root = self.mask_dir
+        images = list(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)))
+        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
+        return image_file, mask_file
+
+
     @staticmethod
     def read_labelmap(input_file):
         label_map = {}