From 8020c8157da257ca625c526ed1f30f7854a00d79 Mon Sep 17 00:00:00 2001 From: Syed Nahin Hossain <50939853+NAHIN-JZS@users.noreply.github.com> Date: Sat, 27 Aug 2022 19:01:21 +0600 Subject: [PATCH 1/2] Update generate_tfrecord.py --- Tensorflow/scripts/generate_tfrecord.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/Tensorflow/scripts/generate_tfrecord.py b/Tensorflow/scripts/generate_tfrecord.py index caad456d..93a35520 100644 --- a/Tensorflow/scripts/generate_tfrecord.py +++ b/Tensorflow/scripts/generate_tfrecord.py @@ -85,10 +85,10 @@ def xml_to_csv(path): int(root.find('size')[0].text), int(root.find('size')[1].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[5][0].text), + int(member[5][1].text), + int(member[5][2].text), + int(member[5][3].text) ) xml_list.append(value) column_name = ['filename', 'width', 'height', From 63428b81aa0d27436e002d5f0deb20eb835c96c6 Mon Sep 17 00:00:00 2001 From: Syed Nahin Hossain <50939853+NAHIN-JZS@users.noreply.github.com> Date: Thu, 1 Sep 2022 14:33:12 +0600 Subject: [PATCH 2/2] Update generate_tfrecord.py In create_tf_example function, there is a bug. xmins->row['min'], xmaxs->row['ymin'] (I don't know why), ymins->row['xmax'], ymaxs -> row['ymax']... --- Tensorflow/scripts/generate_tfrecord.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/Tensorflow/scripts/generate_tfrecord.py b/Tensorflow/scripts/generate_tfrecord.py index 93a35520..203eff26 100644 --- a/Tensorflow/scripts/generate_tfrecord.py +++ b/Tensorflow/scripts/generate_tfrecord.py @@ -125,8 +125,8 @@ def create_tf_example(group, path): for index, row in group.object.iterrows(): xmins.append(row['xmin'] / width) - xmaxs.append(row['xmax'] / width) - ymins.append(row['ymin'] / height) + xmaxs.append(row['ymin'] / width) + ymins.append(row['xmax'] / height) ymaxs.append(row['ymax'] / height) classes_text.append(row['class'].encode('utf8')) classes.append(class_text_to_int(row['class']))