将Caffe Model转化为Pytorch Model以及Torch Model

最新推荐文章于 2024-04-27 09:46:14 发布

转载最新推荐文章于 2024-04-27 09:46:14 发布 · 541 阅读

1 ·

CC 4.0 BY-SA版权

原文链接：https://my.oschina.net/earnp/blog/1457226

文章标签：

#人工智能 #python #lua

本文介绍如何将Caffe模型转换为Pytorch模型，并提供详细的步骤指导，包括安装依赖软件、转换流程及代码实现等。同时分享了一个已测试过的vgg16模型转换案例。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

为什么80%的码农都做不了架构师？>>>

将Caffe转化为Pytorch

1、将Caffe Model转化为Pytorch Model
2、将Caffe Model转化为Torch Model
3、将Torch Model转化为Pytorch Model

我已经测试了vgg16，它表现在分类任务。但我不能保证它在其他任务上执行得很好（如对象检测和语义分割）。您可以尝试并根据bug信息修改代码。如果有你的咖啡模型新的组件，你应该在代码中添加相应的部分。Github地址：https://github.com/fanq15/caffe_to_torch_to_pytorch

安装Torch和Caffe

1、Torch安装教程
2、Caffe安装教程

把Caffe转换为Torch

1、将path改变成自己的path
2、把 .prototxt 和 .caffemodel 放在同一个文件夹下.
3、会生成vgg16_torch.t7文件夹.

th caffemodel_to_t7.lua

把torch转换为pytorch

python convert_torch.py -m vgg16_torch.t7

将会创建2个文件vgg16_torch.py vgg16_torch.pth

在python中加载.pth model

1、确保vgg16_torch.py 和 vgg16_torch.pth 文件在python工作区的同一个文件夹下
2、导入vgg16_torch意味着从vgg16_torch.py导入模型结构。
3、model.load_state_dict 代表从vgg16_torch.pth模型中家在权重

import vgg16_torch

model = vgg16_torch.vgg16_torch
model.load_state_dict(torch.load('vgg16_torch.pth'))
model.eval()
...

参考资源

1、Caffe转换为Torch参考 https://github.com/jcjohnson/pytorch-vgg
2、Torch转Pytorch参考 https://github.com/clcarwin/convert_torch_to_pytorch

在这里，我整理发布了Pytorch中文文档，方便大家查询使用，同时也准备了中文论坛，欢迎大家学习交流！

Pytorch中文文档

Pytorch中文论坛

Pytorch中文文档已经发布，完美翻译，更加方便大家浏览：

Torch中文网：https://ptorch.com/

Pytorch中文网：https://ptorch.com/

Pytorch中文文档：https://ptorch.com/docs/1/

转载于:https://my.oschina.net/earnp/blog/1457226