def test_processing(): inp = 'raw_data/test.json' outp = "raw_data/test.npz" process_creativebirds_json(in_path=inp, out_path=outp) a = np.load(outp, allow_pickle=True, encoding='latin1') drawing = Drawing.from_npz_data(a['train'][1]) drawing.plot()
def test_one(idx=0, in_path='raw_data/test.json'): data = json.loads(open(in_path).read()) # print(data) cc = CreativeItem.from_json_item(data[idx]) cc.plot() ll = LineStringList.fromCreativeItem(cc) ll.simplify() ll.normalize() ll.plot() print(ll.linestring_list[:5]) dd = DatasetItem.fromLineStringList(ll) print(dd.embedding[:5]) dd.plot() print(dd.embedding.shape) print(dd.embedding.dtype) drawing = Drawing.from_npz_data(dd.embedding) drawing.plot()