PSEHead¶
- class mmocr.models.textdet.PSEHead(in_channels, hidden_dim, out_channel, module_loss={'type': 'PSEModuleLoss'}, postprocessor={'text_repr_type': 'poly', 'type': 'PSEPostprocessor'}, init_cfg=None)[source]¶
The class for PSENet head.
- Parameters
in_channels (list[int]) – A list of numbers of input channels.
hidden_dim (int) – The hidden dimension of the first convolutional layer.
out_channel (int) – Number of output channels.
module_loss (dict) – Configuration dictionary for loss type. Supported loss types are “PANModuleLoss” and “PSEModuleLoss”. Defaults to PSEModuleLoss.
postprocessor (dict) – Config of postprocessor for PSENet.
init_cfg (dict or list[dict], optional) – Initialization configs.
- Return type