def cbam(inputs):
inputs_channels=int(inputs.shape[-1])
x=GlobalAveragePooling2D()(inputs)
x=Dense(int(inputs_channels/4))(x)
x=Activation(relu)(x)
x=Dense(int(inputs_channels))(x)
x=Activation(softmax)(x)
x=Reshape((1,1,inputs_channels))(x)
x=Multply()([inputs,x])
return x
keras注意力机制 cnn图像
最新推荐文章于 2024-09-04 14:45:51 发布