Fix visual bug

8c6548b0 · wuzewu · 31bcce98 · 8c6548b0
显示空白变更内容
内联并排

Showing with 1 addition and 0 deletion

pdseg/vis.py pdseg/vis.py +1 -0

未找到文件。
--- 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)