深度学习（八）caffe源码学习-未完待续

最新推荐文章于 2018-06-06 12:02:08 发布

转载最新推荐文章于 2018-06-06 12:02:08 发布 · 1.2k 阅读

文章标签：

#深度学习

深度学习专栏收录该内容

15 篇文章

订阅专栏

本文详细介绍了如何使用 Caffe 进行模型训练，包括如何选择 GPU、加载预训练模型进行 fine-tuning 的方法，并提及了特征提取的相关命令。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

本文主要详细讲解caffe的直接调用方法。

一、训练相关

[python]view plain copy 
   
 #!/usr/bin/env sh  
 TOOLS=../cafferead/build/tools  
 $TOOLS/caffe train --solver=gender_solver.prorotxt  -gpu all  #加入 -gpu 选项  

-gpu 可以选择gpu的id号，如果是 -gpu all表示启用所有的GPU进行训练。

[python]view plain copy 
   
 $TOOLS/caffe train --solver=gender_solver.prorotxt -weights gender_net.caffemodel #加入-weights  

加入-weights，这个功能很好用，也经常会用到，因为现在的CNN相关的文献，很多都是在已有的模型基础上，进行fine-tuning，因为我们大部分人都缺少训练数据，不像谷歌、百度这些土豪公司，有很多人专门做数据标注，对于小公司而言，往往缺少标注好的训练数据。因此我们一般使用fine-tuning的方法，在少量数据的情况下，尽可能的提高精度。我们可以使用：-weights 选项，利用已有的模型训练好的参数，作为初始值，进行继续训练。

二、特征提前相关命令