mirrors
/
mmocr
mirror of https://github.com/open-mmlab/mmocr.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

add db

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
mmocr/models/textdet/detectors/dbnet.py
View File

@ -1,27 +1,12 @@
# Copyright (c) OpenMMLab. All rights reserved. # Copyright (c) OpenMMLab. All rights reserved.
from mmocr.registry import MODELS from mmocr.registry import MODELS
from .single_stage_text_detector import SingleStageTextDetector from .single_stage_text_detector import SingleStageTextDetector
from .text_detector_mixin import TextDetectorMixin
@MODELS.register_module() @MODELS.register_module()
class DBNet(TextDetectorMixin, SingleStageTextDetector): class DBNet(SingleStageTextDetector):
"""The class for implementing DBNet text detector: Real-time Scene Text """The class for implementing DBNet text detector: Real-time Scene Text
Detection with Differentiable Binarization. Detection with Differentiable Binarization.
[https://arxiv.org/abs/1911.08947]. [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)