Skip to content

Commit 822f5d4

Browse files
committed
Use max/min for pose coordinate clamping
Replace verbose conditional expressions with max/min to clamp pose-based ellipse coordinates (x0, x1, y0, y1) within image bounds.
1 parent 03fe25a commit 822f5d4

File tree

1 file changed

+4
-20
lines changed

1 file changed

+4
-20
lines changed

dlclive/display.py

Lines changed: 4 additions & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -95,26 +95,10 @@ def display_frame(self, frame, pose=None):
9595
for j in range(pose.shape[1]):
9696
if pose[i, j, 2] > self.pcutoff:
9797
try:
98-
x0 = (
99-
pose[i, j, 0] - self.radius
100-
if pose[i, j, 0] - self.radius > 0
101-
else 0
102-
)
103-
x1 = (
104-
pose[i, j, 0] + self.radius
105-
if pose[i, j, 0] + self.radius < im_size[0]
106-
else im_size[0]
107-
)
108-
y0 = (
109-
pose[i, j, 1] - self.radius
110-
if pose[i, j, 1] - self.radius > 0
111-
else 0
112-
)
113-
y1 = (
114-
pose[i, j, 1] + self.radius
115-
if pose[i, j, 1] + self.radius < im_size[1]
116-
else im_size[1]
117-
)
98+
x0 = max(0, pose[i, j, 0] - self.radius)
99+
x1 = min(im_size[0], pose[i, j, 0] + self.radius)
100+
y0 = max(0, pose[i, j, 1] - self.radius)
101+
y1 = min(im_size[1], pose[i, j, 1] + self.radius)
118102
coords = [x0, y0, x1, y1]
119103
draw.ellipse(
120104
coords, fill=self.colors[j], outline=self.colors[j]

0 commit comments

Comments
 (0)