训练脚本:
python ./tools/train.py -c ./configs/rec/SVTRv2/rec_repsvtr_gtc.yml
SVTRV2版本导出:
python tools/export_model.py -c ./configs/rec/SVTRv2/rec_repsvtr_gtc.yml -o Global.pretrained_model=output/rec_repsvtr_gtc/best_accuracy Global.save_inference_dir=./inference/rec_repsvtr_infer
导出onnx:
paddle2onnx --model_dir ./inference/rec_repsvtr_infer/ --model_file inference.pdmodel --params_filename inference.pdiparams --save_file ./inference/rec_repsvtr_infer/model.onnx --opset_version 11 --enable_onnx_checker True
onnxruntime推理:
python3.7 tools/infer/predict_system.py --use_gpu=False --use_onnx=True \
--det_model_dir=./inference/det_onnx/model.onnx \
--rec_model_dir=./inference/rec_onnx/model.onnx \
--cls_model_dir=./inference/cls_onnx/model.onnx \
--image_dir=doc/imgs_en/img_12.jpg \</