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

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

你好!

我尝试使用自己的生成数据集微调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的语言字符集:

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

相关问题