fixed ValueError when processing bbox in XML annotation file

c924285b · Xionghc · a66b9e13 · c924285b
Commit c924285b authored Nov 17, 2016 by Xionghc
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 1 deletion

inception/inception/data/process_bounding_boxes.py inception/inception/data/process_bounding_boxes.py +3 -1

No files found.
--- a/inception/inception/data/process_bounding_boxes.py
+++ b/inception/inception/data/process_bounding_boxes.py
@@ -102,7 +102,9 @@ def GetItem(name, root, index=0):


 def GetInt(name, root, index=0):
-  return int(GetItem(name, root, index))
+  #  In some XML annotation files, the point value are not int,  but float.
+  #  So adding a float function to avoid  ValueError.
+  return int(float(GetItem(name, root, index)))


 def FindNumberBoundingBoxes(root):