INNER CODE UNIT · Python
last_ball
chonyy/ML-auto-baseball-pitching-overlay · src/get_pitch_frames.py:113
last_ball = tuple(tracked_balls[-1][:-1])
pitch_frames.append(FrameInfo(frame, True, last_ball, color))
last_tracked_frame = frame_id
result = cv2.cvtColor(frame, cv2.COLOR_RGB2BGR)
detection = cv2.resize((result), (0, 0), fx=0.5, fy=0.5)
cv2.imshow("result", detection)
if cv2.waitKey(50) & 0xFF == ord("q"):
break
frame_id += 1
# Use Polyfit to approximate the untracked balls
fill_lost_tracking(pitch_frames)
# Add five more frames after the last tracked frame
pitch_frames.extend(frames[last_tracked_frame : last_tracked_frame + 10])
return pitch_frames, width, height, fps