PaddleOCR2Pytorch icon indicating copy to clipboard operation
PaddleOCR2Pytorch copied to clipboard

请问可以用这个预训练模型在pytorch上直接训练吗?(不用paddleocr)

Open guangzufang opened this issue 4 years ago • 6 comments

guangzufang avatar Nov 09 '21 08:11 guangzufang

这里没有pytorch的训练代码

frotms avatar Nov 09 '21 13:11 frotms

后面会补吗:)

DCC-lzhy avatar Dec 15 '21 01:12 DCC-lzhy

不会补。只做inference in PyTorch

frotms avatar Dec 15 '21 02:12 frotms

那训练是直接拿的paddleocr训练的吗? 这个项目很棒,作者厉害的。 有考虑后面加pytorch2onnx2tensorrt吗?

DCC-lzhy avatar Dec 16 '21 03:12 DCC-lzhy

还有个小问题,我一开始拿着pt_models/det_models/*.pth去测我的数据,表现很不好;然后我下了PPOCRLabel,试用了自动标注,发现很准,就看了源码里使用的是什么模型,原来都是ch开头的,因为我的数据里都是中文数字的,这个ch是指中文吧。 但我没找到你的readme里对det_models和rec_models里的模型介绍,这里面的是啥呀?

DCC-lzhy avatar Dec 16 '21 11:12 DCC-lzhy

在我仔细看了config里和inference.md,发现原来是针对其他数据集训练得到的,不好意思,是我没发现。

DCC-lzhy avatar Dec 16 '21 11:12 DCC-lzhy