开源
MMdnn:跨框架模型转换/可视化等工具
来源:元经纪     阅读:1098
网站管理员
发布于 2023-01-29 08:15
查看主页

概述

一套用于帮助用户在不同深度学习框架之间进行互操作的工具。例如模型转换和可视化。在 Caffe、Keras、MXNet、TensorFlow 之间进行转换模型。

MMdnn 是一个全面的跨框架工具,用于转换、可视化和诊断深度学习 (DL) 模型。“MM”代表模型管理,“dnn”是深度神经网络的首字母缩写。

主要特点

安装

[hidecontent type="logged" desc="隐藏内容:登录后可查看"]

手动安装

您可以通过以下方式获得 MMdnn 的稳定版本

pip install mmdnn

并确保安装了Python,或者您可以通过以下方式尝试最新版本

[rihide]
pip install -U git+https://github.com/Microsoft/MMdnn.git@master

使用 docker 镜像安装

MMdnn 提供了一个 docker 镜像,其中封装了我们支持的 MMdnn 和深度学习框架以及其他依赖项。您可以通过以下步骤轻松尝试图像:

  1. 安装 Docker 社区版(CE)

    详细了解如何安装 docker

  2. 拉取 MMdnn docker 镜像

    docker pull mmdnn/mmdnn:cpu.small
  3. 以交互模式运行图像

    docker run -it mmdnn/mmdnn:cpu.small

特征

模型转换

在整个工业界和学术界,有许多现有的框架可供开发人员和研究人员设计模型,其中每个框架都有自己的网络结构定义和保存模型格式。框架之间的差距阻碍了模型的互操作。

我们提供了一个模型转换器来帮助开发人员通过中间表示格式在框架之间转换模型。

支持框架

[注意] 您可以点击链接获取每个框架的详细README。

测试模型

当前支持的框架之间的模型转换在一些ImageNet模型上进行了测试。

用法

一条命令实现转换。使用 TensorFlow ResNet V2 152到 PyTorch 作为我们的示例。

$ mmdownload -f tensorflow -n resnet_v2_152 -o ./
$ mmconvert -sf tensorflow -in imagenet_resnet_v2_152.ckpt.meta -iw imagenet_resnet_v2_152.ckpt --dstNodeName MMdnn_Output -df pytorch -om tf_resnet_to_pth.pth

完毕。

正在进行的框架

  • Torch7(需要帮助)
  • Chainer(需要帮助)

正在进行的模型

  • 人脸检测
  • 语义分割
  • 图像风格迁移
  • 物体检测
  • 循环神经网络

模型可视化

我们提供了一个本地可视化工具来显示深度学习模型的网络架构。请参考说明书

例子

官方教程

用户示例

[/hidecontent]

 
免责声明:本文为用户发表,不代表网站立场,仅供参考,不构成引导等用途。 开源
2.0版本“上新”!海量信息触手可及 “解锁”一张指尖上的就业地图
中国平安:数字化转型提升服务质效  金融科技助力高质量发展
连续阴雨天气 烘干机成抢手货
龙年最后一战!我国成功发射通信技术试验卫星十四号
骁龙8Gen4成本近1500元!台积电将提高5nm以下工艺报价:安卓旗舰都要涨价

首页

分类

定制方案

消息

我的