基于Pytorch的OCR工具库,支持常用的文字检测和识别算法
Geek Repo:Geek Repo
Github PK Tool:Github PK Tool
ltm920716 opened this issue 3 years ago · comments
hello, 识别部分是ocr的难点,请问怎样才能在提供的训练好的模型 ch_rec_server_crnn_res34.pth 上直接做finetune呢,从头训练缺少很多数据集,并且也不是很好训练,谢谢
只保留backbone,bn开启eval。
@novioleo 感谢回复,目前问题是模型 state_dict 可以对不上
请问这个是需要人为转换一下key就可以么,谢谢
model.module.load_state_dict()不就解决了吗。。
@novioleo 是的,是因为dataparrallel的问题,git上的代码需要需要优化一下
加载后续还是有一些其他问题(optimize,dataloader等),我尝试解决,非常感谢回复