标注xml bounding box计算问题

[terrianzhang]

以下是自动跟踪标注完的标注xml内容，xmax，和ymax似乎算错了，那个是bounding box的宽度和高度

xxxxxx
0.jpg
xxxxxx\0.jpg

478
850
3


screw
Unspecified
0
0

217
435
61
53

util/xmlfile.py这个程序应该有问题

[M1norkey]

确实有问题，标注出来的xml用labelimg打开位置就不对了
我经过检查确实是util/xmlfile.py这个脚本应该有问题

给的bndbox实际上是 [x, y, w, h]，即左上角坐标(x,y)和宽、高。

但是VOC XML格式要求的是xmin, ymin, xmax, ymax，即左上角和右下角的坐标。

所以，代码里应该把w和h转成右下角坐标才对，换句话说，应该是：

xmin = x
ymin = y
xmax = x + w
ymax = y + h

在写入标准VOC XML时的这段代码改成
def xml_message(save_path,image_name,img_width,img_height,text,x,y,w,h):
file_path = os.path.join(save_path, f"{image_name}.xml")
size = [img_width, img_height, 3]
result = {
'name': text,
'pose': 'Unspecified',
'truncated': 0,
'difficult': 0,
'bndbox': [x, y, x + w, y + h]
}
return result,file_path,size

重点是'bndbox': [x, y, w, h]改成 'bndbox': [x, y, x + w, y + h]
这样改动之后，经过labelimg检测，位置也是对的

同时，读取xml绘制框的逻辑也得相应调整
util/QtFunc.py这个文件改一部分

def list_label(label_path):
with open(label_path, 'r') as file:
content = file.read()
root = ET.fromstring(content)

objects = root.findall('object')

list_labels = []
list_box = []
for obj in objects:
    name = obj.find('name').text
    xmin = int(obj.find('bndbox/xmin').text)
    ymin = int(obj.find('bndbox/ymin').text)
    xmax = int(obj.find('bndbox/xmax').text)
    ymax = int(obj.find('bndbox/ymax').text)
    box = [xmin, ymin, xmax, ymax]  # 标准VOC格式
    list_labels.append(name)
    list_box.append(box)
return list_labels, list_box