当前位置: 代码迷 >> 综合 >> caffe使用训练好的model
  详细解决方案

caffe使用训练好的model

热度:9   发布时间:2024-02-09 02:53:47.0

1、mean file

将所有数据的均值保存为文件,如果需要用到均值文件,图片会先减去均值再进行训练和测试,会提高速度和精度。对于imageNet库则需要均值文件。

在网络文件中,数据层配置均值文件

2、deploy文件

1)输入和data相关的层去掉

2)输出和loss相关的层去掉

3)加入

input:“data”

input_shape{

    dim:1 #batchsize 每次forward输入的图片数

    dim:3 #channels

    dim:28#width

    dim:28#height

}

4)输出层需要重新建立输出

layer

{

    name:"prob"

    type:"Softmax"

    bottom:"ip2"

    top:"prob"

}

可以参考caffe中自带的例子

  相关解决方案