소스 검색

Merge pull request #656 from xionghcx/xionghcx-patch-1

Fix ValueError when processing bbox in XML annotation file
Neal Wu 8 년 전
부모
커밋
015490f3df
1개의 변경된 파일3개의 추가작업 그리고 1개의 파일을 삭제
  1. 3 1
      inception/inception/data/process_bounding_boxes.py

+ 3 - 1
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):