def test_video2ndarray_mp4():
vpath = os.path.join(DIR_PATH, "test.mp4")
if not os.path.exists(vpath):
mp4_src = os.path.join(DOWNLOAD_SERVER_PREFIX,
DOWNLOAD_SRC_DIR,
"test.mp4")
download(mp4_src, vpath)
# read video to varray
varray = backend.video2ndarray(vpath, cin="BGR", cout="RGB")
print(varray.shape)
def test_ndarray2video_mp4():
vpath = os.path.join(DIR_PATH, "test.mp4")
if not os.path.exists(vpath):
mp4_src = os.path.join(DOWNLOAD_SERVER_PREFIX, DOWNLOAD_SRC_DIR,
"test.mp4")
download(mp4_src, vpath)
# read video to varray
varray = backend.video2ndarray(vpath, cin="BGR", cout="RGB")
# dump varray to video
vpath = os.path.join(DIR_PATH, "ndarray2video.mp4")
backend.ndarray2video(varray, vpath)
def test_ndarray2frames_mp4():
vpath = os.path.join(DIR_PATH, "test.mp4")
if not os.path.exists(vpath):
mp4_src = os.path.join(DOWNLOAD_SERVER_PREFIX, DOWNLOAD_SRC_DIR,
"test.mp4")
download(mp4_src, vpath)
# read video to varray
varray = backend.video2ndarray(vpath, cin="BGR", cout="RGB")
# dump varray to frames
fdir = os.path.join(DIR_PATH, "ndarray2frames.frames.mp4.d")
ret, f_n = backend.ndarray2frames(varray, fdir, cin="RGB", cout="BGR")
print('Dumping frames from varray finished, {} frames'.format(f_n))