TensorFlow读取数据在简单图像识别中的应用

• 作者: 来学伟
• 来源期刊: 现代信息科技
• 学科: 工学
• 关键词: MNIST数据集 Softmax回归 训练模型
• 年，卷（期）: 2019,（12）
• 所属期刊栏目: 计算机技术
• 页码范围: 98-99
• 页数: 2页
• 分类号: TP181
• 字数: 2110字
• 语种: 中文
• DOI: 10.3969/j.issn.2096-4706.2019.12.038

摘要

本文以MNIST数据库为例,用TensorFlow读取数据集中的数据,并建立一个简单的图像识别模型.同时以TensorFlow为工具,写一个手写体数字识别程序,使用的机器方法是Softmax回归.通过把占位符的值传递给会话,计算后运行梯度下降后,可以检测模型训练的结果,得到预测的标签和实际标签,接下来通过tf.equal函数来比较它们是否相等,并将结果保存到correct_prediction中.最后,用tf.reduce_mean可以计算数组中的所有元素的平均值,相当于得到了模型的预测准确率.该模型识别的准确率超过90％,具有一定的推广价值.

作者信息

序号	姓名	单位	发文数	被引次数	H指数	G指数
1	来学伟	三门峡职业技术学院信息传媒学院	43	58	5.0	6.0