我正在尝试针对新图像的宽度和高度转换xml文件中的边界框坐标。示例XML文件如下:
<annotations>
<image height="940" id="3" name="C_00080.jpg" width="1820">
<box label="Objects" occluded="0" xbr="801.99255" xtl="777.78656" ybr="506.9955" ytl="481.82132">
<attribute name="Class">B</attribute>
</box>
<box label="Objects" occluded="0" xbr="999.319" xtl="963.38654" ybr="519.2735" ytl="486.68628">
<attribute name="Class">A</attribute>
</box>
</image>
<annotations>
xml中原始图像的宽度和高度为1820x940
,并且框坐标相同。我想将框坐标更改为新图像的宽度和高度,即1080x720
。我已经编写了这段代码,有人可以帮助我验证或告诉我以下代码的更好方法。
import xml.etree.ElementTree as ET
label_file = '1.xml'
tree = ET.parse(label_file)
root = tree.getroot()
for image in root.findall('image'):
image.attrib['width'] = '1080' # Original width = 1820
image.attrib['height'] = '720' # Original width = 940
for allBboxes in image.findall('box'):
xmin = float(allBboxes.attrib['xtl'])
xminNew = float(xmin / (1820/1080))
xminNew = float("{:.5f}".format(xminNew))
allBboxes.attrib['xtl'] = str(xminNew)
ymin = float(allBboxes.attrib['ytl'])
yminNew = float(ymin / (940/720))
yminNew = float("{:.5f}".format(yminNew))
allBboxes.attrib['ytl'] = str(yminNew)
xmax = float(allBboxes.attrib['xbr'])
xmaxNew = float(xmax / (1820/1080))
xmaxNew = float("{:.5f}".format(xmaxNew))
allBboxes.attrib['xbr'] = str(xmaxNew)
ymax = float(allBboxes.attrib['ybr'])
ymaxNew = float(ymax / (940/720))
ymaxNew = float("{:.5f}".format(ymaxNew))
allBboxes.attrib['ybr'] = str(ymaxNew)
tree.write(label_file)
删除无用的浮点转换
(另存为.xsl文件)