voc metadata corrected

Author: agyurjin

superannotate/input_converters/converters/voc_converters/voc_helper.py

@@ -39,3 +39,14 @@ def _iou(bbox1, bbox2):
         (xmax1 - xmin1) * (ymax1 - ymin1) + (xmax2 - xmin2) *
         (ymax2 - ymin2) - x * y
     )
+
+
+def _get_image_shape_from_xml(file_path):
+    with open(os.path.splitext(file_path)[0] + ".xml") as f:
+        tree = ET.parse(f)
+
+    size = tree.find('size')
+    width = int(size.find('width').text)
+    height = int(size.find('height').text)
+
+    return height, width

superannotate/input_converters/converters/voc_converters/voc_to_sa_pixel.py

@@ -3,7 +3,7 @@
 
 from ....common import blue_color_generator, hex_to_rgb, write_to_json
 from ..sa_json_helper import _create_pixel_instance, _create_sa_json
-from .voc_helper import _get_voc_instances_from_xml, _iou
+from .voc_helper import _get_voc_instances_from_xml, _iou, _get_image_shape_from_xml
 
 
 def _generate_polygons(object_mask_path):
@@ -61,7 +61,7 @@ def _generate_instances(polygon_instances, voc_instances, bluemask_colors):
         attributes = voc_instances[ind][0][class_name]
         instances.append(
             {
-                "className": voc_instances[ind][0],
+                "className": class_name,
                 "polygon": polygon,
                 "bbox": voc_instances[ind][1],
                 "blue_color": bluemask_colors[i],
@@ -100,7 +100,10 @@ def voc_instance_segmentation_to_sa_pixel(voc_root, output_dir):
             sa_instances.append(sa_obj)
 
         file_name = "%s.jpg___pixel.json" % (filename.stem)
-        sa_metadata = {'name': filename.stem}
+        height, width = _get_image_shape_from_xml(
+            annotation_dir / filename.name
+        )
+        sa_metadata = {'name': filename.stem, 'height': height, 'width': width}
         sa_json = _create_sa_json(sa_instances, sa_metadata)
         write_to_json(output_dir / file_name, sa_json)
 

superannotate/input_converters/converters/voc_converters/voc_to_sa_vector.py

@@ -4,7 +4,7 @@
 
 from ....common import write_to_json
 from ..sa_json_helper import _create_sa_json, _create_vector_instance
-from .voc_helper import _get_voc_instances_from_xml, _iou
+from .voc_helper import _get_voc_instances_from_xml, _iou, _get_image_shape_from_xml
 
 
 def _generate_polygons(object_mask_path):
@@ -99,7 +99,10 @@ def voc_instance_segmentation_to_sa_vector(voc_root, output_dir):
             sa_instances.append(sa_obj)
 
         file_name = "%s.jpg___objects.json" % filename.stem
-        sa_metadata = {'name': filename.stem}
+        height, width = _get_image_shape_from_xml(
+            annotation_dir / filename.name
+        )
+        sa_metadata = {'name': filename.stem, 'height': height, 'width': width}
         sa_json = _create_sa_json(sa_instances, sa_metadata)
         write_to_json(output_dir / file_name, sa_json)
     return classes
@@ -125,7 +128,10 @@ def voc_object_detection_to_sa_vector(voc_root, output_dir):
             sa_instances.append(sa_obj)
 
         file_name = "%s.jpg___objects.json" % filename.stem
-        sa_metadata = {'name': filename.stem}
+        height, width = _get_image_shape_from_xml(
+            annotation_dir / filename.name
+        )
+        sa_metadata = {'name': filename.stem, 'height': height, 'width': width}
         sa_json = _create_sa_json(sa_instances, sa_metadata)
         write_to_json(output_dir / file_name, sa_json)
     return classes