mirror of
https://github.com/open-mmlab/mmocr.git
synced 2025-06-03 15:44:26 +08:00
25 lines
671 B
Python
25 lines
671 B
Python
label_convertor = dict(
|
|
type='AttnConvertor', dict_type='DICT90', with_unknown=True)
|
|
|
|
hybrid_decoder = dict(type='SequenceAttentionDecoder')
|
|
|
|
position_decoder = dict(type='PositionAttentionDecoder')
|
|
|
|
model = dict(
|
|
type='RobustScanner',
|
|
backbone=dict(type='ResNet31OCR'),
|
|
encoder=dict(
|
|
type='ChannelReductionEncoder',
|
|
in_channels=512,
|
|
out_channels=128,
|
|
),
|
|
decoder=dict(
|
|
type='RobustScannerDecoder',
|
|
dim_input=512,
|
|
dim_model=128,
|
|
hybrid_decoder=hybrid_decoder,
|
|
position_decoder=position_decoder),
|
|
loss=dict(type='SARLoss'),
|
|
label_convertor=label_convertor,
|
|
max_seq_len=30)
|