2ab73c85eee57d710227eabcb5c5e37fb6d6bc52,mmcv/video/io.py,VideoReader,__getitem__,#VideoReader#Any#,241

Before Change

    def __getitem__(self, index):
        if isinstance(index, slice):
            raise RuntimeError("slice has not been supported yet")
        return self.get_frame(index)

    def __iter__(self):
        self._set_real_position(0)

After Change

        return self.frame_cnt

    def __getitem__(self, index):
        if isinstance(index, slice):
            return [self.get_frame(i) for i in range(*index.indices(self.frame_cnt))]
        return self.get_frame(index)

    def __iter__(self):
        self._set_real_position(0)

Instances