fix image shape in infer

585dbc30 · 文幕地方 · 652beeab · 585dbc30
隐藏空白更改
内联并排

Showing with 1 addition and 2 deletion

ppstructure/vqa/utils.py ppstructure/vqa/utils.py +1 -2

未找到文件。
--- a/ppstructure/vqa/utils.py
+++ b/ppstructure/vqa/utils.py
@@ -211,8 +211,7 @@ def preprocess(
    height = ori_img.shape[0]
    width = ori_img.shape[1]

-    img = cv2.resize(ori_img,
-                     (224, 224)).transpose([2, 0, 1]).astype(np.float32)
+    img = cv2.resize(ori_img, img_size).transpose([2, 0, 1]).astype(np.float32)

    segment_offset_id = []
    words_list = []