Method encode_from_tensor

bench/python/nvcodec_utils.py:403–438

(self, tensor)

    # docs_tag: begin_imp_nvvideoencoder

    def encode_from_tensor(self, tensor):

        # Create a CUDA array interface object wit 2 planes one for luma and CrCb for NV12
        objCAI = []
        # Need to compute the address of the Y plane and the interleaved chroma plane
        data = (
            tensor.storage().data_ptr()
            + tensor.storage_offset() * tensor.element_size()
        )
        objCAI.append(
            AppCAI(
                (self.avstream.height, self.avstream.width, 1),
                (self.avstream.width, 1, 1),
                "|u1",
                data,
            )
        )
        chromaAlloc = int(data) + self.avstream.width * self.avstream.height
        objCAI.append(
            AppCAI(
                (int(self.avstream.height / 2), int(self.avstream.width / 2), 2),
                (self.avstream.width, 2, 1),
                "|u1",
                chromaAlloc,
            )
        )
        # Encode the frame takes CUDA array interface object as input
        self.encoded_frame = self.nvEnc.Encode(objCAI)
        self.write_frame(
            self.encoded_frame,
            self.pts_time,
            self.fps,
            self.avstream,
            self.container,
        )
        self.pts_time += self.delta_t

    # docs_tag: end_imp_nvvideoencoder