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

问题描述

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

output nodes of SavedModel

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

解决方法

暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!

如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@)