diff --git a/element_facemap/facemap_inference.py b/element_facemap/facemap_inference.py index 59c79d4..5a5a5ed 100644 --- a/element_facemap/facemap_inference.py +++ b/element_facemap/facemap_inference.py @@ -221,7 +221,7 @@ def infer_output_dir(cls, key, relative=True, mkdir=True): root_dir = find_root_directory(get_facemap_root_data_dir(), video_dir) model_id = (FacemapPoseEstimationTask & key).fetch1("model_id") - processed_dir = Path(fbe.get_facemap_processed_data_dir()) + processed_dir = Path(get_facemap_processed_data_dir()) output_dir = ( processed_dir / video_dir.relative_to(root_dir) / f"facemap_{model_id}" )