- 博客(7)
- 收藏
- 关注
原创 将注意力机制加入CNN网络中
在使用CNN网络进行特征提取之后再加入注意力机制是一个合理且常见的策略。这样做可以结合CNN的局部特征提取能力和注意力机制的全局依赖建模能力,进一步提升模型的性能。
2024-08-22 10:21:16
136
原创 自注意力机制
这个Attention模块实现了一个带有多头机制的自注意力层,常用于Transformer架构中。它通过线性变换将输入数据生成查询、键、值向量,计算注意力权重,再使用这些权重对值向量进行加权求和,最终通过一个线性层投影输出结果。这个模块的目的是让每个位置(如词、图像块)根据序列中的其他位置动态地调整自己的表示,捕捉序列内部的依赖关系。
2024-08-21 20:26:17
687
原创 查看npz文件,图像
如果数据类型是 object,需要进一步处理。# 尝试将对象数组转换为数值数组。# 检查是否是图像数据(3D 或 2D)# 列出 npz 文件中所有的数组。# 查看每个数组的类型和形状。# 加载 npz 文件。
2024-08-08 14:30:33
73
原创 报错解决怎么搞
ImportError: attempted relative import with no known parent package
2024-07-31 13:28:47
292
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人