从Caffe到PyTorch：模型转换工具caffemodel2pytorch

简介：
caffemodel2pytorch是一款专为深度学习开发者设计的高效模型转换工具，它能够轻松实现将基于Caffe框架训练得到的caffemodel文件迁移到PyTorch环境下的操作，极大地方便了科研人员和工程师在不同深度学习平台间的切换与资源共享。 此转换器可用于将 Caffe 代码和层移植到 PyTorch。其主要特性包括： - 将 caffemodel 的权重转储为 hdf5、npy、pt 和 json 格式。 - 加载 Caffe 模型并从 PyTorch 使用它们进行模拟。 - 提供类似于 PyCaffe API，以允许平滑移植使用 Caffe 编写的代码（例如在训练和评估中将后端更改为脚本）：网络、Blob、SGDSolver 等。 - 包装 Caffe 的 Python 层（参见 OICR 示例）。 - 提供 PyTorch 中 ROI 池化的示例，无需手动编译 CUDA 代码（参见 OICR 示例）。 目前层支持还不完全。它已知支持以下 Caffe 层： - 卷积：包括 num_output、kernel_size、stride、pad 和 dilation 参数；常数和高斯权重/偏置填充。 - 内部产品：num_output 参数，提供常数和高斯权重/偏置初始化选项。 - 最大池化与平均池化。