mirror of
https://github.com/open-mmlab/mmocr.git
synced 2025-06-03 21:54:47 +08:00
25 lines
595 B
Python
25 lines
595 B
Python
|
label_convertor = dict(
|
||
|
|
type='AttnConvertor', dict_type='DICT90', with_unknown=True)
|
||
|
|
|
||
|
|
model = dict(
|
||
|
|
type='SARNet',
|
||
|
|
backbone=dict(type='ResNet31OCR'),
|
||
|
|
encoder=dict(
|
||
|
|
type='SAREncoder',
|
||
|
|
enc_bi_rnn=False,
|
||
|
|
enc_do_rnn=0.1,
|
||
|
|
enc_gru=False,
|
||
|
|
),
|
||
|
|
decoder=dict(
|
||
|
|
type='ParallelSARDecoder',
|
||
|
|
enc_bi_rnn=False,
|
||
|
|
dec_bi_rnn=False,
|
||
|
|
dec_do_rnn=0,
|
||
|
|
dec_gru=False,
|
||
|
|
pred_dropout=0.1,
|
||
|
|
d_k=512,
|
||
|
|
pred_concat=True),
|
||
|
|
loss=dict(type='SARLoss'),
|
||
|
|
label_convertor=label_convertor,
|
||
|
|
max_seq_len=30)
|