소스 검색

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 달 전
부모
커밋
b0b4b7ba33
No account linked to committer's email address
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)

Loading…
취소
저장