From 651c4e9abd5ab2dd0dae1e7e0c4105b57e5acf04 Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Fri, 14 May 2021 14:45:11 +0800
Subject: [PATCH] fix(imperative/dataset): fix VOC dataset

GitOrigin-RevId: a46a4c11733cafd7b2ccf9cec1a5f2aaf9a7ef99
---
 .../megengine/data/dataset/vision/voc.py      | 24 +++++++++++++++++++
 1 file changed, 24 insertions(+)

diff --git a/imperative/python/megengine/data/dataset/vision/voc.py b/imperative/python/megengine/data/dataset/vision/voc.py
index 628b51e06..7d1eb744f 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],
+    ]
-- 
GitLab