From 1e4181ff84a41c399ec176fd13a13522f1fda24f Mon Sep 17 00:00:00 2001 From: Jael Gu Date: Mon, 13 Jun 2022 12:20:47 +0800 Subject: [PATCH] Add layout in output Signed-off-by: Jael Gu --- audio_decoder.py | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/audio_decoder.py b/audio_decoder.py index 128176e..f82ee13 100644 --- a/audio_decoder.py +++ b/audio_decoder.py @@ -5,7 +5,13 @@ import av from towhee.operator.base import Operator -AudioOutput = NamedTuple("Outputs", [("audio_frame", 'ndarray'), ('sample_rate', 'int'), ("TIMESTAMP", 'int')]) +AudioOutput = NamedTuple( + "Outputs", + [("audio_frame", 'ndarray'), + ('sample_rate', 'int'), + ('layout', 'str'), + ("TIMESTAMP", 'int')] +) class AudioDecoder(Operator): @@ -21,5 +27,6 @@ class AudioDecoder(Operator): for frame in in_container.decode(stream): timestamp = int(frame.time * 1000) sample_rate = frame.sample_rate + layout = frame.layout.name ndarray = frame.to_ndarray() - yield AudioOutput(ndarray, sample_rate, timestamp) + yield AudioOutput(ndarray, sample_rate, layout, timestamp)