diff --git a/deploy/python/infer.py b/deploy/python/infer.py
index 8a701d359227e532414b726ff8e84a37af3adf59..fc4ff9bc80d470ef838dc4187c30c9eb0ca15b2b 100644
--- a/deploy/python/infer.py
+++ b/deploy/python/infer.py
@@ -45,6 +45,7 @@ def generate_colormap(num_classes):
             color_map[i * 3 + 2] |= (((lab >> 2) & 1) << (7 - j))
             j += 1
             lab >>= 3
+    color_map = [color_map[i:i+3] for i in range(0, len(color_map), 3)]
     return color_map
 
 # Paddle-TRT Precision Map