PointAR/gen_data.py at main · cake-lab/PointAR · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
"""PointAR dataset generation code
"""
import fire
import multiprocessing
from datasets.pointar.preprocess.pack import pack
from datasets.pointar.preprocess.generate import generate


def gen_data():

    print("Generating test dataset")
    generate(
        'test',  # dataset name ["train", "test"],
        index='all',  # data item, 'all' for all items, 0 for data item 0
    )

    # print("Packing test dataset")
    # pack('test', index="all")

    # print("Generating training dataset")
    # generate(
    #     "train",  # dataset name ["train", "test"],
    #     index='all',  # data item, 'all' for all items, 0 for data item 0
    # )

    # print("Packing training dataset")
    # pack("train", index="all")


if __name__ == "__main__":
    multiprocessing.set_start_method("spawn")
    fire.Fire(gen_data)