VisualDL - 跨框架深度学习可视化框架


Apache 2.0
跨平台
C/C++

软件简介

VisualDL,即 Visualize the Deep Learning,是一个面向深度学习任务设计的可视化工具,包含了
scalar、参数分布、模型结构、图像可视化等功能,项目正处于高速迭代中,新的组件会不断加入。

目前大多数 DNN 平台均使用 Python 作为配置语言,VisualDL 原生支持 Python 的使用, 通过在模型的 Python
配置中添加几行,便可以为训练过程提供丰富的可视化支持。

除了 Python SDK 之外,VisualDL 底层采用 C 编写,其暴露的 C SDK 也可以集成到其他平台中, 实现原生的性能和定制效果。

组件

VisualDL 目前提供以下组件:

  • scalar
  • histogram
  • image
  • audio
  • graph
  • high dimensional

graph

兼容 ONNX(Open Neural Network Exchange),通过与
python SDK的结合,VisualDL 可以兼容包括 PaddlePaddle、pytorch、mxnet 在内的大部分主流 DNN 平台。

scalar

可以用于展示训练测试的误差趋势

image

可以用于可视化任何tensor,或模型生成的图片

histogram

用于可视化任何 tensor 中元素分布的变化趋势

audio

音频可用于播放输入音频样本或生成的音频样本。

high dimensional

通过将 high-dimensional 数据投影到 2D / 3D中,high dimensional 可用于可视化数据嵌入。