Skip to content

GitLab

Commit 585dbc30 authored by WenmuZhou's avatar WenmuZhou
Browse files

fix image shape in infer

parent 652beeab
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 2 deletions
ppstructure/vqa/utils.py
View file @ 585dbc30
...@@ -211,8 +211,7 @@ def preprocess( ...@@ -211,8 +211,7 @@ def preprocess(
height = ori_img.shape[0] height = ori_img.shape[0]
width = ori_img.shape[1] width = ori_img.shape[1]
img = cv2.resize(ori_img, img = cv2.resize(ori_img, img_size).transpose([2, 0, 1]).astype(np.float32)
(224, 224)).transpose([2, 0, 1]).astype(np.float32)
segment_offset_id = [] segment_offset_id = []
words_list = [] words_list = []
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment