SwanLab是一个由国内团队开源的机器学习实验跟踪工具,相比于Tensorboard有更丰富的功能、更友好的UI界面,以及更重要的云端同步、多人协作功能。
安装:pip install swanlab
音频图表预览,支持播放:
本教程主要介绍如何用swanlab.Audio
记录音频,支持numpy、文件路径这两种输入方式:
从numpy array创建
记录单个音频:
import numpy as np
import swanlab
run = swanlab.init()
# 创建一个numpy array类型的音频
white_noise = np.random.randn(2, 100000)
# 传入swanlab.Audio,设置采样率
audio = swanlab.Audio(white_noise, sample_rate=44100, caption="white_noise")
run.log({"examples": audio})
记录多个音频:
import numpy as np
import swanlab
run = swanlab.init()
# 创建一个列表
examples = []
for i in range(3):
white_noise = np.random.randn(2, 100000)
audio = swanlab.Audio(white_noise, sample_rate=44100, caption="audio_{i}")
# 列表中添加swanlab.Audio类型对象
examples.append(audio)
run.log({"examples": examples})
从文件路径创建
import swanlab
run = swanlab.init()
audio = swanlab.Audio("path/to/file")
run.log({"examples": audio})