import xml.etree.ElementTree as ET
import os
from PIL import ImageDraw
from PIL import Image
xml_path = "./xmls/val"
imgs = "./images/val"
img_list = os.listdir(imgs)
print(img_list[0].split('.')[0])
for item in img_list:
name = item.split('.')[0]
xml_name = name+".xml"
xml = open(os.path.join(xml_path, xml_name), 'r', encoding='UTF-8')
tree = ET.parse(xml)
root = tree.getroot()
box = root.find('object').find("bndbox")
x_min = int(box.find('xmin').text)
x_max = int(box.find('xmax').text)
y_min = int(box.find('ymin').text)
y_max = int(box.find('ymax').text)
image = Image.open(os.path.join(imgs, item))
t = [x_min, y_min, x_max, y_max]
draw = ImageDraw.Draw(image)
draw.rectangle(t, outline=(255, 0, 0),width=10)
image.save("./images/label_val" + '/' + item)
VOC格式标签绘制box
最新推荐文章于 2024-11-08 13:43:44 发布