pytorch轉caffe
不支持pytorch1.1,但是可以將ptorch1.1的模型轉化爲較低版本1.0->0.4
基於pytorch1.1完成轉換
1.拷貝caffe文件夾和pytorch_2_caffe.py文件到工程中
2.寫轉化腳本pytorch2caffe/light_openpose2Caffe.py
備註:pycharm工程中所有路徑都是相對工程根目錄的路徑
python 不同路徑下加載模塊
ValueError: source code string cannot contain null bytes
vim查看文件加密無法打開
from google.protobuf.internal import enum_type_wrapper ModuleNotFoundError: No module named 'google'
pip install protobuf
python導入上級目錄中的模塊
RuntimeError: Error(s) in loading state_dict for PoseEstimationWithMobileNet:
Missing key(s) in state_dict: "model.0.0.weight", "model.0.1.weight", "model.0.1.bias", "model.0.1.running_mean", "model.0.1.running_var", "model.1.0.weight", "model.1.1.weight", "model.1.1.bias", "model.1.1.running_mean", "model.1.1.running_var", "model.1.3.weight", "model.1.4.weight", "model.1.4.bias", "model.1.4.running_mean", "model.1.4.running_var", "model.2.0.weight", "model.2.1.weight", "model.2.1.bias", "model.2.1.running_mean", "model.2.1.running_var", "model.2.3.weight", "model.2.4.weight", "model.2.4.bias", "model.2.4.running_mean", "model.2.4.running_var", "model.3.0.weight", "model.3.1.weight", "model.3.1.bias", "model.3.1.running_mean", "model.3.1.running_var", "model.3.3.weight", "model.3.4.weight", "model.3.4.bias", "model.3.4.running_mean", "model.3.4.running_var", "model.4.0.weight", "model.4.1.weight", "model.4.1.bias", "model.4.1.running_mean", "model.4.1.running_var", "model.4.3.weight", "model.4.4.weight", "model.4.4.bias", "model.4.4.running_mean", "model.4.4.running_var", "model.5.0.weight", "model.5.1.weight", "model.5.1.bias", "model.5.1.running_mean", "model.5.1.running_var", "model.5.3.weight", "model.5.4.weight", "model.5.4.bias", "model.5.4.running_mean", "model.5.4.running_var", "model.6.0.weight", "model.6.1.weight", "model.6.1.bias", "model.6.1.running_mean", "model.6.1.running_var", "model.6.3.weight", "model.6.4.weight", "model.6.4.bias", "model.6.4.running_mean", "model.6.4.running_var", "model.7.0.weight", "model.7.1.weight", "model.7.1.bias", "model.7.1.running_mean", "model.7.1.running_var", "model.7.3.weight", "model.7.4.weight", "model.7.4.bias", "model.7.4.running_mean", "model.7.4.running_var", "model.8.0.weight", "model.8.1.weight", "model.8.1.bias", "model.8.1.running_mean", "model.8.1.running_var", "model.8.3.weight", "model.8.4.weight", "model.8.4.bias", "model.8.4.running_mean", "model.8.4.running_var", "model.9.0.weight", "model.9.1.weight", "model.9.1.bias", "model.9.1.running_mean", "model.9.1.running_var", "model.9.3.weight", "model.9.4.weight", "model.9.4.bias", "model.9.4.running_mean", "model.9.4.running_var", "model.10.0.weight", "model.10.1.weight", "model.10.1.bias", "model.10.1.running_mean", "model.10.1.running_var", "model.10.3.weight", "model.10.4.weight", "model.10.4.bias", "model.10.4.running_mean", "model.10.4.running_var", "model.11.0.weight", "model.11.1.weight", "model.11.1.bias", "model.11.1.running_mean", "model.11.1.running_var", "model.11.3.weight", "model.11.4.weight", "model.11.4.bias", "model.11.4.running_mean", "model.11.4.running_var", "cpm.align.0.weight", "cpm.align.0.bias", "cpm.trunk.0.0.weight", "cpm.trunk.0.2.weight", "cpm.trunk.1.0.weight", "cpm.trunk.1.2.weight", "cpm.trunk.2.0.weight", "cpm.trunk.2.2.weight", "cpm.conv.0.weight", "cpm.conv.0.bias", "initial_stage.trunk.0.0.weight", "initial_stage.trunk.0.0.bias", "initial_stage.trunk.1.0.weight", "initial_stage.trunk.1.0.bias", "initial_stage.trunk.2.0.weight", "initial_stage.trunk.2.0.bias", "initial_stage.heatmaps.0.0.weight", "initial_stage.heatmaps.0.0.bias", "initial_stage.heatmaps.1.0.weight", "initial_stage.heatmaps.1.0.bias", "initial_stage.pafs.0.0.weight", "initial_stage.pafs.0.0.bias", "initial_stage.pafs.1.0.weight", "initial_stage.pafs.1.0.bias", "refinement_stages.0.trunk.0.initial.0.weight", "refinement_stages.0.trunk.0.initial.0.bias", "refinement_stages.0.trunk.0.trunk.0.0.weight", "refinement_stages.0.trunk.0.trunk.0.0.bias", "refinement_stages.0.trunk.0.trunk.0.1.weight", "refinement_stages.0.trunk.0.trunk.0.1.bias", "refinement_stages.0.trunk.0.trunk.0.1.running_mean", "refinement_stages.0.trunk.0.trunk.0.1.running_var", "refinement_stages.0.trunk.0.trunk.1.0.weight", "refinement_stages.0.trunk.0.trunk.1.0.bias", "refinement_stages.0.trunk.0.trunk.1.1.weight", "refinement_stages.0.trunk.0.trunk.1.1.bias", "refinement_stages.0.trunk.0.trunk.1.1.running_mean", "refinement_stages.0.trunk.0.trunk.1.1.running_var", "refinement_stages.0.trunk.1.initial.0.weight", "refinement_stages.0.trunk.1.initial.0.bias", "refinement_stages.0.trunk.1.trunk.0.0.weight", "refinement_stages.0.trunk.1.trunk.0.0.bias", "refinement_stages.0.trunk.1.trunk.0.1.weight", "refinement_stages.0.trunk.1.trunk.0.1.bias", "refinement_stages.0.trunk.1.trunk.0.1.running_mean", "refinement_stages.0.trunk.1.trunk.0.1.running_var", "refinement_stages.0.trunk.1.trunk.1.0.weight", "refinement_stages.0.trunk.1.trunk.1.0.bias", "refinement_stages.0.trunk.1.trunk.1.1.weight", "refinement_stages.0.trunk.1.trunk.1.1.bias", "refinement_stages.0.trunk.1.trunk.1.1.running_mean", "refinement_stages.0.trunk.1.trunk.1.1.running_var", "refinement_stages.0.trunk.2.initial.0.weight", "refinement_stages.0.trunk.2.initial.0.bias", "refinement_stages.0.trunk.2.trunk.0.0.weight", "refinement_stages.0.trunk.2.trunk.0.0.bias", "refinement_stages.0.trunk.2.trunk.0.1.weight", "refinement_stages.0.trunk.2.trunk.0.1.bias", "refinement_stages.0.trunk.2.trunk.0.1.running_mean", "refinement_stages.0.trunk.2.trunk.0.1.running_var", "refinement_stages.0.trunk.2.trunk.1.0.weight", "refinement_stages.0.trunk.2.trunk.1.0.bias", "refinement_stages.0.trunk.2.trunk.1.1.weight", "refinement_stages.0.trunk.2.trunk.1.1.bias", "refinement_stages.0.trunk.2.trunk.1.1.running_mean", "refinement_stages.0.trunk.2.trunk.1.1.running_var", "refinement_stages.0.trunk.3.initial.0.weight", "refinement_stages.0.trunk.3.initial.0.bias", "refinement_stages.0.trunk.3.trunk.0.0.weight", "refinement_stages.0.trunk.3.trunk.0.0.bias", "refinement_stages.0.trunk.3.trunk.0.1.weight", "refinement_stages.0.trunk.3.trunk.0.1.bias", "refinement_stages.0.trunk.3.trunk.0.1.running_mean", "refinement_stages.0.trunk.3.trunk.0.1.running_var", "refinement_stages.0.trunk.3.trunk.1.0.weight", "refinement_stages.0.trunk.3.trunk.1.0.bias", "refinement_stages.0.trunk.3.trunk.1.1.weight", "refinement_stages.0.trunk.3.trunk.1.1.bias", "refinement_stages.0.trunk.3.trunk.1.1.running_mean", "refinement_stages.0.trunk.3.trunk.1.1.running_var", "refinement_stages.0.trunk.4.initial.0.weight", "refinement_stages.0.trunk.4.initial.0.bias", "refinement_stages.0.trunk.4.trunk.0.0.weight", "refinement_stages.0.trunk.4.trunk.0.0.bias", "refinement_stages.0.trunk.4.trunk.0.1.weight", "refinement_stages.0.trunk.4.trunk.0.1.bias", "refinement_stages.0.trunk.4.trunk.0.1.running_mean", "refinement_stages.0.trunk.4.trunk.0.1.running_var", "refinement_stages.0.trunk.4.trunk.1.0.weight", "refinement_stages.0.trunk.4.trunk.1.0.bias", "refinement_stages.0.trunk.4.trunk.1.1.weight", "refinement_stages.0.trunk.4.trunk.1.1.bias", "refinement_stages.0.trunk.4.trunk.1.1.running_mean", "refinement_stages.0.trunk.4.trunk.1.1.running_var", "refinement_stages.0.heatmaps.0.0.weight", "refinement_stages.0.heatmaps.0.0.bias", "refinement_stages.0.heatmaps.1.0.weight", "refinement_stages.0.heatmaps.1.0.bias", "refinement_stages.0.pafs.0.0.weight", "refinement_stages.0.pafs.0.0.bias", "refinement_stages.0.pafs.1.0.weight", "refinement_stages.0.pafs.1.0.bias".
Unexpected key(s) in state_dict: "state_dict", "optimizer", "scheduler", "iter", "current_epoch".
net.load_state_dict(checkpoint,False)
增加False參數，否則會有
refinement_stage(torch.cat([backbone_features, stages_output[-2], stages_output[-1]], dim=1)))去掉 dim=1
Transform Completed完成轉換
3.文件壓縮
tar -cvf checkpoint_iter_370000.pth.tar checkpoint_iter_370000.pth
打包解包命令