From d3bd5103772fe84ce15e3602bce72f1c595702fb Mon Sep 17 00:00:00 2001 From: zhoujun <572459439@qq.com> Date: Mon, 21 Dec 2020 22:27:11 +0800 Subject: [PATCH] fix bugs in DetResizeForTest --- ppocr/data/imaug/operators.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/ppocr/data/imaug/operators.py b/ppocr/data/imaug/operators.py index 57cd3b4b..8b9175cf 100644 --- a/ppocr/data/imaug/operators.py +++ b/ppocr/data/imaug/operators.py @@ -119,10 +119,10 @@ class DetResizeForTest(object): if 'image_shape' in kwargs: self.image_shape = kwargs['image_shape'] self.resize_type = 1 - if 'limit_side_len' in kwargs: + elif 'limit_side_len' in kwargs: self.limit_side_len = kwargs['limit_side_len'] self.limit_type = kwargs.get('limit_type', 'min') - if 'resize_long' in kwargs: + elif 'resize_long' in kwargs: self.resize_type = 2 self.resize_long = kwargs.get('resize_long', 960) else: -- GitLab