Caffe轉Pytorch模型系列教程 概述

嘗試了github上提供得諸多Caffe轉Pytorch模型的工具之後，我要使用的模型始終不能成功地轉換爲Pytorch模型。只得手動地把Caffe模型轉爲Pytorch模型。

零、知識儲備以及步驟說明

1、知識儲備

• 知道如何搭建Pytorch網絡。
• 知道如何閱讀Caffe的網絡。

2、步驟說明

• 首先把Caffe的網絡模型轉換爲Pytorch模型。
• 提取Caffe的權重，以便在Pytorch模型中使用。
• 調試模型，一般這步的時間最長。

一、把Caffe的模型轉換爲Pytorch模型

1、可視化Caffe的網絡模型

參見：把Caffe的模型轉換爲Pytorch模型。

2、從.caffemodel提取參數

參見：從.caffemodel/.caffemodel.h5提取Caffe模型的參數。

3、調試轉換後的Pytorch模型

參見：調試從Caffe模型轉換過來的Pytorch模型。

本系列博客用到的關鍵資源地址代碼：https://github.com/Mannix1994/SfSNet-Pytorch。
如果大家覺得這個系列的教程還闊以，麻煩大家在github上給我star！！！