WenmuZhou / PytorchOCR

基于Pytorch的OCR工具库,支持常用的文字检测和识别算法

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

finetune rec_model

ltm920716 opened this issue · comments

commented

hello,
识别部分是ocr的难点,请问怎样才能在提供的训练好的模型 ch_rec_server_crnn_res34.pth 上直接做finetune呢,从头训练缺少很多数据集,并且也不是很好训练,谢谢

只保留backbone,bn开启eval。

commented

@novioleo
感谢回复,目前问题是模型 state_dict 可以对不上

image
image

请问这个是需要人为转换一下key就可以么,谢谢

model.module.load_state_dict()不就解决了吗。。

commented

@novioleo
是的,是因为dataparrallel的问题,git上的代码需要需要优化一下

image

加载后续还是有一些其他问题(optimize,dataloader等),我尝试解决,非常感谢回复