Fix image size indexing when drawing keypoints

Corrects swapped im_size indexing when clamping ellipse coordinates in Display.draw. The previous code used im_size[1] for x1 and im_size[0] for y1, which reversed width/height fallbacks and could produce incorrect bounds; x1 now uses im_size[0] (width) and y1 uses im_size[1] (height). This prevents drawing outside the image or using wrong coordinates.

Author: C-Achard

dlclive/display.py

@@ -35,7 +35,9 @@ class Display:
 
     def __init__(self, cmap="bmy", radius=3, pcutoff=0.5):
         if not _TKINTER_AVAILABLE:
-            raise ImportError("tkinter is not available. Display functionality requires tkinter. ")
+            raise ImportError(
+                "tkinter is not available. Display functionality requires tkinter. "
+            )
         self.cmap = cmap
         self.colors = None
         self.radius = radius
@@ -93,12 +95,30 @@ def display_frame(self, frame, pose=None):
                 for j in range(pose.shape[1]):
                     if pose[i, j, 2] > self.pcutoff:
                         try:
-                            x0 = pose[i, j, 0] - self.radius if pose[i, j, 0] - self.radius > 0 else 0
-                            x1 = pose[i, j, 0] + self.radius if pose[i, j, 0] + self.radius < im_size[0] else im_size[1]
-                            y0 = pose[i, j, 1] - self.radius if pose[i, j, 1] - self.radius > 0 else 0
-                            y1 = pose[i, j, 1] + self.radius if pose[i, j, 1] + self.radius < im_size[1] else im_size[0]
+                            x0 = (
+                                pose[i, j, 0] - self.radius
+                                if pose[i, j, 0] - self.radius > 0
+                                else 0
+                            )
+                            x1 = (
+                                pose[i, j, 0] + self.radius
+                                if pose[i, j, 0] + self.radius < im_size[0]
+                                else im_size[0]
+                            )
+                            y0 = (
+                                pose[i, j, 1] - self.radius
+                                if pose[i, j, 1] - self.radius > 0
+                                else 0
+                            )
+                            y1 = (
+                                pose[i, j, 1] + self.radius
+                                if pose[i, j, 1] + self.radius < im_size[1]
+                                else im_size[1]
+                            )
                             coords = [x0, y0, x1, y1]
-                            draw.ellipse(coords, fill=self.colors[j], outline=self.colors[j])
+                            draw.ellipse(
+                                coords, fill=self.colors[j], outline=self.colors[j]
+                            )
                         except Exception as e:
                             print(e)
         img_tk = ImageTk.PhotoImage(image=img, master=self.window)