
使用TensorFlow 2.1实现CIFAR-10的ResNet、SENet和Inception模型训练,准确率达88.6%
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
本项目采用TensorFlow 2.1框架,成功实现了ResNet、SENet及Inception三种深度学习架构在CIFAR-10数据集上的高效训练,并达到了88.6%的高精度识别率。
环境:TensorFlow 2.1,推荐使用GPU。
模型:
- ResNet:将前一层的数据直接传递到下一层,以减少数据在传播过程中丢失。
- SENet:学习每一层通道之间的关系。
- Inception:每层采用不同大小的卷积核(如1×1、3×3和5×5)来防止因使用过小或过大卷积核而无法捕捉图片特征。
使用ResNet,SENet和Inception网络在Cifar10 或 Cifar 100上进行训练。具体表现如下:
- 训练集准确率:约97.11%
- 验证集准确率:约90.22%
- 测试集准确率:88.6%
使用GPU时,训练时间约为一小时多。
权重大小为21。
全部评论 (0)
还没有任何评论哟~


