From b55f6e13a263a5eab7aad8029e501f2f00ebf1cf Mon Sep 17 00:00:00 2001 From: WillieMaddox Date: Tue, 20 Dec 2016 08:30:11 -0600 Subject: [PATCH] Fix bbox bug. --- libs/pascal_voc_io.py | 16 ++++++---------- 1 file changed, 6 insertions(+), 10 deletions(-) diff --git a/libs/pascal_voc_io.py b/libs/pascal_voc_io.py index 7791e8de..11d654d4 100644 --- a/libs/pascal_voc_io.py +++ b/libs/pascal_voc_io.py @@ -117,11 +117,11 @@ class PascalVocReader: def getShapes(self): return self.shapes - def addShape(self, label, rect): - xmin = rect[0] - ymin = rect[1] - xmax = rect[2] - ymax = rect[3] + def addShape(self, label, bndbox): + xmin = int(bndbox.find('xmin').text) + ymin = int(bndbox.find('ymin').text) + xmax = int(bndbox.find('xmax').text) + ymax = int(bndbox.find('ymax').text) points = [(xmin, ymin), (xmax, ymin), (xmax, ymax), (xmin, ymax)] self.shapes.append((label, points, None, None)) @@ -132,13 +132,9 @@ class PascalVocReader: filename = xmltree.find('filename').text for object_iter in xmltree.findall('object'): - rects = [] bndbox = object_iter.find("bndbox") - rects.append([int(it.text) for it in bndbox]) label = object_iter.find('name').text - - for rect in rects: - self.addShape(label, rect) + self.addShape(label, bndbox) return True