Fix for empty instances

talmolab · Dec 19, 2024 · 889fc5a · 889fc5a
1 parent 11af641
commit 889fc5a
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/sleap/nn/data/providers.py b/sleap/nn/data/providers.py
@@ -210,16 +210,16 @@ def py_fetch_lf(ind):
 
             for inst in insts:
 
-                if len(inst) > 0:
+                # Filter OOB
+                pts = inst.numpy()
+                pts[pts < 0] = np.NaN
 
-                    # Filter OOB
-                    pts = inst.numpy()
-                    pts[pts < 0] = np.NaN
+                pts[:, 0][pts[:, 0] > width - 1] = np.NaN
+                pts[:, 1][pts[:, 1] > height - 1] = np.NaN
 
-                    pts[:, 0][pts[:, 0] > width - 1] = np.NaN
-                    pts[:, 1][pts[:, 1] > height - 1] = np.NaN
+                instance = Instance.from_numpy(pts, inst.skeleton, inst.track)
 
-                    instance = Instance.from_numpy(pts, inst.skeleton, inst.track)
+                if len(instance) > 0:
 
                     if self.with_track_only:
                         if instance.track is not None: