EasyOCR 为什么探测器仅限于craft和dbnet18?

e3bfsja2  于 6个月前  发布在  其他
关注(0)|答案(5)|浏览(62)

你好,
首先,感谢这个很棒的项目。
我正在尝试使用DBnet文本检测器。然而,我在想为什么只有dbnet18可以用作如下代码片段中所示。
EasyOCR/easyocr/easyocr.py
第84行 29e39cb
| | self.support_detection_network= ['craft', 'dbnet18'] |
似乎craft的性能比dbnet18要好,但不如dbnet50
参考: https://paperswithcode.com/sota/scene-text-detection-on-total-text?p=mask-textspotter-an-end-to-end-trainable

lx0bsm1f

lx0bsm1f1#

你好,
首先,感谢你提供的这个很棒的项目。我尝试使用DBNet文本检测器,但是我在以下代码片段中遇到了一个问题:为什么只有dbnet18可以使用,而dbnet18dbnet50不能使用?
EasyOCR/easyocr/easyocr.py
第84行:| | self.support_detection_network= ['craft', 'dbnet18'] |
看起来craft的性能比dbnet18要好,但不如dbnet50。参考:https://paperswithcode.com/sota/scene-text-detection-on-total-text?p=mask-textspotter-an-end-to-end-trainable

你好,
亲爱的surfiii3z,
在这个项目中,我在配置文件中看到了3个选项:craft、dbnet18和dbnet50。你可以尝试使用'detect_network'参数进行测试。
EasyOCR/easyocr/config.py
第11行到第27行:| | detection_models= { |
| | 'craft' : { |
| | 'filename': 'craft_mlt_25k.pth', |
| | 'url': 'https://github.com/JaidedAI/EasyOCR/releases/download/pre-v1.1.6/craft_mlt_25k.zip', |
| | 'md5sum': '2f8227d2def4037cdb3b34389dcf9ec1' |
| | }, |
| | 'dbnet18' : { |
| | 'filename': 'pretrained_ic15_res18.pt', |
| | 'url': 'https://github.com/JaidedAI/EasyOCR/releases/download/v1.6.0/pretrained_ic15_res18.zip', |
| | 'md5sum': 'aee04f8ffe5fc5bd5abea73223800425' |
| | }, |
| | 'dbnet50' : { |
| | 'filename': 'pretrained_ic15_res50.pt', |
| | 'url': 'https://github.com/JaidedAI/EasyOCR/releases/download/v1.6.0/pretrained_ic15_res50.zip', |
| | 'md5sum': 'a8e90144c131c2467d1eb7886c2e93a6' |
| | } |
| | } |

rhfm7lfc

rhfm7lfc2#

我正在遇到同样的问题,使用CPU的DBNet模型需要很长时间(比CRAFT模型慢10倍)。有趣的是,我尝试了https://github.com/mindee/doctr,它也使用了DBNet模型,大约需要3到5秒。

cigdeys3

cigdeys33#

@benjamintanweihao ,在你的Macpro M1上使用resent18模型需要多长时间?大约需要50秒。

6bc51xsx

6bc51xsx4#

在周三,2023年2月8日,19:53,lizhiling0609,***@***.***>写道: @benjamintanweihao < https://github.com/benjamintanweihao > 使用resent18模型在CPU上需要多长时间?在我Macpro M1上,大约需要50秒。 — 直接回复此电子邮件,查看GitHub上的<#872 (comment)>,或取消订阅< https://github.com/notifications/unsubscribe-auth/AAGSINAIXJTUIFVNAMLUE6DWWOCLBANCNFSM6AAAAAARB2RJMQ >。您收到此消息是因为您被提及。消息ID: ***@***.***>

suzh9iv8

suzh9iv85#

作为OP:配置的一部分,但未被代码接受(根据我的小样本,可以在不显著提高转录的情况下,从x5到x15重现dbnet18的缓慢性...)

相关问题