Update convert_to_pascalVOC.py

convert_to_pascalVOC.py

@@ -0,0 +1,33 @@
+import xml.etree.ElementTree as ET
+
+def convert_bbox_to_int(xml_file, output_xml_file):
+    # Parse the XML file
+    tree = ET.parse(xml_file)
+    root = tree.getroot()
+
+    # Loop through all 'object' elements to find bboxes
+    for obj in root.findall('object'):
+        # Find the 'bndbox' element which contains the bbox info
+        bndbox = obj.find('bndbox')
+        if bndbox is not None:
+            # Get the coordinates as float and convert them to int
+            xmin = int(float(bndbox.find('xmin').text))
+            ymin = int(float(bndbox.find('ymin').text))
+            xmax = int(float(bndbox.find('xmax').text))
+            ymax = int(float(bndbox.find('ymax').text))
+
+            # Update the XML with the integer values
+            bndbox.find('xmin').text = str(xmin)
+            bndbox.find('ymin').text = str(ymin)
+            bndbox.find('xmax').text = str(xmax)
+            bndbox.find('ymax').text = str(ymax)
+
+    # Save the modified XML to the output file
+    tree.write(output_xml_file)
+    print(f"Converted bounding boxes saved to {output_xml_file}")
+
+# Example usage:
+input_xml_file = 'path/to/your/input_annotation.xml'
+output_xml_file = 'path/to/your/output_annotation.xml'
+
+convert_bbox_to_int(input_xml_file, output_xml_file)