@@ -184,3 +184,27 @@ class PascalVOC(VisionDataset): | |||||
"train", | "train", | ||||
"tvmonitor", | "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], | |||||
] |