hook-easyocr.py 793 B

123456789101112131415161718
  1. from PyInstaller.utils.hooks import collect_data_files, get_hook_config
  2. # Recognition backends are imported with `importlib.import_module()`.
  3. hiddenimports = ['easyocr.model.vgg_model', 'easyocr.model.model']
  4. def hook(hook_api):
  5. lang_codes = get_hook_config(hook_api, 'easyocr', 'lang_codes')
  6. if not lang_codes:
  7. lang_codes = ['*']
  8. extra_datas = list()
  9. extra_datas += collect_data_files('easyocr', include_py_files=False, subdir='character',
  10. includes=[f'{lang_code}_char.txt' for lang_code in lang_codes])
  11. extra_datas += collect_data_files('easyocr', include_py_files=False, subdir='dict',
  12. includes=[f'{lang_code}.txt' for lang_code in lang_codes])
  13. hook_api.add_datas(extra_datas)