From a795aa700a6ee2b6a002ff1bb612ec14d1fab3e5 Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Fri, 8 May 2020 18:06:58 +0800 Subject: [PATCH] fix(mge/data): fix wrong class name in voc GitOrigin-RevId: 229038bcfa84d109f299cba42bdfe7c51e37b4a0 --- python_module/megengine/data/dataset/vision/voc.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/python_module/megengine/data/dataset/vision/voc.py b/python_module/megengine/data/dataset/vision/voc.py index c1eebe56..95f373be 100644 --- a/python_module/megengine/data/dataset/vision/voc.py +++ b/python_module/megengine/data/dataset/vision/voc.py @@ -93,9 +93,7 @@ class PascalVOC(VisionDataset): elif k == "boxes_category": anno = self.parse_voc_xml(ET.parse(self.annotations[index]).getroot()) boxes_category = [obj["name"] for obj in anno["annotation"]["object"]] - boxes_category = [ - self.class_names.index(bc) - 1 for bc in boxes_category - ] + boxes_category = [self.class_names.index(bc) for bc in boxes_category] boxes_category = np.array(boxes_category, dtype=np.int32) target.append(boxes_category) elif k == "mask":