MMOCR / configs /_base_ /recog_models /robust_scanner.py
tomofi's picture
Add application file
2366e36
raw
history blame
671 Bytes
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)