opensource/PaddleOCR
3
0
Fork 0
mirror of https://github.com/PaddlePaddle/PaddleOCR.git synced 2026-09-24 23:33:08 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #1890 from tink2123/srn_ch

Browse Source 
Adaptation of Chinese char
This commit is contained in:
xiaoting
2021-02-01 11:31:25 +08:00
committed by GitHub
parent 2a0c3d4dac 7f2304ab3f
commit 0c5c9f694d
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
ppocr/modeling/heads/rec_srn_head.py
+1 -1
View File
@@ -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