基于caffe的图像分类(3)——修改网络并训练模型

本文介绍了如何基于Caffe进行GoogLeNet图像分类的实践操作,包括复制模型文件到指定目录、修改train_val.txt数据层、调整solver配置,以及微调模型的注意事项。训练完成后,需要结合labels.txt、mydata_mean.binaryproto和预训练模型文件进行分类。
摘要由CSDN通过智能技术生成

前面做好了lmdb和均值文件,下面以Googlenet为例修改网络并训练模型。


我们将caffe-master\models下的bvlc_googlenet文件夹复制到caffe-master\examples\imagenet下。(因为我们的lmdb和均值都在这里,放一起方便些)

打开train_val.txt,修改:

1.修改data层:

layer {
  name: "data"
  type: "Data"
  top: "data"
  top: "label"
  include {
    phase: TRAIN
  }
  transform_param {
    mirror: true
    crop_size: 224
	mean_file: "examples/imagenet/mydata_mean.binaryproto" #均值文件
    #mean_value: 104 #这些注释掉
    #mean_value: 117
    #mean_value: 123
  }
  data_param {
    source: "examples/imagenet/mydata_train_lmdb" #训练集的lmdb
    batch_size: 32 #根据GPU修改
    backend: LMDB
  }
}

layer {
  name: "data"
  type: "Data"
  top: "data"
  top: "label"
  include {
    phase: TEST
  }
  transform_param {
    mirror: f
评论 10
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值