diff --git a/element_deeplabcut/plotting/dlc_plots.py b/element_deeplabcut/plotting/dlc_plots.py index c0867df..c5391d4 100644 --- a/element_deeplabcut/plotting/dlc_plots.py +++ b/element_deeplabcut/plotting/dlc_plots.py @@ -9,14 +9,15 @@ def plotting_results(pose_estimation_key: dict): plots_dir (Path): Path to the folder containing the plots """ import deeplabcut - from element_deeplabcut import dlc_reader, model + from element_deeplabcut import model + from element_deeplabcut.readers import dlc_reader output_dir = (model.PoseEstimationTask & pose_estimation_key).fetch1( "pose_estimation_output_dir" ) output_dir = model.find_full_path(model.get_dlc_root_data_dir(), output_dir) - dlc_result = dlc_reader.PoseEstimation(output_dir) + dlc_result = dlc_reader.PoseEstimation(output_dir.as_posix()) plots_dir = output_dir / "plots" plots_dir.mkdir(exist_ok=True)