diff --git a/video_decoder.py b/video_decoder.py index d36f305..ae37526 100644 --- a/video_decoder.py +++ b/video_decoder.py @@ -39,7 +39,7 @@ class VideoDecoder(PyOperator): elif self._sample_type.lower() == SAMPLE_TYPE.UNIFORM_TEMPORAL_SUBSAMPLE: end_time = self._end_time if self._end_time is not None and self._end_time <= duration else duration start_time = self._start_time if self._start_time is not None else 0 - nums = int(stream.rate * (end_time - start_time)) + nums = int(stream.average_rate * (end_time - start_time)) return partial(self._uniform_temporal_subsample, total_frames=nums) elif self._sample_type.lower() == SAMPLE_TYPE.TIME_STEP_SAMPLE: start_time = self._start_time if self._start_time is not None else 0