MMdnn - 用于转换深度神经网络模型的跨框架解决方案


MIT
跨平台
Python

软件简介

MMdnn 是微软开源的一套帮助用户在不同深度学习框架之间进行互操作的工具,包括模型转换和可视化。目前支持在
Caffe、Keras、MXNet、Tensorflow、CNTK、PyTorch 和 CoreML 等框架之间进行模型转换。

MMdnn 是一个用于转换、可视化和诊断深度神经网络模型的综合跨框架解决方案。MMdnn 中的「MM」代表模型管理,「dnn」是「deep neural
network」深度神经网络的缩写。

MMdnn 可将一个框架训练的 DNN 模型转换到成其他框架可用的模型。主要特性如下:

  • 模型文件转换器 ,转换 DNN 模型使其适合不同框架;

  • 模型代码块生成器 ,为框架生成适合的训练或推断代码块;

  • 模型可视化 ,可视化框架的 DNN 网络架构和参数;

  • 模型兼容性测试 (正在进行)。