diff --git a/ppstructure/vqa/utils.py b/ppstructure/vqa/utils.py index b49ea315d26a0fbd5866a9be1a4ef7d0ea7db71b..0af180ada2eae740c042378c73b884239ddbf7b9 100644 --- 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 = []