提交 bd3b285e 编写于 作者: S sjtubinlong

fix infer.py visualization bug

上级 a26d964b
...@@ -45,6 +45,7 @@ def generate_colormap(num_classes): ...@@ -45,6 +45,7 @@ def generate_colormap(num_classes):
color_map[i * 3 + 2] |= (((lab >> 2) & 1) << (7 - j)) color_map[i * 3 + 2] |= (((lab >> 2) & 1) << (7 - j))
j += 1 j += 1
lab >>= 3 lab >>= 3
color_map = [color_map[i:i+3] for i in range(0, len(color_map), 3)]
return color_map return color_map
# Paddle-TRT Precision Map # Paddle-TRT Precision Map
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册