After trying a lot of tools Caffe turn Pytorch model was provided on github, I always want to use the model can not be successfully converted to Pytorch model. Caffe model had to manually put into Pytorch model.
table of Contents
Zero knowledge base and step instructions
1, knowledge base
- Pytorch know how to build a network.
- You know how to read Caffe network.
2, Step Description
- First, the Caffe network model into Pytorch model.
- Caffe extracted weights for use in Pytorch model.
- Debugging model, this step is generally the longest time.
First, the model into Caffe model for Pytorch
1, the network model visualization Caffe
See also: the Caffe model into Pytorch model .
2, parameters are extracted from .caffemodel
See also: Parameter Caffe model extracted from .caffemodel / .caffemodel.h5 .
3, after the commissioning model conversion Pytorch
See also: Pytorch model debugging switch from Caffe model over .
The address code key resources used in this blog series: https://github.com/Mannix1994/SfSNet-Pytorch
If you think this tutorial series also wide to trouble you give me a star on github! ! !