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

作者：来学伟

来源：现代信息科技, 2019, 3(12): 98-99.

• MNIST数据集
• Softmax回归
• 训练模型

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

• 单位
  三门峡职业技术学院