From 1cd4580be5d03f4afc36b87a7e04aa5cf9300b67 Mon Sep 17 00:00:00 2001
From: Channingss <chen_lingchi@163.com>
Date: Mon, 11 May 2020 07:44:37 +0000
Subject: [PATCH] modify segment

---
 paddlex/cv/transforms/seg_transforms.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/paddlex/cv/transforms/seg_transforms.py b/paddlex/cv/transforms/seg_transforms.py
index 63c4fab..89733e3 100644
--- a/paddlex/cv/transforms/seg_transforms.py
+++ b/paddlex/cv/transforms/seg_transforms.py
@@ -333,7 +333,10 @@ class ResizeByShort:
         """
         Args:
             im (numnp.ndarraypy): 图像np.ndarray数据。
-            im_info (dict, 可选): 存储与图像相关的信息。
+            im_info (list): 存储图像reisze或padding前的shape信息，如
+                [('resize', [200, 300]), ('padding', [400, 600])]表示
+                图像在过resize前shape为(200, 300)， 过padding前shape为
+                (400, 600)
             label (np.ndarray): 标注图像np.ndarray数据。
 
         Returns:
-- 
GitLab