文件名称：caffemodel2pytorch:将Caffe模型转换为PyTorch
文件大小：11KB
文件格式：ZIP
更新时间：2024-06-19 03:40:09
converter caffe pytorch Python
此转换器可用于将 Caffe 代码和层移植到 PyTorch。 特征： 将 caffemodel 的权重转储为 hdf5、npy、pt 和 json 格式 加载 Caffe 模型并从 PyTorch 使用它们 模拟 PyCaffe API 以允许平滑移植使用 Caffe 的代码（用于将 train/eval 中的后端更改为脚本如下）： 网络、Blob、SGDSolver 包装 Caffe 的 Python 层（参见 OICR 示例） PyTorch 中 ROI 池化的示例，无需手动编译 CUDA 代码（参见 OICR 示例） 层支持不像那样完整。 目前它支持以下 Caffe 层： 卷积（num_output、kernel_size、stride、pad、dilation；常数和高斯权重/偏置填充） 内部产品（num_output；常数和高斯权重/偏置填充物） 最大/平均池化（
【文件预览】：
caffemodel2pytorch-master
----README.md(16KB)
----caffemodel2pytorch.py(20KB)