NN-Tools:神经网络工具集(转换/构建/分析器)

PyTorch入门实战教程

NN-Tools 是一款方便易用的神经网络工具集,提供了面向 Caffe/PyTorch/TensorFlow 等主流神经网络框架的转换器/构建器/分析器。具有以下特性:

  1. 转换不同框架之间的模型。(Caffe,PyTorch,MXNet)
  2. 提供了方便的工具,可用于操作 caffe 模型和 ProtoTXT 文件(比如获取网络层的权重)。
  3. 分析模型,获取每一层的运算数。

模型转换

Pytorch to Caffe

这个工具支持 PyTorch 几乎所有的版本(从 0.2.0 到 0.4.0)。

  • 支持的层: conv2d, linear, max_pool2d, avg_pool2d, dropout, relu, prelu, threshold(only value=0), softmax, batch_norm
  • 支持的操作: torch.split, torch.max, torch.cat
  • 支持的tensor操作: var.view, + (add), += (iadd), -(sub), -=(isub) * (mul) *= (imul)

示例:将 Alexnet 的 PyTorch 模型转换成 Caffe

示例:将 Inception3 的 PyTorch 模型转换成 Caffe

示例:将 VGG19 的 PyTorch 模型转换成 Caffe

工具地址:Github

PyTorch入门实战教程
除特别注明外,本站所有文章均为 PyTorch 中文网原创,转载请注明出处:https://www.pytorchtutorial.com/nn_tools/

Leave a Reply

Your email address will not be published. Required fields are marked *

返回顶部