keras cnn注意力机制_keras, TensorFlow中加入注意力机制

第一步：找到要修改文件的源代码

在里面添加通道注意力机制和空间注意力机制

所需库

from 
           

通道注意力机制

def 
           

空间注意力机制

def 
           

构建CBAM

def 
           

在相应的位置添加CBAM

inputs 
           

这样就在任意位置加入了注意力机制啦。