导出识别模型时都是 KeyError: 'predict'
Created by: goodtogood
paddlepaddle-gpu 1.7.2.post107 PaddleOCR 3b40c32
python tools/export_model.py -c configs/rec/rec_r34_vd_tps_bilstm_attn.yml -o Global.checkpoints="./output/rec/rec_r34_vd_tps_bilstm_attn/best_accuracy" Global.save_inference_dir="./inference/rec/rec_r34_vd_tps_bilstm_attn"
python tools/export_model.py -c configs/rec/rec_mv3_tps_bilstm_attn.yml -o Global.checkpoints="./output/rec/rec_mv3_tps_bilstm_attn/best_accuracy" Global.save_inference_dir="./inference/rec/rec_mv3_tps_bilstm_attn"
Traceback (most recent call last):
File "tools/export_model.py", line 94, in <module>
main()
File "tools/export_model.py", line 68, in main
config, eval_program, startup_prog)
File "PaddleOCR\tools\program.py", line 193, in build_export
image, outputs = model(mode='export')
File "PaddleOCR\ppocr\modeling\architectures\rec_model.py", line 112
, in __call__
predict = predicts['predict']
KeyError: 'predict'