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

Fix bbox bug.

Browse Source
This commit is contained in:
WillieMaddox 2016-12-20 08:30:11 -06:00
parent d9eaef0e40
commit b55f6e13a2
1 changed files with 6 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
libs/pascal_voc_io.py
View File

@ -117,11 +117,11 @@ class PascalVocReader:
def getShapes(self): def getShapes(self):
return self.shapes return self.shapes
def addShape(self, label, rect): def addShape(self, label, bndbox):
xmin = rect[0] xmin = int(bndbox.find('xmin').text)
ymin = rect[1] ymin = int(bndbox.find('ymin').text)
xmax = rect[2] xmax = int(bndbox.find('xmax').text)
ymax = rect[3] ymax = int(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)) self.shapes.append((label, points, None, None))
@ -132,13 +132,9 @@ class PascalVocReader:
filename = xmltree.find('filename').text filename = xmltree.find('filename').text
for object_iter in xmltree.findall('object'): for object_iter in xmltree.findall('object'):
rects = []
bndbox = object_iter.find("bndbox") bndbox = object_iter.find("bndbox")
rects.append([int(it.text) for it in bndbox])
label = object_iter.find('name').text label = object_iter.find('name').text
self.addShape(label, bndbox)
for rect in rects:
self.addShape(label, rect)
return True return True