如何使用 Tensorflow aocr API 从 SavedModel 生成注意力掩码?

时间:2021-07-15 15:33:59

标签: tensorflow ocr attention-model

我已经使用 Tensorflow AOCR API 训练了一个模型。接下来,我将其导出为 SavedModel 格式。作为输出节点,我得到每个字符的 attention_masks (attention_mask_0 ... n)。请参阅附图。

output nodes of SavedModel

为什么掩码是一维的,如何生成掩码以查看每个字符的注意力?

0 个答案:

没有答案