Просмотр исходного кода

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
tags/v0.18.0
Stephen Hu 7 месяцев назад
Родитель
Сommit
b0b4b7ba33
Аккаунт пользователя с таким Email не найден
1 измененных файлов: 3 добавлений и 4 удалений
  1. 3
    4
      deepdoc/vision/recognizer.py

+ 3
- 4
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)

Загрузка…
Отмена
Сохранить