21 lines
482 B
Python
21 lines
482 B
Python
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)
|