Spaces:
Runtime error
Runtime error
File size: 1,435 Bytes
2366e36 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 |
# Copyright (c) OpenMMLab. All rights reserved.
from mmocr.models.builder import HEADS
from . import PANHead
@HEADS.register_module()
class PSEHead(PANHead):
"""The class for PSENet head.
Args:
in_channels (list[int]): A list of 4 numbers of input channels.
out_channels (int): Number of output channels.
downsample_ratio (float): Downsample ratio.
loss (dict): Configuration dictionary for loss type. Supported loss
types are "PANLoss" and "PSELoss".
postprocessor (dict): Config of postprocessor for PSENet.
train_cfg, test_cfg (dict): Depreciated.
init_cfg (dict or list[dict], optional): Initialization configs.
"""
def __init__(self,
in_channels,
out_channels,
downsample_ratio=0.25,
loss=dict(type='PSELoss'),
postprocessor=dict(
type='PSEPostprocessor', text_repr_type='poly'),
train_cfg=None,
test_cfg=None,
init_cfg=None,
**kwargs):
super().__init__(
in_channels=in_channels,
out_channels=out_channels,
downsample_ratio=downsample_ratio,
loss=loss,
postprocessor=postprocessor,
train_cfg=train_cfg,
test_cfg=test_cfg,
init_cfg=init_cfg,
**kwargs)
|