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

opt rec_att_head

Browse Source
This commit is contained in:
LDOUBLEV
2021-02-01 08:08:18 +00:00
parent 0d89f3f913
commit a094d27755
1 changed files with 8 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
ppocr/modeling/heads/rec_att_head.py
+8 -4
View File
@@ -64,8 +64,10 @@ class AttentionHead(nn.Layer):
(outputs, hidden), alpha = self.attention_cell(hidden, inputs,
char_onehots)
probs_step = self.generator(outputs)
probs = paddle.unsqueeze(
probs_step, axis=1) if probs is None else paddle.concat(
if probs is None:
probs = paddle.unsqueeze(probs_step, axis=1)
else:
probs = paddle.concat(
[probs, paddle.unsqueeze(
probs_step, axis=1)], axis=1)
next_input = probs_step.argmax(axis=1)
@@ -152,8 +154,10 @@ class AttentionLSTM(nn.Layer):
char_onehots)
probs_step = self.generator(hidden[0])
hidden = (hidden[1][0], hidden[1][1])
probs = paddle.unsqueeze(
probs_step, axis=1) if probs is None else paddle.concat(
if probs is None:
probs = paddle.unsqueeze(probs_step, axis=1)
else:
probs = paddle.concat(
[probs, paddle.unsqueeze(
probs_step, axis=1)], axis=1)