目标检测：图像根据xml里的boundingbox信息画框

Posted 2022-12-11 告白少年

根据xml boundingbox信息画框

import os
import os.path
import xml.etree.cElementTree as ET
import cv2
def draw(image_path, root_saved_path):
    """
    图片根据标注画框
    """
    src_img_path = image_path
    for file in os.listdir(src_img_path):
        print(file)
        file_name, suffix = os.path.splitext(file)
        if suffix == '.xml':
            # print(file)
            xml_path = os.path.join(src_img_path, file)
            image_path = os.path.join(src_img_path, file_name+'.jpg')
            img = cv2.imread(image_path)
            tree = ET.parse(xml_path)
            root = tree.getroot()
            for obj in root.iter('object'):
                name = obj.find('name').text
                xml_box = obj.find('bndbox')
                x1 = int(xml_box.find('xmin').text)
                x2 = int(xml_box.find('xmax').text)
                y1 = int(xml_box.find('ymin').text)
                y2 = int(xml_box.find('ymax').text)
                cv2.rectangle(img, (x1, y1), (x2, y2), (255, 0, 0), thickness=2)
                # 字为绿色
                cv2.putText(img, name, (x1, y1), cv2.FONT_HERSHEY_COMPLEX, 0.7, (0, 255, 0), thickness=2)
            cv2.imwrite(os.path.join(root_saved_path, file_name+'.jpg'), img)


if __name__ == '__main__':
    image_path = r"E:\\data_set\\test1"
    root_saved_path = r"E:\\data_set\\result2"
    draw(image_path, root_saved_path)