Skip to content

fanq15/caffe_to_torch_to_pytorch

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

12 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Convert caffe to pytorch

Convert caffe to torch

Convert torch to pytorch

  • 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。

Convert caffe to torch

  • 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

Convert torch to pytorch

python convert_torch.py -m vgg16_torch.t7

Two file will be created vgg16_torch.py vgg16_torch.pth

Load the .pth model in python

  • Make sure the vgg16_torch.py and vgg16_torch.pth files in the same folder with the python workspace.
  • The import vgg16_torch means importing the model structure from the vgg16_torch.py.
  • The model.load_state_dict means loading weights from vgg16_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()
...

Acknowledgement

About

No description or website provided.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published