Update convert_to_pascalVOC.py

convert_to_pascalVOC.py

@@ -31,3 +31,43 @@ 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)
+
+#######################################################################################################
+
+## for rename labels 
+import os
+import xml.etree.ElementTree as ET
+
+# Path to the folder containing XML annotation files
+annotations_folder = '/path/to/annotations_folder'
+
+# Dictionary for renaming labels: {old_label: new_label}
+label_mapping = {
+    'old_label_1': 'new_label_1',
+    'old_label_2': 'new_label_2',
+    # Add more label mappings as needed
+}
+
+# Function to rename labels in the XML file
+def rename_labels_in_xml(xml_file, label_mapping):
+    tree = ET.parse(xml_file)
+    root = tree.getroot()
+
+    # 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 back to file
+    tree.write(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)
+        rename_labels_in_xml(xml_path, label_mapping)
+
+print('Label renaming completed.')