- Convert caffe model to pytorch model
- Convert caffe model to torch model
- Convert torch model to pytorch model
- I have tested on vgg16, it behaves well on classification tasks. But I can't guarantee it performs well on other tasks(such as object detection and semantic segmentation). You can try it and modify the code according the bug info. If there are new components in your caffe model, you should add corresponding parts in the code。
-
Change the path to your own path.
-
Put the
.prototxt
and.caffemodel
file in the same folder. -
You will get the
vgg16_torch.t7
file.
th caffemodel_to_t7.lua
python convert_torch.py -m vgg16_torch.t7
Two file will be created vgg16_torch.py
vgg16_torch.pth
- Make sure the
vgg16_torch.py
andvgg16_torch.pth
files in the same folder with the python workspace. - The
import vgg16_torch
means importing the model structure from thevgg16_torch.py
. - The
model.load_state_dict
means loading weights fromvgg16_torch.pth
into the model structure.
import vgg16_torch
model = vgg16_torch.vgg16_torch
model.load_state_dict(torch.load('vgg16_torch.pth'))
model.eval()
...
-
The caffe to torch code is modified from https://github.com/jcjohnson/pytorch-vgg
-
The torch to pytorch code is borrowed from https://github.com/clcarwin/convert_torch_to_pytorch