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

refine

Browse Source
This commit is contained in:
MissPenguin
2021-06-22 08:14:07 +00:00
parent 97a668747c
commit 79e83a8d6c
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
ppocr/modeling/heads/rec_ctc_head.py
+6 -1
View File
@@ -45,7 +45,12 @@ class CTCHead(nn.Layer):
self.out_channels = out_channels
def forward(self, x, targets=None):
predicts = self.fc(x)
if self.mid_channels is None:
predicts = self.fc(x)
else:
predicts = self.fc1(x)
predicts = self.fc2(predicts)
if not self.training:
predicts = F.softmax(predicts, axis=2)
return predicts