Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[Bug]: 使用Mac的Metal在OCR阶段会报错,识别出来的内容也不正确 #569

Open
tg111 opened this issue Jan 27, 2024 · 0 comments
Labels
bug Something isn't working

Comments

@tg111
Copy link

tg111 commented Jan 27, 2024

问题

设备:iMac,AMD显卡。增加--use-gpu参数会有报错,识别出来的文本也不正确,使用cpu不报错,OCR内容也正确。

命令行参数

python -m manga_translator --verbose --translator=baidu --target-lang=CHS --use-gpu -i /Users/gii/Documents/code/pycode/manga-image-translator-main/2.png

报错日志

[batch] Running text detection
[DefaultDetector] Detection resolution: 1280x1536
[batch] Running ocr
[Model48pxOCR] prob: 0.22398340702056885 曲ちゅ fg: (10, 7, 4) bg: (245, 245, 244)
[Model48pxOCR] prob: 0.48187121748924255 ラḮっ! fg: (0, 1, 0) bg: (108, 109, 109)
[Model48pxOCR] prob: 20.847171783447266 入哘ゃ寸ぃぞ fg: (3, 6, 4) bg: (231, 226, 226)
[Model48pxOCR] prob: 11.87712287902832 で𝘺りかげで fg: (2, 4, 3) bg: (174, 174, 172)
ERROR: [batch] IndexError: list index out of range
Traceback (most recent call last):
  File "/Users/gii/Documents/code/pycode/manga-image-translator-main/manga_translator/__main__.py", line 73, in <module>
    loop.run_until_complete(dispatch(args))
  File "/usr/local/opt/python@3.8/Frameworks/Python.framework/Versions/3.8/lib/python3.8/asyncio/base_events.py", line 616, in run_until_complete
    return future.result()
  File "/Users/gii/Documents/code/pycode/manga-image-translator-main/manga_translator/__main__.py", line 42, in dispatch
    await translator.translate_path(path, dest, args_dict)
  File "/Users/gii/Documents/code/pycode/manga-image-translator-main/manga_translator/manga_translator.py", line 152, in translate_path
    await self.translate_file(path, _dest, params)
  File "/Users/gii/Documents/code/pycode/manga-image-translator-main/manga_translator/manga_translator.py", line 203, in translate_file
    return await self._translate_file(path, dest, ctx)
  File "/Users/gii/Documents/code/pycode/manga-image-translator-main/manga_translator/manga_translator.py", line 246, in _translate_file
    ctx = await self.translate(img, ctx)
  File "/Users/gii/Documents/code/pycode/manga-image-translator-main/manga_translator/manga_translator.py", line 303, in translate
    return await self._translate(ctx)
  File "/Users/gii/Documents/code/pycode/manga-image-translator-main/manga_translator/manga_translator.py", line 392, in _translate
    ctx.textlines = await self._run_ocr(ctx)
  File "/Users/gii/Documents/code/pycode/manga-image-translator-main/manga_translator/manga_translator.py", line 470, in _run_ocr
    textlines = await dispatch_ocr(ctx.ocr, ctx.img_rgb, ctx.textlines, ctx, self.device, self.verbose)
  File "/Users/gii/Documents/code/pycode/manga-image-translator-main/manga_translator/ocr/__init__.py", line 36, in dispatch
    return await ocr.recognize(image, regions, args, verbose)
  File "/Users/gii/Documents/code/pycode/manga-image-translator-main/manga_translator/ocr/common.py", line 45, in recognize
    return await self._recognize(image, textlines, args, verbose)
  File "/Users/gii/Documents/code/pycode/manga-image-translator-main/manga_translator/ocr/common.py", line 56, in _recognize
    return await self.infer(*args, **kwargs)
  File "/Users/gii/Documents/code/pycode/manga-image-translator-main/manga_translator/utils/inference.py", line 342, in infer
    return await self._infer(*args, **kwargs)
  File "/Users/gii/Documents/code/pycode/manga-image-translator-main/manga_translator/ocr/model_48px.py", line 116, in _infer
    ch = self.model.dictionary[chid]
IndexError: list index out of range

Command Line Arguments

No response

Console logs

No response

@tg111 tg111 added the bug Something isn't working label Jan 27, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
bug Something isn't working
Projects
None yet
Development

No branches or pull requests

1 participant