quick draw数据集

最新推荐文章于 2025-01-07 15:14:49 发布

weixin_lld

最新推荐文章于 2025-01-07 15:14:49 发布

阅读量464

点赞数

本文链接：https://blog.csdn.net/weixin_40829516/article/details/125079813

版权

ndjson json转换图像处理 png numpy

关键词由CSDN通过智能技术生成

这段代码实现了将.ndjson文件转换为.json文件，并对数据进行处理，包括将json数据转化为图像并保存为.png格式。首先，使用Node.js的ndjson库解析.ndjson文件，然后将数据保存为.json。接着，代码遍历.json文件中的绘图数据，生成对应的.png图像。最后，对于.npy文件，同样读取数据并转换为.png图像。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

.ndjson转为.json

var fs = require('fs');
var ndjson = require('ndjson'); // npm install ndjson

function parseSimplifiedDrawings(fileName, callback) {
  var drawings = [];
  var fileStream = fs.createReadStream(fileName)
  fileStream
    .pipe(ndjson.parse())
    .on('data', function(obj) {
      drawings.push(obj)
    })
    .on("error", callback)
    .on("end", function() {
      callback(null, drawings)
    });
}

parseSimplifiedDrawings("G:/quickdraw/full_simplified_airplane.ndjson", function(err, drawings) {
  if(err) return console.error(err);
  drawings.forEach(function(d) {
    // Do something with the drawing
    console.log(d.key_id, d.countrycode);
  })
  console.log("# of drawings:", drawings);
  var filename = "G:/quickdraw/full_simplified_airplane.json";//这里保存
  fs.writeFileSync(filename, JSON.stringify(drawings));//这里保存
})

打开cmd，node test.js

json转png


import json
from scipy import interpolate
import pylab as pl
f = open("G:/quickdraw/full_simplified_airplane.json")
setting = json.load(f)
for j in range(0,len(setting)): #转化保存1000个图
    for i in range(0,len(setting[j]['drawing'])):
        x = setting[j]['drawing'][i][0]
        y = setting[j]['drawing'][i][1]
        f=interpolate.interp1d(x,y,kind="slinear")
        pl.plot(x,y,'k')
    ax = pl.gca()
    ax.xaxis.set_ticks_position('top')
    ax.invert_yaxis()
    pl.axis('off')
    pl.savefig("G:/quickdraw/data/%d.png"%j)
    pl.close()

.npy转png

import matplotlib.pyplot as plt
import numpy as np
import scipy.misc
import os

file_dir = "G:/quickdraw/"  # npy文件路径
dest_dir = "G:/quickdraw/data1/"  # 文件存储的路径


def npy_png(file_dir, dest_dir):
    # 如果不存在对应文件，则创建对应文件
    if not os.path.exists(file_dir):
        os.makedirs(file_dir)
    if not os.path.exists(dest_dir):
        os.makedirs(dest_dir)

    file = file_dir + 'airplane.npy'  # npy文件
    con_arr = np.load(file)  # 读取npy文件
    for i in range(0, 151623):  # 图片张数
        arr = con_arr[i, :]  # 获得第i张的单一数组
        disp_to_img = arr.reshape(28, 28)  # 根据需要的尺寸进行修改
        plt.imsave(os.path.join(dest_dir, "{}_disp.png".format(i)), disp_to_img, cmap='plasma')  # 定义命名规则，保存图片为彩色模式
        print('photo {} finished'.format(i))


if __name__ == "__main__":
    npy_png(file_dir, dest_dir)