mirrors
/
PaddleOCR
mirror of https://github.com/PaddlePaddle/PaddleOCR.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix: release memory after predict (#10688) 

* fix: release memory after predict

* fix: release memory
pull/10809/head
Chung Chăm Chỉ 2023-08-30 09:22:15 +07:00 committed by GitHub
parent c13a50c7a8
commit 15963b0d24
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tools/infer/predict_rec.py
View File

@ -624,6 +624,7 @@ class TextRecognizer(object):
preds = outputs preds = outputs
else: else:
preds = outputs[0] preds = outputs[0]
self.predictor.try_shrink_memory()
rec_result = self.postprocess_op(preds) rec_result = self.postprocess_op(preds)
for rno in range(len(rec_result)): for rno in range(len(rec_result)):
rec_res[indices[beg_img_no + rno]] = rec_result[rno] rec_res[indices[beg_img_no + rno]] = rec_result[rno]