diff --git a/imperative/python/megengine/data/dataset/vision/voc.py b/imperative/python/megengine/data/dataset/vision/voc.py index 628b51e065abc6febb4910e6b30c4b5c6ffece0b..7d1eb744f9e51e4b5b4270af2e4c1fa51bb184b1 100644 --- a/imperative/python/megengine/data/dataset/vision/voc.py +++ b/imperative/python/megengine/data/dataset/vision/voc.py @@ -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], + ]