如何加载caffemodel权重(与求解器一起)以便随后使用Python进行再训练?
换句话说,Pycaffe中以下命令行的对应部分是什么?
build/tools/caffe train \
-solver models/finetune_flickr_style/solver.prototxt \
-weights models/bvlc_reference_caffenet/bvlc_reference_caffenet.caffemodel
答案 0 :(得分:4)
可以通过get_solver()
和net.copy_from()
函数在Pycaffe中加载用于使用特定解算器进行微调的caffemodel:
solver=caffe.get_solver('prototxtfile.prototxt')
solver.net.copy_from('weights.caffemodel')