上传者: 42116681
|
上传时间: 2022-01-18 12:42:22
|
文件大小: 11KB
|
文件类型: -
此转换器可用于将 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;常数和高斯权重/偏置填充物)
最大/平均池化(