Update convert_to_pascalVOC.py

convert_to_pascalVOC.py

@@ -59,4 +59,23 @@ def rename_labels_in_xml(xml_file, new_xml_file, label_mapping):
     tree = ET.parse(xml_file)
     root = tree.getroot()
 
-    # Iterate over each object
+    # Iterate over each object in the XML file
+    for obj in root.findall('object'):
+        label = obj.find('name').text
+        # If label matches any in the label_mapping dictionary, rename it
+        if label in label_mapping:
+            print(f'Renaming label "{label}" to "{label_mapping[label]}" in {xml_file}')
+            obj.find('name').text = label_mapping[label]
+
+    # Write the updated XML to the new location
+    tree.write(new_xml_file)
+
+# Loop through all XML files in the annotations folder
+for filename in os.listdir(annotations_folder):
+    if filename.endswith('.xml'):
+        xml_path = os.path.join(annotations_folder, filename)
+        new_xml_path = os.path.join(new_annotations_folder, filename)  # New file path
+        rename_labels_in_xml(xml_path, new_xml_path, label_mapping)
+
+print('Label renaming and saving completed.')
+