请教一下caffe中的model的参数和deploy文件的网络是怎么对应的

用别人的训练好model和deploy文件运行自己的数据,在deploy文件中删了最后一层,仍能运行。所以有个疑问,请教一下caffe中的model的参数和deploy文件的网络是怎么对应的
已邀请:

Evence

赞同来自: 很遥远

按照层名字对应的,也就是说model训练的时候train.prototxt有conv1、conv2、ip1、ip2,而你使用这个model的时候网络中只会对新的prototxt中的对应叫做“conv1、conv2、ip1、ip2”层进行model的参数初始化。

如果我的另一个网络train.prototxt是conv1、conv3、conv5、ip1、ip3,那我用这个model初始化的时候只会初始化conv1和ip1的参数,其他层该怎么初始化就怎么初始化。

这也就是为什么有些网络用VGG来跑Imagenet之后可以用来作为与训练模型,就是名称对应上就可以用了,后面的softmax那些不对应的就不会有影响。

很遥远

赞同来自:

维文训练的时候,我是二分类,同事是三分类,都可以在同一个模型预训练的模型上继续训练。前面可以对应上的参数都对应上,最后一个全连接不同可能就初始化为零什么的吧

要回复问题请先登录注册