From 7f2304ab3fbb2846b365f8293b4835a3120160df Mon Sep 17 00:00:00 2001 From: tink2123 Date: Mon, 1 Feb 2021 03:28:31 +0000 Subject: [PATCH] Adaptation of Chinese char --- ppocr/modeling/heads/rec_srn_head.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ppocr/modeling/heads/rec_srn_head.py b/ppocr/modeling/heads/rec_srn_head.py index 8aaf65e1..d2c7fc02 100644 --- a/ppocr/modeling/heads/rec_srn_head.py +++ b/ppocr/modeling/heads/rec_srn_head.py @@ -246,7 +246,7 @@ class SRNHead(nn.Layer): num_encoder_tus=self.num_encoder_TUs, num_decoder_tus=self.num_decoder_TUs, hidden_dims=self.hidden_dims) - self.vsfd = VSFD(in_channels=in_channels) + self.vsfd = VSFD(in_channels=in_channels, char_num=self.char_num) self.gsrm.wrap_encoder1.prepare_decoder.emb0 = self.gsrm.wrap_encoder0.prepare_decoder.emb0