Caffe turn Pytorch model series Tutorial Overview

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.

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! ! !

Published 88 original articles · won praise 132 · views 180 000 +

Guess you like

Origin blog.csdn.net/DumpDoctorWang/article/details/88716962
Recommended