From 2027d5311d419addfb67ad94d45a198592b2e459 Mon Sep 17 00:00:00 2001 From: Maxime Desroches Date: Thu, 30 Jun 2022 15:03:15 -0700 Subject: [PATCH] tools: add support for nv12 in compressed_vipc (#24962) reshape for nv12 --- tools/camerastream/compressed_vipc.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/tools/camerastream/compressed_vipc.py b/tools/camerastream/compressed_vipc.py index 290610e45f..d321d6fd2b 100755 --- a/tools/camerastream/compressed_vipc.py +++ b/tools/camerastream/compressed_vipc.py @@ -73,6 +73,10 @@ def decoder(addr, sock_name, vipc_server, vst, nvidia): continue assert len(frames) == 1 img_yuv = frames[0].to_ndarray(format=av.video.format.VideoFormat('yuv420p')).flatten() + uv_offset = H*W + y = img_yuv[:uv_offset] + uv = img_yuv[uv_offset:].reshape(2, -1).ravel('F') + img_yuv = np.hstack((y, uv)) vipc_server.send(vst, img_yuv.data, cnt, int(time_q[0]*1e9), int(time.monotonic()*1e9)) cnt += 1