如何在 Python 中繪制 Google QuickDraw 數據集？

Question

我已經從 QuickDraw 數據集（JSON 格式）中提取了筆畫數據，但我不知道如何在 Python 中繪制它。 我嘗試了幾次，但根本不起作用。 我需要有關在 Python 中繪制筆畫的幫助。 以下是我提取筆划數據的代碼：

import json
file = open(filepath)
lines = file.readlines()

data = json.loads(lines[0])
strokes = data['drawing']
x = strokes[0][0]
y = strokes[0][1]

以下文本是數據文件的第一行：

{"word":"airplane","countrycode":"US","timestamp":"2017-03-08 21:12:07.26604      UTC","recognized":true,"key_id":"5152802093400064","drawing":[[[167,109,80,69,58,31,57,117,99,52,30,6,1,2,66,98,253,254,246,182,165],[140,194,227,232,229,229,206,124,123,149,157,159,153,110,82,77,74,109,121,127,120]],[[207,207,210,221,238],[74,103,114,128,135]],[[119,107,76,70,49,39,60,93],[72,41,3,0,1,5,38,70]]]}

Answer 1

除了 Swati 的回答之外，還有一個quickdraw python 包可以讓這一切變得簡單：

from quickdraw import QuickDrawData

qd = QuickDrawData()

anvil = qd.get_drawing("anvil")
anvil.image.show()

如果你想手動完成，你可以zip()所有的 x 值和 y 值並渲染它（使用 PIL 或其他任何東西）：

import json
from PIL import Image, ImageDraw

# read ndjson lines
lines = open('full_simplified_cat.ndjson','r').readlines()
# grab the first line, JSON parse it and fetch the 'drawing' array
raw_drawing = json.loads(lines[0])['drawing']

print('before',raw_drawing)
# zip x,y coordinates for each point in every polyline
polylines = (zip(polyline[0], polyline[1]) for polyline in raw_drawing)
# notice how the data is shuffled to (x1,y1),(x2,y2) order
print('after',polylines)

# make a new image
pil_img = Image.new("RGB", (240, 270), (255,255,255))
# get a drawing context
d = ImageDraw.Draw(pil_img)
# render each polyline
for polyline in polylines:
    d.line(polyline,fill=(0, 0, 0),width=3)
# display image
pil_img.show()

Answer 2

查看https://colab.research.google.com/github/zaidalyafeai/Notebooks/blob/master/QuickDraw10.ipynb#scrollTo=Rxktd50yL-Za

它包含數據的所有預處理、訓練和數據測試。