From 851660d3244d7ff22c37fc5ef38acdfcbbef1cb4 Mon Sep 17 00:00:00 2001 From: zimenglan Date: Mon, 26 Nov 2018 18:50:04 +0800 Subject: [PATCH] make pixel indexes 0-based for bounding box in pascal voc dataset (#209) --- maskrcnn_benchmark/data/datasets/voc.py | 20 +++++++++++--------- 1 file changed, 11 insertions(+), 9 deletions(-) diff --git a/maskrcnn_benchmark/data/datasets/voc.py b/maskrcnn_benchmark/data/datasets/voc.py index 6b57764..459985b 100644 --- a/maskrcnn_benchmark/data/datasets/voc.py +++ b/maskrcnn_benchmark/data/datasets/voc.py @@ -88,22 +88,24 @@ class PascalVOCDataset(torch.utils.data.Dataset): boxes = [] gt_classes = [] difficult_boxes = [] + TO_REMOVE = 1 + for obj in target.iter("object"): difficult = int(obj.find("difficult").text) == 1 if not self.keep_difficult and difficult: continue name = obj.find("name").text.lower().strip() bb = obj.find("bndbox") + # Make pixel indexes 0-based + # Refer to "https://github.com/rbgirshick/py-faster-rcnn/blob/master/lib/datasets/pascal_voc.py#L208-L211" + box = [ + bb.find("xmin").text, + bb.find("ymin").text, + bb.find("xmax").text, + bb.find("ymax").text, + ] bndbox = tuple( - map( - int, - [ - bb.find("xmin").text, - bb.find("ymin").text, - bb.find("xmax").text, - bb.find("ymax").text, - ], - ) + map(lambda x: x - TO_REMOVE, list(map(int, box))) ) boxes.append(bndbox) -- GitLab