EasyOCR 微调加载state_dict时出现的错误(DataParallel)

evrscar2  于 5个月前  发布在  其他
关注(0)|答案(1)|浏览(124)

你好!

我尝试使用自己的生成数据集微调Cyrillic_g2模型。我一直在使用你的教程。
https://github.com/JaidedAI/EasyOCR/blob/master/custom_model.md
https://github.com/JaidedAI/EasyOCR/tree/master/trainer
然而,我遇到了这个错误⬇️ ⬇️ ⬇️

我该如何解决这个问题?
my config file

ufj5ltwl

ufj5ltwl1#

看起来你的语言字符集对于Cyrillic的原始模型训练由JADED AI不匹配,这就是你得到这个错误的原因。你可以将你的语言字符集更改为与JADED AI的Cyrillic字符集匹配,或者将"new_prediction"设置为True。

从他们的配置文件中,这是g1和g2 Cyrillic的语言字符集:

'cyrillic_g1':{
 'filename': 'cyrillic.pth',
 'model_script': 'cyrillic',
 'url': ' [https://github.com/JaidedAI/EasyOCR/releases/download/pre-v1.1.6/cyrillic.zip](https://github.com/JaidedAI/EasyOCR/releases/download/pre-v1.1.6/cyrillic.zip) ',
 'md5sum': '5a046f7be2a4f7da6ed50740f487efa8',
 'symbols': "0123456789!"#$%&'()*+,-./:;<=>?@[\]^_ `{|}~ ", 'characters': '0123456789!"#$%&\'()*+,-./:;<=>?@[\\]^_` {|}~ abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZЁЂЄІЇЈЉЊЋЎЏАБВГДЕЖЗИЙКЛМНОПРСТУФХЦЧШЩЪЫЬЭЮЯабвгдежзийклмнопрстуфхцчшщъыьэюяёђєіїјљњћўџҐґҮүө'
 }
'cyrillic_g2':{
 'filename': 'cyrillic_g2.pth',
 'model_script': 'cyrillic',
 'url': ' [https://github.com/JaidedAI/EasyOCR/releases/download/v1.6.1/cyrillic_g2.zip](https://github.com/JaidedAI/EasyOCR/releases/download/v1.6.1/cyrillic_g2.zip) ',
 'md5sum': '19f85f43d9128a89ac21b8d6a06973fe',
 'symbols': '0123456789!"#$%&'()*+,-./:;<=>?@[\]^_ `{|}~ €₽', 'characters': '0123456789!"#$%&\'()*+,-./:;<=>?@[\\]^_` {|}~ €₽ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyzАБВГДЕЁЖЗИЙКЛМНОПРСТУФХЦЧШЩЪЫЬЭЮЯабвгдеёжзийклмнопрстуфхцчшщъыьэюяЂђЃѓЄєІіЇїЈјЉљЊњЋћЌќЎўЏџҐґҒғҚқҮүҲҳҶҷӀӏӢӣӨөӮӯ'
 },

相关问题