pull/1178/head
gaotongxiao 2022-05-25 17:53:00 +08:00
parent a4952a6dd6
commit d34fad1451
1 changed files with 1 additions and 16 deletions

View File

@ -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)