Method get_next_frames

bench/python/nvcodec_utils.py:202–211

(self, N)

                    raise ValueError("Unexpected tensor layout, NCHW expected.")

    def get_next_frames(self, N):
        decoded_frames = list(itertools.islice(self.generate_decoded_frames(), N))
        if len(decoded_frames) == 0:
            return None
        elif len(decoded_frames) == 1:  # this case we dont need stack the tensor
            return decoded_frames[0]
        else:
            # convert from list of tensors to a single tensor (NHWC)
            tensorNHWC = cvcuda.stack(decoded_frames)
            return tensorNHWC


# docs_tag: end_imp_nvvideodecoder