gpt4 book ai didi

python - 在 Python 中使用 caffe.NetSpec() 时,train.prototxt 中的两个顶部 blob

转载 作者:太空宇宙 更新时间:2023-11-04 05:13:03 24 4
gpt4 key购买 nike

我在 python 中使用 caffe.NetSpec 使用以下代码在体系结构中定义和导出:

conv1_1 = L.Convolution(data,top='conv1_1',name='conv1_1',
convolution_param=
{'kernel_size':3,'num_output':64,'pad':1},
param=[{'lr_mult':1, 'decay_mult':1},
{'lr_mult':2,'decay_mult':0}])

但是,当生成 train.protxt 时,层中出现两个 Top blob,如下所示:

layer {
name: "conv1_1"
type: "Convolution"
bottom: "Data1"
top: "Convolution1"
top: "conv1_1"
param {
lr_mult: 1
decay_mult: 1
}
param {
lr_mult: 2
decay_mult: 0
}
convolution_param {
num_output: 64
pad: 1
kernel_size: 3
}
}

这里出了什么问题?谢谢

最佳答案

如何使用 NetSpec() 对象?

import caffe

ns = caffe.NetSpec() # use this object to store the layers
ns.data, ns.label = L.Data(name='data', ntop=2,
data_param={'source':'', 'batch_size': 32})
ns.conv1_1 = L.Convolution(ns.data, name='conv1_1',
convolution_param=
{'kernel_size':3,'num_output':64,'pad':1},
param=[{'lr_mult':1, 'decay_mult':1},
{'lr_mult':2,'decay_mult':0}])
print str(ns.to_proto()) # print the net stored in ns object

关于python - 在 Python 中使用 caffe.NetSpec() 时,train.prototxt 中的两个顶部 blob,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/42494235/

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com