Copy corresponding xml and jpg files(predicated on labels)

copy_xml_and_jpg.py

from re import T
import os
import subprocess

from glob import glob

import xml.etree.ElementTree as ET
labels = [
"some",
"labels",
"here",
"hia"]

image_dir = "/directory/to/copy/images/from/"
files = glob('/directory/with/xml/files/*.xml')
# annotations_folder and image_folder can point to the same folder
annotations_folder = "output_annotations_folder"
image_folder = "output_images_folder"
for file in files:
    tree = ET.parse(file)
    root = tree.getroot()
    objects = root.findall('object')

    modified_file_id="'"+file+"'" # Modification because the filenames have spaces in them

    filename = root.find('filename').text
    modified_image_id="'"+image_dir+filename+"'" # Same reason there is a modification here

    for object in objects:
        label = object.find('name')

        if(label.text in labels):
            os.system('cp {} {}'.format(modified_file_id,annotations_folder))
            os.system('cp {} {}'.format(modified_image_id,image_folder))
            break;