请问pth模型如何使用paddle进行训练？

frotms / PaddleOCR2Pytorch

PaddleOCR inference in PyTorch. Converted from [PaddleOCR](https://github.com/PaddlePaddle/PaddleOCR)

Apache License 2.0

877 stars 174 forks source link

Open zhangqiqi1228 opened 1 year ago

zhangqiqi1228 commented 1 year ago

作者，您好！看到您之前的回复有说这里的模型，可以直接使用paddle框架进行训练。请问具体是用哪个脚本进行训练呢？ trian.py脚本好像只支持paddle的模型格式。

command： python tools/train.py -c configs/det/det_r50_vd_db.yml -o Global.pretrained_model=./pt_models/det_r50_vd_db_v2.0_infer.pth

报错： AssertionError: The ./pt_models/det_r50_vd_db_v2.0_infer.pth.pdparams does not exists!

frotms commented 1 year ago

@zhangqiqi1228 需要用PaddleOCR的工程训练，将训好的模型再转成PyTorch

zhangqiqi1228 commented 1 year ago

@zhangqiqi1228 需要用PaddleOCR的工程训练，将训好的模型再转成PyTorch

明白了，谢谢！