add lmdb dataset support and EpisodicDataset

tool/lmdbify.py

@@ -0,0 +1,20 @@
+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)