通道注意力机制keras_深度残差收缩网络：一种新的深度注意力机制算法

#!/usr/bin/env python3#-*- coding: utf-8 -*-

"""Created on Mon Dec 23 21:23:09 2019

Implemented using TensorFlow 1.0 and TFLearn 0.3.2

M. Zhao, S. Zhong, X. Fu, B. Tang, M. Pecht, Deep Residual Shrinkage Networks for Fault Diagnosis,

IEEE Transactions on Industrial Informatics, 2019, DOI: 10.1109/TII.2019.2943898

@author: super_9527"""

from __future__ importdivision, print_function, absolute_importimporttflearnimportnumpy as npimporttensorflow as tffrom tflearn.layers.conv importconv_2d#Data loading

from tflearn.datasets importcifar10

(X, Y), (testX, testY)=cifar10.load_data()#Add noise

X = X + np.random.random((50000, 32, 32, 3))*0.1testX= testX + np.random.random((10000, 32, 32, 3))*0.1

#Transform labels to one-hot format

Y = tflearn.data_utils.to_categorical(Y,10)

testY= tflearn.data_uti