diff --git a/util_scripts/create_csv.py b/util_scripts/create_csv.py index d919fcb1..13dac4c4 100644 --- a/util_scripts/create_csv.py +++ b/util_scripts/create_csv.py @@ -13,13 +13,13 @@ def xml_to_csv(path): root = tree.getroot() for member in root.findall('object'): value = (root.find('filename').text, - int(root.find('size')[0].text), - int(root.find('size')[1].text), + int(root.find('size').find('width').text), + int(root.find('size').find('width').text), member[0].text, - int(member[4][0].text), - int(member[4][1].text), - int(member[4][2].text), - int(member[4][3].text) + int(member.find("bndbox").find('xmin').text), + int(member.find("bndbox").find('xmin').text), + int(member.find("bndbox").find('xmin').text), + int(member.find("bndbox").find('xmin').text) ) xml_list.append(value) column_name = ['filename', 'width', 'height', 'class', 'xmin', 'ymin', 'xmax', 'ymax']