VOC格式标签绘制box

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)

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值