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,