Filter oob while computing crop size

talmolab · Dec 17, 2024 · ed56b36 · ed56b36
1 parent f00afd6
commit ed56b36
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/sleap/nn/data/instance_cropping.py b/sleap/nn/data/instance_cropping.py
@@ -41,9 +41,15 @@ def find_instance_crop_size(
 
     # Calculate crop size
     min_crop_size_no_pad = min_crop_size - padding
+    height, width, _ = labels[0].image.shape
     max_length = 0.0
     for inst in labels.user_instances:
         pts = inst.points_array
+
+        pts[pts < 0] = np.NaN
+        pts[:, 0][pts[:, 0] > height - 1] = np.NaN
+        pts[:, 1][pts[:, 1] > width - 1] = np.NaN
+
         pts *= input_scaling
         max_length = np.maximum(max_length, np.nanmax(pts[:, 0]) - np.nanmin(pts[:, 0]))
         max_length = np.maximum(max_length, np.nanmax(pts[:, 1]) - np.nanmin(pts[:, 1]))