SENet-Tensorflow
使用Cifar10的简单Tensorflow实现
我实现了以下SENet
如果您想查看原始作者的代码,请参考此
要求
Tensorflow 1.x
Python 3.x
tflearn(如果您易于使用全局平均池,则应安装tflearn )
问题
图片尺寸
在纸上,尝试了ImageNet
但是,由于Inception网络中的图像大小问题,因此我对Cifar10使用零填充
input_x = tf . pad ( input_x , [[ 0 , 0 ], [ 32 , 32 ], [ 32 , 32 ], [ 0 , 0 ]]) # size 32x32
1