refine

2021-06-22 08:14:07 +00:00 · 2021-06-22 08:14:07 +00:00 · 79e83a8d6c
parent 97a668747c
commit 79e83a8d6c
1 changed files with 6 additions and 1 deletions
--- a/ppocr/modeling/heads/rec_ctc_head.py
+++ b/ppocr/modeling/heads/rec_ctc_head.py
@ -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