diff --git a/Tensorflow/scripts/generate_tfrecord.py b/Tensorflow/scripts/generate_tfrecord.py
index caad456d..9d7895ac 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(float(member[4][0].text)),
+                     int(float(member[4][1].text)),
+                     int(float(member[4][2].text)),
+                     int(float(member[4][3].text))
                      )
             xml_list.append(value)
     column_name = ['filename', 'width', 'height',