如何使用 Pycaffe 加载用于重新训练的 caffemodel 权重？

Question

如何加载 caffemodel 权重（与求解器一起）以用于随后使用 Python 进行再训练？

换句话说，Pycaffe 中以下命令行的对应物是什么？

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

Answer 1

可以在 Pycaffe 中通过get_solver()和net.copy_from()函数获得加载 caffemodel 以使用特定求解器进行微调：

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