According to the category information and coordinate information of the XML file, the category area corresponding to the image is pulled out and stored in the folder.

"""  according to xml Category information and location coordinate information of the file , The corresponding category object crop come out , And save it into a folder named after the category information   Purpose ： It is convenient to recheck the drawing quality after drawing .  Input ： Folder containing pictures imgs, contain xml Information folder xmls  Output ： Folder crop_img,crop_img A subfolder named after the label category under , as well as crop The children of  """

import cv2
import os
import xml.etree.ElementTree as ET

xml_file_dir = "./xmls"
img_file_dir = "./imgs"

xmls = os.listdir(xml_file_dir)
for xml in xmls:
    #  analysis xml
    tree = ET.parse(os.path.join(xml_file_dir,xml))
    root = tree.getroot()

    img_name = root.find("filename").text
    img = cv2.imread(os.path.join(img_file_dir,img_name))

    objects = root.findall('object')
    for i, obj in enumerate(objects):
        id = 0
        label = obj.find('name').text
        if label == ":":
            label = "colon"

        bb = obj.find('bndbox')
        xmin = bb.find('xmin').text
        ymin = bb.find('ymin').text
        xmax = bb.find('xmax').text
        ymax = bb.find('ymax').text

        #  preservation crop img
        crop_img = img[int(ymin):int(ymax), int(xmin):int(xmax)]

        # crop_img_name Naming method ：img_name+"_"+label+"_"+i
        crop_img_name = os.path.splitext(img_name)[0] + "_" + label+"_"+str(i)+".jpg"

        crop_img_save_dir = os.path.join(os.getcwd(),"crop_img",label)
        print(crop_img_save_dir)
        if os.path.exists(crop_img_save_dir) == 0:
            os.makedirs(crop_img_save_dir)
        crop_img_save_path = os.path.join(crop_img_save_dir, crop_img_name)
        cv2.imwrite(crop_img_save_path,crop_img)