mirror of
https://github.com/PaddlePaddle/PaddleOCR.git
synced 2026-09-24 23:33:08 +08:00
@@ -92,11 +92,11 @@ class TextSystem(object):
|
||||
self.draw_crop_rec_res(self.args.crop_res_save_dir, img_crop_list,
|
||||
rec_res)
|
||||
filter_boxes, filter_rec_res = [], []
|
||||
for box, rec_reuslt in zip(dt_boxes, rec_res):
|
||||
text, score = rec_reuslt
|
||||
for box, rec_result in zip(dt_boxes, rec_res):
|
||||
text, score = rec_result
|
||||
if score >= self.drop_score:
|
||||
filter_boxes.append(box)
|
||||
filter_rec_res.append(rec_reuslt)
|
||||
filter_rec_res.append(rec_result)
|
||||
return filter_boxes, filter_rec_res
|
||||
|
||||
|
||||
|
||||
+2
-2
@@ -73,8 +73,8 @@ def main():
|
||||
images = paddle.to_tensor(images)
|
||||
preds = model(images)
|
||||
post_result = post_process_class(preds)
|
||||
for rec_reuslt in post_result:
|
||||
logger.info('\t result: {}'.format(rec_reuslt))
|
||||
for rec_result in post_result:
|
||||
logger.info('\t result: {}'.format(rec_result))
|
||||
logger.info("success!")
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user