From bd3b285ed4a25445460a59b47e3856894b542de0 Mon Sep 17 00:00:00 2001
From: sjtubinlong <binlong.plus@gmail.com>
Date: Tue, 10 Dec 2019 18:02:23 +0800
Subject: [PATCH] fix infer.py visualization bug

---
 deploy/python/infer.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/deploy/python/infer.py b/deploy/python/infer.py
index 8a701d35..fc4ff9bc 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
-- 
GitLab