TensorFlow-CNN CIFAR-10数据集 学习
1 CIFAR-10 数据集
CIFAR-10数据集是机器学习中的一个通用的用于图像识别的基础数据集,官网链接为:The CIFAR-10 dataset,官方教程Convolutional Neural Networks
image的个数:60000
image的大小:32*32*3
class的个数:10 (飞机;汽车;鸟;猫;鹿;狗;青蛙;马;船;卡车)
注:卡车的类中只存在卡车,皮卡和suv都不在此类中。
下载使用的版本是:
将其解压后(代码中包含自动解压代码),内容为:
cifar-10 dataset的组成:
train batch有5个
test batch有1个
batches.meta为class的名字
每个batch的有10000个data和相对应的label
label从0到9
每一个data的大小为3073个字节,等于32*32*3。
2 测试代码
测试代码公布在GitHub:yhlleo
主要代码及作用:
文件 | 作用 |
---|---|
cifar10_input.py
|
读取本地或者在线下载CIFAR-10的二进制文件格式数据集 |
cifar10.py
|
建立CIFAR-10的模型 |
cifar10_train.py
|
在CPU或GPU上训练CIFAR-10的模型 |
cifar10_multi_gpu_train.py
|
在多个GPU上训练CIFAR-10的模型 |
cifar10_eval.py
|
评估CIFAR-10模型的预测性能 |
该部分的代码,介绍了如何使用TensorFlow在CPU和GPU上训练和评估卷积神经网络(convolutional neural network, CNN)。
3 相关网页及教程
更加详细地介绍说明,请浏览网页:Convolutional Neural Networks
中文网站极客学院也有该部分的汉译版:卷积神经网络
代码源自tensorflow官网:tensorflow/models/image/cifar10
-
cifar10_train.py
和cifar10_multi_gpu_train.py
源代码里的最大迭代次数max_steps
为1000000
,需要训练几个小时,不忍心折腾我的破笔记本,就改为了20000
。
其他改动,例如导入模块或者文件路径等,都很容易理解,就不列举了~
原文:http://blog.****.net/yhl_leo/article/details/50738311
http://blog.****.net/zeuseign/article/details/72773342
http://blog.****.net/muyiyushan/article/details/64124365