fix(imperative/dataset): fix VOC dataset

GitOrigin-RevId: a46a4c11733cafd7b2ccf9cec1a5f2aaf9a7ef99

fix(imperative/dataset): fix VOC dataset
GitOrigin-RevId: a46a4c11733cafd7b2ccf9cec1a5f2aaf9a7ef99
651c4e9a · Megvii Engine Team · 4566cc21 · 651c4e9a
显示空白变更内容
内联并排

Showing with 24 addition and 0 deletion

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

未找到文件。
--- 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],
+    ]