From f509d97963ba362f740aaccd6e0d8718e46ea251 Mon Sep 17 00:00:00 2001 From: ET0E <119522284+ET0E@users.noreply.github.com> Date: Mon, 3 Jun 2024 13:16:01 -0400 Subject: [PATCH] Update create_csv.py This will solve the issue I had ""IndexError: child index out of range " --- util_scripts/create_csv.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) 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']