init_code

data_utils/gather_renders.py

@@ -0,0 +1,25 @@
+"""In simulated dataset, gather all the scene images in one folder"""
+
+import pygarment.data_config as config
+from pathlib import Path
+import shutil
+
+from pattern_data_sim import gather_renders
+
+
+system_props = config.Properties('./system.json')
+dataset = 'unpacking_test'
+datapaths = [
+    Path(system_props['output']) / dataset / 'default_body', 
+    Path(system_props['output']) / dataset / 'random_body'
+]
+
+for datapath in datapaths:
+    # Check packing
+    tar_path = datapath / 'data.tar.gz'
+    if tar_path.exists():
+        shutil.unpack_archive(tar_path, datapath)
+        # Finally -- clean up
+        tar_path.unlink()
+
+    gather_renders(datapath)