Fixing Pascal VOC to BigML-COCO annotations translator

HISTORY.rst

@@ -3,6 +3,11 @@
 History
 -------
 
+5.8.1 (2024-05-31)
+~~~~~~~~~~~~~~~~~~
+
+- Fixing Pascal VOC to BigML-COCO translator.
+
 5.8.0 (2024-04-05)
 ~~~~~~~~~~~~~~~~~~
 

bigmler/__init__.py

@@ -1,2 +1,2 @@
 # -*- coding: utf-8 -*-
-__version__ = '5.8.0'
+__version__ = '5.8.1'

bigmler/processing/annotations.py

@@ -173,10 +173,9 @@ def bigml_coco_file(args, session_file):
 
 def get_image_info(annotation_root):
     """Returns the basic image descriptors"""
+    filename = annotation_root.findtext('filename')
     path = annotation_root.findtext('path')
-    if path is None:
-        filename = annotation_root.findtext('filename')
-    else:
+    if filename is None and path is not None:
         filename = os.path.basename(path)
     img_name = os.path.basename(filename)
     img_id = os.path.splitext(img_name)[0]
@@ -530,6 +529,7 @@ def voc_to_cocojson(voc_dir, args, session_file):
             if args.images_dir is not None:
                 base_dir = os.path.join(base_dir, args.images_dir)
             image_filename_base = os.path.join(folder, img_info['filename'])
+            filename = image_filename_base
             if args.images_dir is not None:
                 filename = os.path.join(base_dir, image_filename_base)
 

data/images/VOC_annotations/000004.xml

@@ -1,6 +1,7 @@
 <annotation>
 	<folder>f1</folder>
 	<filename>fruits1b.png</filename>
+	<path>C:\fruits1b.png</path>
 	<source>
 		<database>The VOC2007 Database</database>
 		<annotation>PASCAL VOC2007</annotation>