From b0b4b7ba337cb513af462881cc4dc3e299524d38 Mon Sep 17 00:00:00 2001 From: Stephen Hu Date: Tue, 18 Mar 2025 09:39:49 +0800 Subject: [PATCH] Feat: Improve Recognizer.py performance (#6185) ### What problem does this PR solve? For the create_inputs method based on np operation to replace for loop ### Type of change - [x] Performance Improvement --- deepdoc/vision/recognizer.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/deepdoc/vision/recognizer.py b/deepdoc/vision/recognizer.py index 1052350b8..6911d8698 100644 --- a/deepdoc/vision/recognizer.py +++ b/deepdoc/vision/recognizer.py @@ -194,10 +194,9 @@ class Recognizer: inputs['scale_factor'] = np.array( (im_info[0]['scale_factor'],)).astype('float32') return inputs - - for e in im_info: - im_shape.append(np.array((e['im_shape'],)).astype('float32')) - scale_factor.append(np.array((e['scale_factor'],)).astype('float32')) + + im_shape = np.array([info['im_shape'] for info in im_info], dtype='float32') + scale_factor = np.array([info['scale_factor'] for info in im_info], dtype='float32') inputs['im_shape'] = np.concatenate(im_shape, axis=0) inputs['scale_factor'] = np.concatenate(scale_factor, axis=0)