TensorFlow CNN 测试CIFAR-10数据集

TensorFlow CNN 测试CIFAR-10数据集




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


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



1 CIFAR-10 数据

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

cifar10

下载使用的版本是:

version

将其解压后(代码中包含自动解压代码),内容为:

cifar10 data

cifar10 data2

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

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'
  • cifar10_train.pycifar10_multi_gpu_train.py

源代码里的最大迭代次数max_steps1000000,需要训练几个小时,不忍心折腾我的破笔记本,就改为了20000

其他改动,例如导入模块或者文件路径等,都很容易理解,就不列举了~

运行结果,与官网上公布的一致,也不再列举。附上一张运行结果截图:

cifartrain

转载于:https://www.cnblogs.com/hehehaha/p/6332160.html

版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 举报,一经查实,本站将立刻删除。

发布者:全栈程序员-用户IM,转载请注明出处:https://javaforall.cn/109139.html原文链接:https://javaforall.cn

【正版授权,激活自己账号】: Jetbrains全家桶Ide使用,1年售后保障,每天仅需1毛

【官方授权 正版激活】: 官方授权 正版激活 支持Jetbrains家族下所有IDE 使用个人JB账号...

(0)


相关推荐

  • react 纯函数组件_react类组件

    react 纯函数组件_react类组件纯函数PureFunction定义:一个函数的返回结果只依赖于它的参数,并且在执行的过程中没有副作用,我们就把该函数称作纯函数。特点1.函数的返回结果只依赖于它的参数。letfoo=(a,b)=>a+bfoo(1,2)//=32.函数执行过程里面没有副作用。什么是副作用除了修改外部的变量,一个函数在执行过程中还有很多方式产生外部可观察的变化,比如说调用DOMAPI修改页面,或者你发送了Ajax请求,还有调用window.reload刷新浏览器,

    2022年10月28日
  • js的数组拼接「建议收藏」

    js的数组拼接「建议收藏」js的数组拼接方法方法一:concat方法拼接(返回一个新数组)vara1=[1,2,3,4,5];vara2=[‘a’,’b’,’c’];varnewa=a1.concat(a2);返回结果:{“0″:1,”1″:2,”2″:3,”3″:4,”4″:5,”5″:”a”,”6″:”b”,”7″:”c”,”length”:8}结果类型:objectconcat方法:用于连接两个或多个数组,生成新数组,concat后面的数组时操作的时候数组的元素,而不是数组方.

  • CCriticalSection的使用

    CCriticalSection的使用CCriticalSection是对关键段CRITICAL_SECTION的封装。 关键段(critivalsection)是一小段代码,他在执行之前需要独占对一些共享资源的访问权。这种方式可以让多行代码以“原子方式”来对资源进行操控。这里的“原子方式”,指的是代码知道除了当前线程之外没有其他任何线程会同时访问该资源。当然,系统仍然可以暂停当前线程去调度其他线程。但是,在当前线程离开关键

  • pfx 证书 转 jks 证书

    pfx 证书 转 jks 证书今天在tomcat配置httpspfx证书时总是配置失败很是头疼,配置参数如下:<Connectorport=”443″protocol=”org.apache.coyote.http11.Http11Protocol”SSLEnabled=”true” maxThreads=”150″scheme=”https”secure=”true” keyst…

  • 网站服务器建立数据库连接时出错,WordPress提示建立数据库连接出错的解决办法…[通俗易懂]

    网站服务器建立数据库连接时出错,WordPress提示建立数据库连接出错的解决办法…[通俗易懂]很多新手使用wordpress程序建站初期,会遇到页面提示:建立数据库连接出错,英文提示:“Errorestablishingadatabaseconnection”。那么遇到这种情况大家会很头疼,我到底哪里操作不对呀?本文中老魏就实际遇到的此类情况做详细解释。出错的原因有几种,从字面意思能看出来和数据库无法正常读取有关。也许是因为数据库登陆信息不对,也许是数据库挂了无法读取等等。Wor…

  • 图解Linux命令之–dos2unix命令

    图解Linux命令之–dos2unix命令dos2unix命令–&gt;DostoUNIX的缩写,用来将DOS格式的文本文件转换为UNIX格式find.-typef-execdos2unix{}\;find.-typef|xargs|dos2unix备注:1)为什么需要用dos2unix进行格式转换在Linux系统下经常会出现下列提示:/bin/sh^M:badinter…

发表回复

您的电子邮箱地址不会被公开。

关注全栈程序员社区公众号