谷歌上个月推出的 AutoDraw ，这是一个能把你的随手涂鸦变成绘画的人工智能技术工具。谷歌也曾发布论文和博客介绍了其背后的技术，参见机器之心文章《谷歌官方揭秘 AutoDraw 人工智能绘画工具原理：让机器学会理解艺术》。实际上，AutoDraw 所用的技术基于谷歌先前的涂鸦实验「Quick, Draw!」。近日，谷歌发布了该项目背后的数据集，其中包含 5000 万张矢量画。机器之心对该项目的介绍文档进行了编译介绍。
数据集地址：https://github.com/googlecreativelab/quickdraw-dataset
数据集官网：https://quickdraw.withgoogle.com/data
Quick, Draw! 在线体验：https://quickdraw.withgoogle.com
AutoDraw 在线体验：https://www.autodraw.com
相关论文：https://arxiv.org/abs/1704.03477
Quick Draw 数据集是一个包含 5000 万张图画的集合，分成了 345 个类别，这些图画都来自于 Quick, Draw! 游戏的玩家。这些画都是加了时间戳的矢量图，并带有一些元数据标注，包括玩家被要求绘画的内容和玩家所在的国家。你可以在数据集官网浏览被识别出的绘画。
我们在这里将这个数据集共享给开发者、研究者和艺术家，以供探索、研究和学习。如果你使用这个数据集创造出了什么东西，请发邮件告知我们：quickdraw-support@google.com 或提交到 A.I. Experiments：https://aiexperiments.withgoogle.com/submit
请注意，尽管这个绘画集合中每一张都被审核过，但其中可能仍有不当内容。
原始的审核过的数据集
原始数据在 ndjson 文件中，并按类别进行了分割，按照如下格式：
每一张包含一副绘画，下面给出了一个例子：
{
"key_id":"5891796615823360",
"word":"nose",
"countrycode":"AE",
"timestamp":"2017-03-01 20:41:36.70725 UTC",
"recognized":true,
"drawing":[[[129,128,129,129,130,130,131,132,132,133,133,133,133,...]]]
}
绘画数组的格式如下：
[
[ // First stroke
[x0, x1, x2, x3, ...],
[y0, y1, y2, y3, ...],
[t0, t1, t2, t3, ...]
],
[ // Second stroke
[x0, x1, x2, x3, ...],
[y0, y1, y2, y3, ...],
[t0, t1, t2, t3, ...]
],
... // Additional strokes
]
其中 x 和 y 是像素坐标，t 是自第一个点开始后的时间，单位：毫秒。x 和 y 是实数值，而 t 是整型值。因为用于显示和输入的设备各不相同，原始绘画的边框大小和点的数量可能有非常大的差异。
预处理后的数据集
我们已经对该数据集进行了预处理，并且将其分成了不同的文件和格式，以便人们能更快更轻松地下载和使用。
简化图文件（.ndjson）
我们对这些矢量图进行了简化，移除了时间信息，并且对数据进行了定位和缩放处理，得到了大小 256×256 的图像。该数据以 ndjson 的格式导出，带有与原始格式一样的元数据。简化过程如下：
1. 将绘画对齐到左上角，最小值为 0；
2. 均匀缩放绘画，最大值为 255；
3. 以 1 像素的间隔重采样所有的笔画；
4. 使用 Ramer–Douglas–Peucker 算法简化所有的笔画，设置其 ε 值为 2.0
二进制文件（.bin）
我们也提供了简化后的绘画和元数据的定制二进制格式，可用于高效的压缩和加载。examples/binary_file_parser.py 给出了如何用 Python 加载该文件的示例。
Numpy 位图（.npy）
所有简化过的绘画也都被转换成了 28×28 的灰度位图，保存为 numpy 的 .npy 格式。该文件可以通过 np.load() 函数加载。这些图像是从简化过的数据中生成的，但对齐的是绘画边框的中心，而非左上角。
获取数据
该数据集在谷歌云存储服务中，在 ndjson 文件中分类存储。请参阅 Cloud Console 中的文件列表，你也可以使用其他方法使用它（参阅：https://cloud.google.com/storage/docs/access-public-data）。
数据集分类
Raw files (.ndjson)
Simplified drawings files (.ndjson)
Binary files (.bin)
Numpy bitmap files (.npy)
Sketch-RNN QuickDraw 数据集
这些数据也被用于训练 Sketch-RNN 模型。该模型开源的 TensorFlow 应用将在 Magenta Project 上近期推出。了解该模型可参阅文章《谷歌官方揭秘 AutoDraw 人工智能绘画工具原理：让机器学会理解艺术》。数据存储在压缩文件.npz 中，这种格式适合输入进循环神经网络中。在这一数据集中 75000 个用例（70000 用于训练、2500 个用于验证、2500 个用于测试）在每个类别里随机选择，使用值为 2.0 的 epsilon 参数经过 Ramer–Douglas–Peucker 算法线简化处理。
许可证
这些数据由谷歌提供，使用 Creative Commons Attribution 4.0 International 许可证。
共同学习，写下你的评论
评论加载中...
作者其他优质文章
