zxt881108 / caffe2pytorch-tsn

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

caffe2pytorch-tsn

This scipt first gets .hdf5file from caffemodel in python, then use this .hdf5 file to get torch model in lua, finally we transfer torch model to pytorch model in python.

(Maybe you can use 'load_lua' and 'torch.save' directly in pytorch, but you can find that there will be error when you use this pytorchmodel on test_videos.py.)

To be honest, this script is so complicated. I'm looking forward to see yjxiong's transferring work.

You also need to modify test_videos.py because of the problem of state_dict, please see details in test_videos.py

reference: https://github.com/kmatzen/googlenet-caffe2torch

About


Languages

Language:Python 53.3%Language:Lua 43.6%Language:Shell 3.1%