mirror of
https://github.com/HiMeditator/auto-caption.git
synced 2026-02-04 04:14:42 +08:00
f2aa075e65
- 重构 GummyTranslator 类，增加启动和停止方法 - 优化 AudioStream 类，添加读取音频数据方法 - 更新 main-gummy.py，使用新的 GummyTranslator 和 AudioStream 接口 - 更新文档和 TODO 列表
23 lines
598 B
Python
23 lines
598 B
Python
import numpy as np
|
|
|
|
def mergeStreamChannels(data, channels):
|
|
"""
|
|
将当前多通道流数据合并为单通道流数据
|
|
|
|
Args:
|
|
data: 多通道数据
|
|
channels: 通道数
|
|
|
|
Returns:
|
|
mono_data_bytes: 单通道数据
|
|
"""
|
|
# (length * channels,)
|
|
data_np = np.frombuffer(data, dtype=np.int16)
|
|
# (length, channels)
|
|
data_np_r = data_np.reshape(-1, channels)
|
|
# (length,)
|
|
mono_data = np.mean(data_np_r.astype(np.float32), axis=1)
|
|
mono_data = mono_data.astype(np.int16)
|
|
mono_data_bytes = mono_data.tobytes()
|
|
return mono_data_bytes
|