Browse Source

fix(imperative/dataset): fix VOC dataset

GitOrigin-RevId: a46a4c1173
release-1.4
Megvii Engine Team 4 years ago
parent
commit
aa0e11febe
1 changed files with 24 additions and 0 deletions
  1. +24
    -0
      imperative/python/megengine/data/dataset/vision/voc.py

+ 24
- 0
imperative/python/megengine/data/dataset/vision/voc.py View File

@@ -184,3 +184,27 @@ class PascalVOC(VisionDataset):
"train",
"tvmonitor",
)

class_colors = [
[0, 0, 0], # background
[0, 0, 128],
[0, 128, 0],
[0, 128, 128],
[128, 0, 0],
[128, 0, 128],
[128, 128, 0],
[128, 128, 128],
[0, 0, 64],
[0, 0, 192],
[0, 128, 64],
[0, 128, 192],
[128, 0, 64],
[128, 0, 192],
[128, 128, 64],
[128, 128, 192],
[0, 64, 0],
[0, 64, 128],
[0, 192, 0],
[0, 192, 128],
[128, 64, 0],
]

Loading…
Cancel
Save