【问题标题】:How to load caffemodel weights for retraining using Pycaffe?如何加载 caffemodel 权重以使用 Pycaffe 进行再训练?
【发布时间】:2017-03-06 08:20:46
【问题描述】:

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

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

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

【问题讨论】:

  • 没有。请注意,提出的重复问题没有回答我的问题。
  • 它在最后一部分
  • “求解器”、“其他功能”?在这两部分中,都没有解释如何加载 caffemodel 进行再训练。如果你看到它,请把它变成一个答案,

标签: caffe


【解决方案1】:

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

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

【讨论】:

    猜你喜欢
    • 2017-09-30
    • 1970-01-01
    • 1970-01-01
    • 2017-07-13
    • 1970-01-01
    • 2020-01-12
    • 2020-10-16
    • 1970-01-01
    • 2019-11-17
    相关资源
    最近更新 更多