Sfoglia il codice sorgente

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 mesi fa
parent
commit
b0b4b7ba33
Nessun account collegato all'indirizzo email del committer
1 ha cambiato i file con 3 aggiunte e 4 eliminazioni
  1. 3
    4
      deepdoc/vision/recognizer.py

+ 3
- 4
deepdoc/vision/recognizer.py Vedi File

@@ -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)

Loading…
Annulla
Salva