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

第一步：找到要修改檔案的源代碼

在裡面添加通道注意力機制和空間注意力機制

所需庫

from 
           

通道注意力機制

def 
           

空間注意力機制

def 
           

建構CBAM

def 
           

在相應的位置添加CBAM

inputs 
           

這樣就在任意位置加入了注意力機制啦。