Skip to content

Commit

Permalink
Merge pull request #23 from pupil-labs/video_tonumpy_fix
Browse files Browse the repository at this point in the history
fix attr name to extract video frames to numpy array
  • Loading branch information
domstoppable authored Oct 16, 2024
2 parents d10ecfa + 2ed293a commit 5b63298
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -85,7 +85,7 @@ def ts(self):

def to_numpy(self):
if self.audio_or_video == "video":
return np.array([frame.rgb for frame in self])
return np.array([frame.bgr for frame in self])
elif self.audio_or_video == "audio":
return np.array([frame.to_ndarray() for frame in self])

Expand Down

0 comments on commit 5b63298

Please sign in to comment.