p2p-ld/numpydantic
1
0
Fork 0
mirror of https://github.com/p2p-ld/numpydantic.git synced 2025-01-09 21:44:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

lint

Browse source
This commit is contained in:
sneakers-the-rat 2024-05-20 19:27:30 -07:00
parent 2d39c46b89
commit ce74a0482a
Signed by untrusted user who does not match committer: jonny
GPG key ID: 6DCB96EF1E4D232D
1 changed files with 4 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
src/numpydantic/interface/video.py
View file

@ -84,8 +84,9 @@ class VideoProxy:
Note that this order flips the order of height and width from typical resolution
specifications: eg. 1080p video is typically 1920x1080, but here it would be
1080x1920. This follows opencv's ordering, which matches expectations when
eg. an image is read and plotted with matplotlib: the first index is the position
in the 0th dimension - the height, or "y" axis - and the second is the width/x.
eg. an image is read and plotted with matplotlib: the first index is the
position in the 0th dimension - the height, or "y" axis - and the second is the
width/x.
"""
if self._shape is None:
self._shape = (self.n_frames, *self.sample_frame.shape)
@ -117,7 +118,7 @@ class VideoProxy:
self._n_frames = int(n_frames)
return self._n_frames
def _get_frame(self, frame: int):
def _get_frame(self, frame: int) -> np.ndarray:
self.video.set(cv2.CAP_PROP_POS_FRAMES, frame)
status, frame = self.video.read()
if not status: # pragma: no cover