transformer学习笔记II

transformer做ocr任务

transformer的结构天然适合做seq2seq的学习任务,直觉上讲,格式标准的图像任务需要做一系列转换才能使用transformer任务,或者说在输入侧适配encoder。ocr任务包含解码的过程,因此在输出侧使用掩码可以适配transformer decoder。

看完datawhale 教程,最为tricky的一部分是作者观察到宽高1:1的原图范围内对应着3个英文字符,因此将图片宽高设置为3:1,保证经过resnet18 感受野降低32倍后能够使得feature map每个点(512维)对应一个英文字符。

暂时还未手撕代码,先挖个坑待填充。没有看懂在模型训练完成前后decoder操作机制的不同,在训练的时候可以用上三角矩阵mask attention,而在输出的时候是使用rnn的思路,不断输入隐变量然后argmax解码出来字符?如果是这样的话,又如何处理字符长度不定长的问题呢?

引用

datawhale transformer 做ocr教程

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
李宏毅是一位著名的机器学习深度学习专家,他在教学视频中也提到了Transformer模型。下面是一些关于李宏毅关于Transformer笔记总结: 1. Transformer 是一种基于注意力机制(attention mechanism)的序列到序列(sequence-to-sequence)模型。它在自然语言处理任务中取得了很大的成功。 2. Transformer 模型的核心思想是完全摒弃了传统的循环神经网络(RNN)结构,而是采用了自注意力机制(self-attention mechanism)来建模输入序列之间的依赖关系。 3. 自注意力机制能够将输入序列中的每个位置与其他位置建立联系,从而捕捉到全局上下文的信息。它能够解决传统的RNN模型在处理长序列时的梯度消失和梯度爆炸问题。 4. Transformer 模型由编码器(Encoder)和解码器(Decoder)两部分组成。编码器负责将输入序列表示为高维向量,解码器则根据编码器的输出生成目标序列。 5. 编码器和解码器由多个层堆叠而成,每一层都包含了多头自注意力机制和前馈神经网络。多头自注意力机制可以并行地学习输入序列中不同位置之间的关系。 6. Transformer 模型还引入了残差连接(residual connection)和层归一化(layer normalization)来帮助模型更好地进行训练和优化。 这些是李宏毅关于Transformer的一些主要笔记总结,希望对你有所帮助。注意,这些总结仅代表了我对李宏毅在其教学视频中所讲述内容的理解,如有误差请以李宏毅本人的观点为准。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值