Status and usage of cudacodec::VideoWriter

Python
, , ,
18

@tvercaut If you have time can you test

The test below demonstrates how to use it

import cv2 as cv
import numpy as np
src = 'big_buck_bunny.mp4'
reader = cv.cudacodec.createVideoReader(src)
reader.set(cv.cudacodec.ColorFormat_NV_NV12)
fmt = reader.format()
stream = cv.cuda.Stream()
writer = cv.cudacodec.createVideoWriter('output.mp4', fmt.targetSz, codec=cv.cudacodec.HEVC, fps=30,
                                        colorFormat=cv.cudacodec.ColorFormat_NV_YUV410_10BIT, stream=stream)
ret, frame_nv12 = reader.nextFrame()
frame_nv12_16bit = cv.cuda_GpuMat(frame_nv12.size(), cv.CV_16U)
frame_yuv410_10bit = cv.cuda_GpuMat(frame_nv12.size(),cv.CV_16U)
for i in range(100):
    frame_nv12.convertTo(cv.CV_16U, stream, dst=frame_nv12_16bit)
    cv.cuda.lshift(frame_nv12_16bit, 8, dst=frame_yuv410_10bit, stream=stream)
    writer.write(frame_yuv410_10bit)
    reader.nextFrame(frame_nv12, stream=stream)
writer.release()