mirror of https://github.com/open-mmlab/mmocr.git
add db
parent
a4952a6dd6
commit
d34fad1451
|
@ -1,27 +1,12 @@
|
|||
# Copyright (c) OpenMMLab. All rights reserved.
|
||||
from mmocr.registry import MODELS
|
||||
from .single_stage_text_detector import SingleStageTextDetector
|
||||
from .text_detector_mixin import TextDetectorMixin
|
||||
|
||||
|
||||
@MODELS.register_module()
|
||||
class DBNet(TextDetectorMixin, SingleStageTextDetector):
|
||||
class DBNet(SingleStageTextDetector):
|
||||
"""The class for implementing DBNet text detector: Real-time Scene Text
|
||||
Detection with Differentiable Binarization.
|
||||
|
||||
[https://arxiv.org/abs/1911.08947].
|
||||
"""
|
||||
|
||||
def __init__(self,
|
||||
backbone,
|
||||
neck,
|
||||
bbox_head,
|
||||
train_cfg=None,
|
||||
test_cfg=None,
|
||||
pretrained=None,
|
||||
show_score=False,
|
||||
init_cfg=None):
|
||||
SingleStageTextDetector.__init__(self, backbone, neck, bbox_head,
|
||||
train_cfg, test_cfg, pretrained,
|
||||
init_cfg)
|
||||
TextDetectorMixin.__init__(self, show_score)
|
||||
|
|
Loading…
Reference in New Issue