feat: add ability to load autolabels (#80)

* feat: add ability to load autolabels * fix: pr updates
TRI-ML · May 4, 2022 · de57b01 · de57b01
1 parent 6201e77
commit de57b01
Show file tree

Hide file tree

Showing 6 changed files with 463 additions and 22 deletions.
diff --git a/dgp/__init__.py b/dgp/__init__.py
@@ -30,7 +30,6 @@
 CLASSIFICATION_FOLDER = "classification"
 
 # Scene Directory structure constants
-AUTOLABEL_FOLDER = 'autolabels'
 CALIBRATION_FOLDER = 'calibration'
 ONTOLOGY_FOLDER = 'ontology'
 SCENE_JSON_FILENAME = 'scene.json'
@@ -47,3 +46,7 @@
 ANNOTATION_FILE_NAME = '{image_content_hash}_{annotation_content_hash}.json'
 TRI_DGP_AGENTS_JSON_NAME = "agents_{agent_hash}.json"
 TRI_DGP_AGENTS_SLICES_JSON_NAME = "agents_slices_{slice_hash}.json"
+
+# Autolabel constants
+AUTOLABEL_FOLDER = 'autolabels'
+AUTOLABEL_SCENE_JSON_NAME = 'scene.json'
diff --git a/dgp/annotations/depth_annotation.py b/dgp/annotations/depth_annotation.py
@@ -60,7 +60,7 @@ def save(self, save_dir):
             Full path to the output NPZ file.
         """
         pointcloud_path = os.path.join(save_dir, '{}.npz'.format(self.hexdigest))
-        np.savez_compressed(pointcloud_path, self.depth)
+        np.savez_compressed(pointcloud_path, data=self.depth)
         return pointcloud_path
 
     def render(self):