Find wrong labels in dataset

find_wrong_labels.py

#from importlib.metadata import files
import xml.etree.ElementTree as ET
import glob


# Sort through a dataset's xml files and print out file names that contain labels which are invalid
valid_labels = ["frodo","baggins"]

# Loop through all xml files
files = glob.glob('/home/isaac/Desktop/work_area/working_on/*.xml')

for file in files:
    tree = ET.parse(file)
    root = tree.getroot()
    
    # Check to make sure that the .xml file has an object
    objects = root.findall('object')

    if objects:
        for object in objects:
            label_name = object.find('name')
            if label_name.text not in valid_labels:
                print(file)