create output directory and update task in pose estimation step

sidhulyalkar · Jan 5, 2024 · 76bf0dc · 76bf0dc
1 parent 9a49c20
commit 76bf0dc
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/element_facemap/facemap_inference.py b/element_facemap/facemap_inference.py
@@ -336,6 +336,15 @@ def make(self, key):
             "task_mode", "pose_estimation_output_dir"
         )
 
+        if not output_dir:
+            output_dir = FacemapPoseEstimationTask.infer_output_dir(
+                key, relative=True, mkdir=True
+            )
+            # update pose_estimation_output_dir
+            FacemapPoseEstimationTask.update1(
+                {**key, "pose_estimation_output_dir": output_dir.as_posix()}
+            )
+
         output_dir = find_full_path(fbe.get_facemap_root_data_dir(), output_dir)
         video_files = (FacemapPoseEstimationTask * fbe.VideoRecording.File & key).fetch(
             "file_path"