Browse Source

fix(mge/data): fix wrong class name in voc

GitOrigin-RevId: 229038bcfa
tags/v0.4.0
Megvii Engine Team Xinran Xu 5 years ago
parent
commit
a795aa700a
1 changed files with 1 additions and 3 deletions
  1. +1
    -3
      python_module/megengine/data/dataset/vision/voc.py

+ 1
- 3
python_module/megengine/data/dataset/vision/voc.py View File

@@ -93,9 +93,7 @@ class PascalVOC(VisionDataset):
elif k == "boxes_category": elif k == "boxes_category":
anno = self.parse_voc_xml(ET.parse(self.annotations[index]).getroot()) anno = self.parse_voc_xml(ET.parse(self.annotations[index]).getroot())
boxes_category = [obj["name"] for obj in anno["annotation"]["object"]] 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) boxes_category = np.array(boxes_category, dtype=np.int32)
target.append(boxes_category) target.append(boxes_category)
elif k == "mask": elif k == "mask":


Loading…
Cancel
Save