Fix issue with pose generation not writing parent directory

PTG-Kitware · Dec 10, 2024 · 36df09d · 36df09d
1 parent cde2144
commit 36df09d
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tcn_hpl/data/utils/pose_generation/generate_pose_data.py b/tcn_hpl/data/utils/pose_generation/generate_pose_data.py
@@ -516,6 +516,7 @@ def img_done_cb(out_dset: kwcoco.CocoDataset) -> None:
         pose_model_device=pose_device,
     )
     output_dset = pg.predict_coco(input_dset, img_done_cb)
+    output_coco_filepath.parent.mkdir(parents=True, exist_ok=True)
     output_dset.dump(
         output_coco_filepath,
         newlines=True,