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

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

PyTorch 到 Caffe 的模型转换工具

PyTorch2Caffe 是一个可以将 Pytorch 模型转换为 Caffe 模型的工具,支持多种网络结构(好像对upsampling支持还不太友好)。具体方法可以见下方代码实例:

返回顶部