From 8c6548b04997aeab81906ce62ac9b4d90ff43840 Mon Sep 17 00:00:00 2001
From: wuzewu <wuzewu@baidu.com>
Date: Wed, 4 Sep 2019 17:27:51 +0800
Subject: [PATCH] Fix visual bug

---
 pdseg/vis.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/pdseg/vis.py b/pdseg/vis.py
index 727edae1..e798f2ed 100644
--- a/pdseg/vis.py
+++ b/pdseg/vis.py
@@ -198,6 +198,7 @@ def visualize(cfg,
                 interpolation=cv2.INTER_NEAREST)
 
             if grt is not None:
+                grt = grt[0:valid_shape[0], 0:valid_shape[1]]
                 grt = cv2.resize(
                     grt, (org_shape[1], org_shape[0]),
                     interpolation=cv2.INTER_NEAREST)
-- 
GitLab