电子人社区›论坛 › 基础技术 › 人工智能 › TensorFlow开放社区 › TensorFlow CNN 测试CIFAR-10数据集

TensorFlow开放社区

今日 : 0|主题 : 276|排名 : 302

发帖回复收藏

TensorFlow CNN 测试CIFAR-10数据集

发表于 2018-1-30 16:06:57 | 739731 只看该作者

[复制链接]

发表于 2016-3-13 15:46:01 | 只看该作者回帖奖励

|倒序浏览 |阅读模式

电子人社区网讯：　　

本系列文章由 @yhl_leo 出品，转载请注明出处。

文章链接： http://blog.csdn.net/yhl_leo/article/details/50738311

　　
1 CIFAR-10 数据集
　　CIFAR-10数据集是机器学习中的一个通用的用于图像识别的基础数据集，官网链接为：The CIFAR-10 dataset

　　下载使用的版本是：

　　将其解压后（代码中包含自动解压代码），内容为：

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模型的预测性能　　[size=125%]
　　该部分的代码，介绍了如何使用TensorFlow在CPU和GPU上训练和评估卷积神经网络（convolutional neural network, CNN）。
3 相关网页及教程
　　更加详细地介绍说明，请浏览网页：Convolutional Neural Networks
　　中文网站极客学院也有该部分的汉译版：卷积神经网络
　　代码源自tensorflow官网：tensorflow/models/image/cifar10
4 代码修改说明
　　GitHub公布代码相对源码（本人的Tensorflow版本还是0.5），主要进行了以下修正：

cifar10.py

# indices = tf.reshape(tf.range(FLAGS.batch_size), [FLAGS.batch_size, 1])
indices = tf.reshape(range(FLAGS.batch_size), [FLAGS.batch_size, 1])
# or
indices = tf.reshape(tf.range(0, FLAGS.batch_size, 1), [FLAGS.batch_size, 1])
　　此处，源码编译时会出现以下错误：

...
File ".../cifar10.py", line 271, in loss
indices = tf.reshape(tf.range(FLAGS.batch_size), [FLAGS.batch_size, 1])
TypeError: range() takes at least 2 arguments (1 given)

cifar10_input_test.py

#self.assertEqual("%s:%d" % (filename, i), tf.compat.as_text(key))
import compat as cp
...
self.assertEqual("%s:%d" % (filename, i), cp.as_text(key))
　　不然的话，我测试的时候就会出现这的错误：

AttributeError: 'module' object has no attribute 'compat'