diff --git a/src/jabs_postprocess/compare_gt.py b/src/jabs_postprocess/compare_gt.py index 3cdc47e..2456eb1 100644 --- a/src/jabs_postprocess/compare_gt.py +++ b/src/jabs_postprocess/compare_gt.py @@ -454,7 +454,7 @@ def generate_iou_scan( # Aggregate over animals performance_df = ( performance_df.groupby(["stitch", "filter", "threshold"])[["tp", "fn", "fp"]] - .apply(np.sum) + .sum() .reset_index() ) # Re-calculate PR/RE/F1 diff --git a/src/jabs_postprocess/utils/metadata.py b/src/jabs_postprocess/utils/metadata.py index b18fbef..edf8871 100644 --- a/src/jabs_postprocess/utils/metadata.py +++ b/src/jabs_postprocess/utils/metadata.py @@ -5,7 +5,7 @@ import yaml -POSE_REGEX_STR = "_pose_est_v([2-6]).h5" +POSE_REGEX_STR = "_pose_est_v([2-8]).h5" PREDICTION_REGEX_STR = "_behavior.h5" FEATURE_REGEX_STR = "features.h5" DATE_REGEX_STR = "[0-9]{4}-[0-9]{2}-[0-9]{2}"