0

を使用してprototxt concate層を生成するには:私はPythonでカフェNetSpecを使用してprototxt上に生成したい次のように私はprototxtを持ってどのようにPythonの

layer { 
    name: "data" 
    type: "HDF5Data" 
    top: "data1" 
    top: "data2" 
    top: "label" 
    include { 
    phase: TRAIN 
    } 
    hdf5_data_param { 
    source: "./source_list.txt" 
    batch_size: 2 
    shuffle: true 
    } 
} 
layer { 
    name: "concat" 
    type: "Concat" 
    bottom: "data1" 
    bottom: "data2" 
    top: "data" 
    concat_param { 
    concat_dim:1 
    } 
} 

。しかし、それは間違っていた。これは私のコードです。それを修正するのを助けてください。おかげ

from caffe import layers as L 
... 
n = caffe.NetSpec() 
n.data, n.label = L.HDF5Data(top=["data1", "data2"], batch_size=2, 
          source="./source_list.txt", ntop=2,shuffle= True, 
          include={'phase': caffe.TRAIN}) 
n.concat = L.Concat(n.data["data1"], n.data["data2"]) 

答えて

3

は、あなたはそれが今で働いていた2つのdata出力

n.data1, n.data2, n.label = L.HDF5Data(ntop=3, name="data", 
             hdf5_data_param={'source': "./source_list.txt", 
                 'shuffle': True, 
                 'batch_size': 2} 
             include={'phase': caffe.TRAIN}) 
n.data = L.Concat(n.data1, n.data2, name="concat", concat_param={'concat_dim':1}) 
+1

を持っている必要があります。ありがとう – user8264

+0

@ user8264それはあなたのために働いてうれしい。古いコメントを削除できるようになりました。 – Shai

関連する問題