from mmdet.models.builder import DETECTORS
from mmocr.models.textdet.detectors.single_stage_text_detector import \
SingleStageTextDetector
from mmocr.models.textdet.detectors.text_detector_mixin import \
TextDetectorMixin
[docs]@DETECTORS.register_module()
class PSENet(TextDetectorMixin, SingleStageTextDetector):
"""The class for implementing PSENet text detector: Shape Robust Text
Detection with Progressive Scale Expansion Network.
[https://arxiv.org/abs/1806.02559].
"""
def __init__(self,
backbone,
neck,
bbox_head,
train_cfg=None,
test_cfg=None,
pretrained=None,
show_score=False):
SingleStageTextDetector.__init__(self, backbone, neck, bbox_head,
train_cfg, test_cfg, pretrained)
TextDetectorMixin.__init__(self, show_score)