Caffe tools/caffe的使用

Ezio_1320

于 2019-07-26 11:38:19 发布

阅读量400

点赞数

CC 4.0 BY-SA版权

分类专栏： Caffe

本文链接：https://blog.youkuaiyun.com/weixin_37029453/article/details/97376410

Caffe 专栏收录该内容

1 篇文章

订阅专栏

本文详细介绍Caffe工具的使用方法，包括如何通过命令行调用Caffe进行模型训练、测试、设备查询及时间基准测试等操作。同时，文章还列举了各种参数设置，如GPU选择、迭代次数、模型定义、预训练权重加载等。

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

Caffe tools/caffe 的使用

当caffe编译之后会有在 /caffe/build/tools/ 路径下会有一个caffe可执行文件，我们在命令行输入 ./caffe/build/tools/caffe 会有caffe这个可执行文件的说明，如下图。

image_1cfs1erohu8hetn25111868029.png-131.3kB

看一看都有神马命令！

caffe: command line brew
usage: caffe <command> <args>

commands:
  train           train or finetune a model                    训练或者调整模型
  test            score a model                                给一个模型打分
  device_query    show GPU diagnostic information              显示GPU的诊断信息
  time            benchmark model execution time               检测一个模型的运行时间

  Flags from tools/caffe.cpp:
  
    控制你用哪一个GPU，如果全用则-gpu all：
    -gpu (Optional; run in GPU mode on given device IDs separated by ','.Use
      '-gpu all' to run on all available GPUs. The effective training batch
      size is multiplied by the number of devices.) type: string default: ""
      
    设置你训练和测试的迭代次数 
    -iterations (The number of iterations to run.) type: int32 default: 50
    
    -level (Optional; network level.) type: int32 default: 0
    
    -model 后接 net.prototxt 文件，此文件设置了网络的结构
    -model (The model definition protocol buffer text file.) type: string
      default: ""
      
    -phase (Optional; network phase (TRAIN or TEST). Only used for 'time'.)
      type: string default: ""
    -sighup_effect (Optional; action to take when a SIGHUP signal is received:
      snapshot, stop or none.) type: string default: "snapshot"
    -sigint_effect (Optional; action to take when a SIGINT signal is received:
      snapshot, stop or none.) type: string default: "stop"
    -snapshot (Optional; the snapshot solver state to resume training.)
      type: string default: ""
    
    -solver 后接你的solver.prototxt文件
    -solver (The solver definition protocol buffer text file.) type: string
      default: ""
      
    -stage (Optional; network stages (not to be confused with phase), separated
      by ','.) type: string default: ""
      
    -weights 后接你之前训练好的 .caffemodel文件，存储的是之前得到的参数们。
    -weights (Optional; the pretrained weights to initialize finetuning,
      separated by ','. Cannot be set simultaneously with snapshot.)
      type: string default: ""