
SENet-TensorFlow:在Cifar10数据集上应用(如ResNeXt,Inception-v4,Inception-ResNet...)
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
SENet-TensorFlow是基于TensorFlow实现的一种深度学习模型,专为Cifar10数据集优化,借鉴了ResNeXt、Inception-v4和Inception-ResNet等网络结构的优点。
我使用TensorFlow实现了SENet,并在Cifar10数据集上进行了简单实现。如果需要查看原始作者的代码,请自行查找相关资源。
该实现要求Tensorflow 1.x版本,Python 3.x环境以及tflearn(如果方便的话可以安装tflearn以简化全局平均池的操作)。
需要注意的是,在处理图片尺寸时遇到了一些问题:最初尝试使用ImageNet的数据集进行实验。然而由于Inception网络中对于图像大小的要求不同,我最终决定在Cifar10数据集中应用零填充来调整输入的尺寸:
```python
input_x = tf.pad(input_x, [[0, 0], [32, 32], [32, 32], [0, 0]]) # 尺寸为32x32
```
这样做的目的是为了确保输入图片符合模型对图像尺寸的要求。
全部评论 (0)
还没有任何评论哟~


