CBAM_MNIST.py

时间:2023-06-05 12:22:50
【文件属性】:

文件名称:CBAM_MNIST.py

文件大小:12KB

文件格式:PY

更新时间:2023-06-05 12:22:50

tensorflow 注意力机制

构建一个四层神经网络识别手写体数据集MNIST,然后将注意力模块CBAM插入到网络的第一层之后,查看注意力模块的性能。可以改变CBAM模块插入的位置,做到任意插入。


网友评论