如何使用Pycaffe加载caffemodel权重进行再培训?

时间:2017-03-06 08:20:46

标签: caffe

如何加载caffemodel权重(与求解器一起)以便随后使用Python进行再训练?

换句话说,Pycaffe中以下命令行的对应部分是什么?

build/tools/caffe train \
-solver models/finetune_flickr_style/solver.prototxt \
-weights models/bvlc_reference_caffenet/bvlc_reference_caffenet.caffemodel 

1 个答案:

答案 0 :(得分:4)

可以通过get_solver()net.copy_from()函数在Pycaffe中加载用于使用特定解算器进行微调的caffemodel:

solver=caffe.get_solver('prototxtfile.prototxt')
solver.net.copy_from('weights.caffemodel')