apoorva/greenhouse
2
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

added float parsing before int conversion to enable support of VOC files that specify coordinates in float

Browse Source
This commit is contained in:
Joakim Eriksson 2019-01-03 11:49:39 +01:00 committed by darrenl
parent 4bb6be6986
commit c2aca6e1d3
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
libs/pascal_voc_io.py
View File

@ -91,9 +91,9 @@ class PascalVocWriter:
pose = SubElement(object_item, 'pose') pose = SubElement(object_item, 'pose')
pose.text = "Unspecified" pose.text = "Unspecified"
truncated = SubElement(object_item, 'truncated') truncated = SubElement(object_item, 'truncated')
if int(each_object['ymax']) == int(self.imgSize[0]) or (int(each_object['ymin'])== 1): if int(float(each_object['ymax'])) == int(float(self.imgSize[0])) or (int(float(each_object['ymin']))== 1):
truncated.text = "1" # max == height or min truncated.text = "1" # max == height or min
elif (int(each_object['xmax'])==int(self.imgSize[1])) or (int(each_object['xmin'])== 1): elif (int(float(each_object['xmax']))==int(float(self.imgSize[1]))) or (int(float(each_object['xmin']))== 1):
truncated.text = "1" # max == width or min truncated.text = "1" # max == width or min
else: else:
truncated.text = "0" truncated.text = "0"
@ -141,10 +141,10 @@ class PascalVocReader:
return self.shapes return self.shapes
def addShape(self, label, bndbox, difficult): def addShape(self, label, bndbox, difficult):
xmin = int(bndbox.find('xmin').text) xmin = int(float(bndbox.find('xmin').text))
ymin = int(bndbox.find('ymin').text) ymin = int(float(bndbox.find('ymin').text))
xmax = int(bndbox.find('xmax').text) xmax = int(float(bndbox.find('xmax').text))
ymax = int(bndbox.find('ymax').text) ymax = int(float(bndbox.find('ymax').text))
points = [(xmin, ymin), (xmax, ymin), (xmax, ymax), (xmin, ymax)] points = [(xmin, ymin), (xmax, ymin), (xmax, ymax), (xmin, ymax)]
self.shapes.append((label, points, None, None, difficult)) self.shapes.append((label, points, None, None, difficult))