We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 242a5a7 commit 0f8a072Copy full SHA for 0f8a072
test/test_videoapi.py
@@ -64,7 +64,9 @@ def test_frame_reading(self):
64
av_pts, vr_pts = [], []
65
# get av frames
66
for av_frame in av_reader.decode(av_reader.streams.video[0]):
67
- av_frames.append(torch.tensor(av_frame.to_rgb(src_colorspace="ITU709").to_ndarray()).permute(2, 0, 1))
+ av_frames.append(
68
+ torch.tensor(av_frame.to_rgb(src_colorspace="ITU709").to_ndarray()).permute(2, 0, 1)
69
+ )
70
av_pts.append(av_frame.pts * av_frame.time_base)
71
72
# get vr frames
0 commit comments