首页 PaddleOCR 帖子详情
求助,使用infer_kie_token_ser.py 推理图片时报错
收藏
快速回复
PaddleOCR 问答OCR检测识别 4021 0
求助,使用infer_kie_token_ser.py 推理图片时报错
收藏
快速回复
PaddleOCR 问答OCR检测识别 4021 0

推理的模型是用网上下载的,增值税发票(zzsfp) 数据集训练的,训练完成后使用训练出来的模型对zzsfp里的任意一张图片进行推理,执行:python tools/infer_kie_token_ser.py -c L:/PaddleOCR-2.7.1/train_data/zzsfp/ser_vi_layoutxlm_udml.yml -o Architecture.Models.Student.Backbone.checkpoints=L:/PaddleOCR-2.7.1/train_data/zzsfp/output/ser_vi_layoutxlm_xfund_zh_udml/best_accuracy/ Global.infer_img=L:/PaddleOCR-2.7.1/train_data/zzsfp/imgs/b137.jpg ;命令后,在控制台窗口python输出了:

Traceback (most recent call last):
File "L:\PaddleOCR-2.7.1\tools\infer_kie_token_ser.py", line 151, in
result = result[0]
KeyError: 0

报错时result 的值为 {'Student': [[{'transcription': '用发票', 'bbox': [2243, 455, 2605, 596], 'points': [[2243.0, 471.0], [2600.0, 455.0], [2605.0, 579.0], [2248.0, 596.0]], 'pred_id': 1, 'pred': 'QUESTION'}, {'transcription': 'No 122638724400154130', 'bbox': [2835, 469, 3933, 617], 'points': [[2835.0, 506.0], [3929.0, 469.0], [3933.0, 579.0], [2839.0, 617.0]], 'pred_id': 0, 'pred': 'O'}, {'transcription': '东增', 'bbox': [1655, 508, 1895, 613], 'points': [[1655.0, 520.0], [1891.0, 508.0], [1895.0, 601.0], [1660.0, 613.0]], 'pred_id': 1, 'pred': 'QUESTION'......

请教各位大佬这个问题该如何解决

0
收藏
回复
需求/bug反馈?一键提issue告诉我们
发现bug?如果您知道修复办法,欢迎提pr直接参与建设飞桨~
在@后输入用户全名并按空格结束,可艾特全站任一用户