將Caffe轉化為Pytorch

1. 1、將Caffe Model轉化為Pytorch Model
2. 2、將Caffe Model轉化為Torch Model
3. 3、將Torch Model轉化為Pytorch Model

我已經測試了vgg16，它表現在分類任務。但我不能保證它在其他任務上執行得很好（如物件檢測和語義分割）。您可以嘗試並根據bug資訊修改程式碼。如果有你的咖啡模型新的元件，你應該在程式碼中新增相應的部分。Github地址：https://github.com/fanq15/caffe_to_torch_to_pytorch

安裝Torch和Caffe

把Caffe轉換為Torch

1. 將path改變成自己的path
2. 把 .prototxt 和 .caffemodel 放在同一個資料夾下.
3. 會生成vgg16_torch.t7資料夾.

th caffemodel_to_t7.lua

把torch轉換為pytorch

python convert_torch.py -m vgg16_torch.t7

將會建立2個檔案vgg16_torch.py vgg16_torch.pth

在python中載入.pth model

1. 確保vgg16_torch.py 和 vgg16_torch.pth 檔案在python工作區的同一個資料夾下
2. 匯入vgg16_torch意味著從vgg16_torch.py匯入模型結構。
3. model.load_state_dict
    代表從vgg16_torch.pth模型中家在權重

import vgg16_torch

model = vgg16_torch.vgg16_torch
model.load_state_dict(torch.load('vgg16_torch.pth'))
model.eval()
...

參考資源

在這裡，我整理髮布了Pytorch中文文件，方便大家查詢使用，同時也準備了中文論壇，歡迎大家學習交流！

Pytorch中文文件已經發布，完美翻譯，更加方便大家瀏覽：

Torch中文網：https://ptorch.com/

Pytorch中文網：https://ptorch.com/

Pytorch中文文件：https://ptorch.com/docs/1/