forgetest5
/
PaddleOCR
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Merge pull request #3215 from YuxinZou/release/2.1 

style: fix spelling mistake
This commit is contained in:
MissPenguin 2021-07-01 12:04:51 +08:00 committed by GitHub
parent 39161f5c9b 638ebee5ca
commit 0ae76be591
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 10 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tools/infer/predict_system.py
View File

@ -111,11 +111,11 @@ class TextSystem(object):
len(rec_res), elapse)) len(rec_res), elapse))
# self.print_draw_crop_rec_res(img_crop_list, rec_res) # self.print_draw_crop_rec_res(img_crop_list, rec_res)
filter_boxes, filter_rec_res = [], [] filter_boxes, filter_rec_res = [], []
for box, rec_reuslt in zip(dt_boxes, rec_res): for box, rec_result in zip(dt_boxes, rec_res):
text, score = rec_reuslt text, score = rec_result
if score >= self.drop_score: if score >= self.drop_score:
filter_boxes.append(box) filter_boxes.append(box)
filter_rec_res.append(rec_reuslt) filter_rec_res.append(rec_result)
return filter_boxes, filter_rec_res return filter_boxes, filter_rec_res

4
tools/infer_cls.py
View File

@ -73,8 +73,8 @@ def main():
images = paddle.to_tensor(images) images = paddle.to_tensor(images)
preds = model(images) preds = model(images)
post_result = post_process_class(preds) post_result = post_process_class(preds)
for rec_reuslt in post_result: for rec_result in post_result:
logger.info('\t result: {}'.format(rec_reuslt)) logger.info('\t result: {}'.format(rec_result))
logger.info("success!") logger.info("success!")

10
tools/infer_rec.py
View File

@ -107,11 +107,11 @@ def main():
else: else:
preds = model(images) preds = model(images)
post_result = post_process_class(preds) post_result = post_process_class(preds)
for rec_reuslt in post_result: for rec_result in post_result:
logger.info('\t result: {}'.format(rec_reuslt)) logger.info('\t result: {}'.format(rec_result))
if len(rec_reuslt) >= 2: if len(rec_result) >= 2:
fout.write(file + "\t" + rec_reuslt[0] + "\t" + str( fout.write(file + "\t" + rec_result[0] + "\t" + str(
rec_reuslt[1]) + "\n") rec_result[1]) + "\n")
logger.info("success!") logger.info("success!")