Skip to content

Provide a Separate Interface in Cameras that Outputs PIL Images #287

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 1 commit into from
Jul 15, 2024
Merged

Provide a Separate Interface in Cameras that Outputs PIL Images #287

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 4 additions & 1 deletion leads_video/camera.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
from typing import override as _override

from PIL.Image import fromarray as _fromarray
from PIL.Image import fromarray as _fromarray, Image as _Image
from cv2 import VideoCapture as _VideoCapture, cvtColor as _cvtColor, COLOR_BGR2RGB as _COLOR_BGR2RGB
from numpy import ndarray as _ndarray, pad as _pad, array as _array

Expand Down Expand Up @@ -51,6 +51,9 @@ def read(self) -> _ndarray | None:
def read_numpy(self) -> _ndarray | None:
return self.read()

def read_pil(self) -> _Image | None:
return None if (frame := self.read_numpy()) is None else _fromarray(frame)

@_override
def close(self) -> None:
self._video_capture.release()
Expand Down