时间戳

首页 > 解决方案 > 扩展mnist数据库

python - 扩展mnist数据库

问题描述

我想知道如何扩展 mnist 数据库以识别数字和只有少数字符，如 A、B。我将本教程用于mnist，如何添加到此数据库字符 A 和 B？谢谢！

标签： pythontensorflowmnist

解决方案

你需要做三件事：

  1. 将字母 A 和 B 的附加图像添加到数据集（训练、验证和测试），标签为 11 和 12。扩展数据集的最简单方法可能是使用 EMNIST 数据集https://www.nist 的子集。 gov/itl/iad/image-group/emnist-dataset

  2. 将 logit 层上的输出数量更改为 12 以适应新字符（如果您想要“以上都不是”输出，则为 13，因为您打算在不需要明确识别的符号上进行训练）

    logits = tf.layers.dense（输入=辍学，单位=12）

  3. 从头开始重新训练网络，因为它在其他层中的权重仅在考虑数字的情况下生成，否则可能很难区分 8 和 B。但是，您可能会从先前训练的网络中进行迁移学习，尝试一下可能会很有趣.

推荐阅读