fix bug

Signed-off-by: junjie.jiang <junjie.jiang@zilliz.com>
2 years ago · b257e1d110
1 changed files with 3 additions and 0 deletions
--- a/video_decoder.py
+++ b/video_decoder.py
@ -76,6 +76,9 @@ class VideoDecoder(PyOperator):
                yield frame
            count += 1

+        if cur_index < len(indexs):
+            yield frame
+
    def _filter(self, frames):
        for f in frames:
            if self._end_time_ms and f.timestamp > self._end_time_ms: