import fire
from omegaconf import OmegaConf
from data.dataset import ImprovedImageFolder, LMDBDataset

pipeline = """
pipeline:
    - Load
"""


def transform(dataset_path, save_path):
    print(save_path, dataset_path)
    conf = OmegaConf.create(pipeline)
    print(conf.pipeline.pretty())
    origin_dataset = ImprovedImageFolder(dataset_path, conf.pipeline)
    LMDBDataset.lmdbify(origin_dataset, conf.pipeline, save_path)


if __name__ == '__main__':
    fire.Fire(transform)