clips_at_random_timestamps¶
- torchcodec.samplers.clips_at_random_timestamps(decoder, *, num_clips: int = 1, num_frames_per_clip: int = 1, seconds_between_frames: Optional[float] = None, sampling_range_start: Optional[float] = None, sampling_range_end: Optional[float] = None, policy: Literal['repeat_last', 'wrap', 'error'] = 'repeat_last') FrameBatch [source]¶
Sample clips at random timestamps.
- Parameters:
decoder (VideoDecoder) – The
VideoDecoder
instance to sample clips from.num_clips (int, optional) – The number of clips to return. Default: 1.
num_frames_per_clip (int, optional) – The number of frames per clips. Default: 1.
seconds_between_frames (float or None, optional) – The time (in seconds) between each frame within a clip. More accurately, this defines the time between the frame sampling point, i.e. the timestamps at which we sample the frames. Because frames span intervals in time , the resulting start of frames within a clip may not be exactly spaced by
seconds_between_frames
- but on average, they will be. Default is None, which is set to the average frame duration (1/average_fps
).sampling_range_start (float or None, optional) – The start of the sampling range, which defines the first timestamp (in seconds) that a clip may start at. Default: None, which corresponds to the start of the video. (Note: some videos start at negative values, which is why the default is not 0).
sampling_range_end (float or None, optional) – The end of the sampling range, which defines the last timestamp (in seconds) that a clip may start at. This value is exclusive, i.e. a clip may only start within [
sampling_range_start
,sampling_range_end
). If None (default), the value is set automatically such that the clips never span beyond the end of the video, i.e. it is set toend_video_seconds - (num_frames_per_clip - 1) * seconds_between_frames
. When a clip spans beyond the end of the video, thepolicy
parameter defines how to construct such clip.policy (str, optional) –
Defines how to construct clips that span beyond the end of the video. This is best described with an example: assuming the last valid (seekable) timestamp in a video is 10.9, and a clip was sampled to start at timestamp 10.5, with
num_frames_per_clip=5
andseconds_between_frames=0.2
, the sampling timestamps of the frames in the clip are supposed to be [10.5, 10.7, 10.9, 11.1, 11.2]. But 11.1 and 11.2 are invalid timestamps, so thepolicy
parameter defines how to replace those frames, with valid sampling timestamps:”repeat_last”: repeats the last valid frame of the clip. We would get frames sampled at timestamps [10.5, 10.7, 10.9, 10.9, 10.9].
”wrap”: wraps around to the beginning of the clip. We would get frames sampled at timestamps [10.5, 10.7, 10.9, 10.5, 10.7].
”error”: raises an error.
Default is “repeat_last”. Note that when
sampling_range_end=None
(default), this policy parameter is unlikely to be relevant.
- Returns:
The sampled clips, as a 5D
FrameBatch
. The shape of thedata
field is (num_clips
,num_frames_per_clips
, …) where … is (H, W, C) or (C, H, W) depending on thedimension_order
parameter ofVideoDecoder
. The shape of thepts_seconds
andduration_seconds
fields is (num_clips
,num_frames_per_clips
).- Return type:
Examples using
clips_at_random_timestamps
:How to sample video clips