From 36df09d1d85b17f90dea169911a07bd6054eaa8b Mon Sep 17 00:00:00 2001 From: Paul Tunison Date: Tue, 10 Dec 2024 16:32:28 -0500 Subject: [PATCH] Fix issue with pose generation not writing parent directory --- tcn_hpl/data/utils/pose_generation/generate_pose_data.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tcn_hpl/data/utils/pose_generation/generate_pose_data.py b/tcn_hpl/data/utils/pose_generation/generate_pose_data.py index eba6eea8e..6effb8114 100755 --- 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,