logo
Browse Source

Add layout in output

Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
main
Jael Gu 3 years ago
parent
commit
1e4181ff84
  1. 11
      audio_decoder.py

11
audio_decoder.py

@ -5,7 +5,13 @@ import av
from towhee.operator.base import Operator 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): class AudioDecoder(Operator):
@ -21,5 +27,6 @@ class AudioDecoder(Operator):
for frame in in_container.decode(stream): for frame in in_container.decode(stream):
timestamp = int(frame.time * 1000) timestamp = int(frame.time * 1000)
sample_rate = frame.sample_rate sample_rate = frame.sample_rate
layout = frame.layout.name
ndarray = frame.to_ndarray() ndarray = frame.to_ndarray()
yield AudioOutput(ndarray, sample_rate, timestamp)
yield AudioOutput(ndarray, sample_rate, layout, timestamp)

Loading…
Cancel
Save